1*b1cdbd2cSJim Jagielski#!/usr/bin/env python 2*b1cdbd2cSJim Jagielski# ************************************************************* 3*b1cdbd2cSJim Jagielski# 4*b1cdbd2cSJim Jagielski# Licensed to the Apache Software Foundation (ASF) under one 5*b1cdbd2cSJim Jagielski# or more contributor license agreements. See the NOTICE file 6*b1cdbd2cSJim Jagielski# distributed with this work for additional information 7*b1cdbd2cSJim Jagielski# regarding copyright ownership. The ASF licenses this file 8*b1cdbd2cSJim Jagielski# to you under the Apache License, Version 2.0 (the 9*b1cdbd2cSJim Jagielski# "License"); you may not use this file except in compliance 10*b1cdbd2cSJim Jagielski# with the License. You may obtain a copy of the License at 11*b1cdbd2cSJim Jagielski# 12*b1cdbd2cSJim Jagielski# http://www.apache.org/licenses/LICENSE-2.0 13*b1cdbd2cSJim Jagielski# 14*b1cdbd2cSJim Jagielski# Unless required by applicable law or agreed to in writing, 15*b1cdbd2cSJim Jagielski# software distributed under the License is distributed on an 16*b1cdbd2cSJim Jagielski# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 17*b1cdbd2cSJim Jagielski# KIND, either express or implied. See the License for the 18*b1cdbd2cSJim Jagielski# specific language governing permissions and limitations 19*b1cdbd2cSJim Jagielski# under the License. 20*b1cdbd2cSJim Jagielski# 21*b1cdbd2cSJim Jagielski# ************************************************************* 22*b1cdbd2cSJim Jagielski 23*b1cdbd2cSJim Jagielski 24*b1cdbd2cSJim Jagielskiimport srclexer, srcparser, globals 25*b1cdbd2cSJim Jagielski 26*b1cdbd2cSJim Jagielskiclass TestCase: 27*b1cdbd2cSJim Jagielski 28*b1cdbd2cSJim Jagielski @staticmethod 29*b1cdbd2cSJim Jagielski def run (tokens, defines): 30*b1cdbd2cSJim Jagielski mcExpander = srcparser.MacroExpander(tokens, defines) 31*b1cdbd2cSJim Jagielski mcExpander.debug = True 32*b1cdbd2cSJim Jagielski mcExpander.expand() 33*b1cdbd2cSJim Jagielski tokens = mcExpander.getTokens() 34*b1cdbd2cSJim Jagielski print(tokens) 35*b1cdbd2cSJim Jagielski 36*b1cdbd2cSJim Jagielski @staticmethod 37*b1cdbd2cSJim Jagielski def simpleNoArgs (): 38*b1cdbd2cSJim Jagielski tokens = ['FUNC_FOO', '(', 'left', ',', 'right', ')'] 39*b1cdbd2cSJim Jagielski defines = {} 40*b1cdbd2cSJim Jagielski macro = globals.Macro('FUNC_FOO') 41*b1cdbd2cSJim Jagielski macro.tokens = ['Here', 'comes', 'X', 'and', 'Y'] 42*b1cdbd2cSJim Jagielski defines['FUNC_FOO'] = macro 43*b1cdbd2cSJim Jagielski TestCase.run(tokens, defines) 44*b1cdbd2cSJim Jagielski 45*b1cdbd2cSJim Jagielski @staticmethod 46*b1cdbd2cSJim Jagielski def simpleArgs (): 47*b1cdbd2cSJim Jagielski tokens = ['FUNC_FOO', '(', 'left', ',', 'right', ')'] 48*b1cdbd2cSJim Jagielski defines = {} 49*b1cdbd2cSJim Jagielski macro = globals.Macro('FUNC_FOO') 50*b1cdbd2cSJim Jagielski macro.tokens = ['Here', 'comes', 'X', 'and', 'Y'] 51*b1cdbd2cSJim Jagielski macro.vars['X'] = 0 52*b1cdbd2cSJim Jagielski macro.vars['Y'] = 1 53*b1cdbd2cSJim Jagielski defines['FUNC_FOO'] = macro 54*b1cdbd2cSJim Jagielski TestCase.run(tokens, defines) 55*b1cdbd2cSJim Jagielski 56*b1cdbd2cSJim Jagielski @staticmethod 57*b1cdbd2cSJim Jagielski def multiTokenArgs (): 58*b1cdbd2cSJim Jagielski tokens = ['FUNC_FOO', '(', 'left1', 'left2', 'left3', ',', 'right', ')'] 59*b1cdbd2cSJim Jagielski defines = {} 60*b1cdbd2cSJim Jagielski macro = globals.Macro('FUNC_FOO') 61*b1cdbd2cSJim Jagielski macro.tokens = ['Here', 'comes', 'X', 'and', 'Y'] 62*b1cdbd2cSJim Jagielski macro.vars['X'] = 0 63*b1cdbd2cSJim Jagielski macro.vars['Y'] = 1 64*b1cdbd2cSJim Jagielski defines['FUNC_FOO'] = macro 65*b1cdbd2cSJim Jagielski TestCase.run(tokens, defines) 66*b1cdbd2cSJim Jagielski 67*b1cdbd2cSJim Jagielski @staticmethod 68*b1cdbd2cSJim Jagielski def nestedTokenArgs (): 69*b1cdbd2cSJim Jagielski tokens = ['FUNC_BAA', '(', 'left', ',', 'right', ')'] 70*b1cdbd2cSJim Jagielski defines = {} 71*b1cdbd2cSJim Jagielski macro = globals.Macro('FUNC_FOO') 72*b1cdbd2cSJim Jagielski macro.tokens = ['Here', 'comes', 'X', 'and', 'Y'] 73*b1cdbd2cSJim Jagielski macro.vars['X'] = 0 74*b1cdbd2cSJim Jagielski macro.vars['Y'] = 1 75*b1cdbd2cSJim Jagielski defines['FUNC_FOO'] = macro 76*b1cdbd2cSJim Jagielski macro = globals.Macro('FUNC_BAA') 77*b1cdbd2cSJim Jagielski macro.tokens = ['FUNC_FOO'] 78*b1cdbd2cSJim Jagielski defines['FUNC_BAA'] = macro 79*b1cdbd2cSJim Jagielski TestCase.run(tokens, defines) 80*b1cdbd2cSJim Jagielski 81*b1cdbd2cSJim Jagielskidef main (): 82*b1cdbd2cSJim Jagielski print("simple expansion with no arguments") 83*b1cdbd2cSJim Jagielski TestCase.simpleNoArgs() 84*b1cdbd2cSJim Jagielski print("simple argument expansion") 85*b1cdbd2cSJim Jagielski TestCase.simpleArgs() 86*b1cdbd2cSJim Jagielski print("multi-token argument expansion") 87*b1cdbd2cSJim Jagielski TestCase.multiTokenArgs() 88*b1cdbd2cSJim Jagielski print("nested argument expansion") 89*b1cdbd2cSJim Jagielski TestCase.nestedTokenArgs() 90*b1cdbd2cSJim Jagielski 91*b1cdbd2cSJim Jagielskiif __name__ == '__main__': 92*b1cdbd2cSJim Jagielski main() 93