import subprocess import sys try: # Make terminal colors work on windows import colorama colorama.init() except ImportError: pass def add_nanopb_builders(env): '''Add the necessary builder commands for nanopb tests.''' # Build command for building .pb from .proto using protoc def proto_actions(source, target, env, for_signature): dirs = ' '.join(['-I' + env.GetBuildPath(d) for d in env['PROTOCPATH']]) return '$PROTOC $PROTOCFLAGS %s -o%s %s' % (dirs, target[0], source[0]) proto_file_builder = Builder(generator = proto_actions, suffix = '.pb', src_suffix = '.proto') env.Append(BUILDERS = {'Proto': proto_file_builder}) env.SetDefault(PROTOC = 'protoc') env.SetDefault(PROTOCPATH = ['.']) # Build command for running nanopb generator import os.path def nanopb_targets(target, source, env): basename = os.path.splitext(str(source[0]))[0] target.append(basename + '.pb.h') return target, source nanopb_file_builder = Builder(action = '$NANOPB_GENERATOR $NANOPB_FLAGS $SOURCE', suffix = '.pb.c', src_suffix = '.pb', emitter = nanopb_targets) env.Append(BUILDERS = {'Nanopb': nanopb_file_builder}) env.SetDefault(NANOPB_GENERATOR = 'python ' + env.GetBuildPath("#../generator/nanopb_generator.py")) env.SetDefault(NANOPB_FLAGS = '-q') # Combined method to run both protoc and nanopb generator def run_protoc_and_nanopb(env, source): b1 = env.Proto(source) b2 = env.Nanopb(source) return b1 + b2 env.AddMethod(run_protoc_and_nanopb, "NanopbProto") # Build command that runs a test program and saves the output def run_test(target, source, env): if len(source) > 1: infile = open(str(source[1])) else: infile = None pipe = subprocess.Popen(str(source[0]), stdin = infile, stdout = open(str(target[0]), 'w'), stderr = sys.stderr) result = pipe.wait() if result == 0: print '\033[32m[ OK ]\033[0m Ran ' + str(source[0]) else: print '\033[31m[FAIL]\033[0m Program ' + str(source[0]) + ' returned ' + str(result) return result run_test_builder = Builder(action = run_test, suffix = '.output') env.Append(BUILDERS = {'RunTest': run_test_builder}) # Build command that decodes a message using protoc def decode_actions(source, target, env, for_signature): dirs = ' '.join(['-I' + env.GetBuildPath(d) for d in env['PROTOCPATH']]) return '$PROTOC $PROTOCFLAGS %s --decode=%s %s <%s >%s' % (dirs, env['MESSAGE'], source[1], source[0], target[0]) decode_builder = Builder(generator = decode_actions, suffix = '.decoded') env.Append(BUILDERS = {'Decode': decode_builder}) # Build command that asserts that two files be equal def compare_files(target, source, env): data1 = open(str(source[0]), 'rb').read() data2 = open(str(source[1]), 'rb').read() if data1 == data2: print '\033[32m[ OK ]\033[0m Files equal: ' + str(source[0]) + ' and ' + str(source[1]) return 0 else: print '\033[31m[FAIL]\033[0m Files differ: ' + str(source[0]) + ' and ' + str(source[1]) return 1 compare_builder = Builder(action = compare_files, suffix = '.equal') env.Append(BUILDERS = {'Compare': compare_builder})