1*c8d645caSAndroid Build Coastguard Worker# Test the 'oneof' feature for generating C unions. 2*c8d645caSAndroid Build Coastguard Worker 3*c8d645caSAndroid Build Coastguard WorkerImport('env') 4*c8d645caSAndroid Build Coastguard Worker 5*c8d645caSAndroid Build Coastguard Workerimport re 6*c8d645caSAndroid Build Coastguard Worker 7*c8d645caSAndroid Build Coastguard Workermatch = None 8*c8d645caSAndroid Build Coastguard Workerif 'PROTOC_VERSION' in env: 9*c8d645caSAndroid Build Coastguard Worker match = re.search('([0-9]+).([0-9]+).([0-9]+)', env['PROTOC_VERSION']) 10*c8d645caSAndroid Build Coastguard Worker 11*c8d645caSAndroid Build Coastguard Workerif match: 12*c8d645caSAndroid Build Coastguard Worker version = list(map(int, match.groups())) 13*c8d645caSAndroid Build Coastguard Worker 14*c8d645caSAndroid Build Coastguard Worker# Oneof is supported by protoc >= 2.6.0 15*c8d645caSAndroid Build Coastguard Workerif env.GetOption('clean') or (match and (version[0] > 2 or (version[0] == 2 and version[1] >= 6))): 16*c8d645caSAndroid Build Coastguard Worker env.NanopbProto('oneof') 17*c8d645caSAndroid Build Coastguard Worker 18*c8d645caSAndroid Build Coastguard Worker enc = env.Program(['encode_oneof.c', 19*c8d645caSAndroid Build Coastguard Worker 'oneof.pb.c', 20*c8d645caSAndroid Build Coastguard Worker '$COMMON/pb_encode.o', 21*c8d645caSAndroid Build Coastguard Worker '$COMMON/pb_common.o']) 22*c8d645caSAndroid Build Coastguard Worker 23*c8d645caSAndroid Build Coastguard Worker dec = env.Program(['decode_oneof.c', 24*c8d645caSAndroid Build Coastguard Worker 'oneof.pb.c', 25*c8d645caSAndroid Build Coastguard Worker '$COMMON/pb_decode.o', 26*c8d645caSAndroid Build Coastguard Worker '$COMMON/pb_common.o']) 27*c8d645caSAndroid Build Coastguard Worker 28*c8d645caSAndroid Build Coastguard Worker env.RunTest("message1.pb", enc, ARGS = ['1']) 29*c8d645caSAndroid Build Coastguard Worker env.RunTest("message1.txt", [dec, 'message1.pb'], ARGS = ['1']) 30*c8d645caSAndroid Build Coastguard Worker env.RunTest("message2.pb", enc, ARGS = ['2']) 31*c8d645caSAndroid Build Coastguard Worker env.RunTest("message2.txt", [dec, 'message2.pb'], ARGS = ['2']) 32*c8d645caSAndroid Build Coastguard Worker env.RunTest("message3.pb", enc, ARGS = ['3']) 33*c8d645caSAndroid Build Coastguard Worker env.RunTest("message3.txt", [dec, 'message3.pb'], ARGS = ['3']) 34