def check_protos(self, src, expected): with tempfile.NamedTemporaryFile() as tf: tf.write(src) tf.flush() expected_proto = build_pb2.BuildLanguage() text_format.Merge(expected, expected_proto) extractor = macro_extractor.MacroDocExtractor() extractor.parse_bzl(tf.name) proto = extractor.proto() self.assertEqual(expected_proto, proto)
def check_protos(self, src, expected, load_symbols=[]): with tempfile.NamedTemporaryFile(mode='w+') as tf: tf.write(src) tf.flush() expected_proto = build_pb2.BuildLanguage() text_format.Merge(expected, expected_proto) extractor = rule_extractor.RuleDocExtractor() extractor.parse_bzl(tf.name, load_symbols) proto = extractor.proto() self.assertEqual(expected_proto, proto)
def check_protos(self, src, expected): tf = tempfile.NamedTemporaryFile(mode='w+', delete=False) tf.write(src) tf.flush() tf.close() expected_proto = build_pb2.BuildLanguage() text_format.Merge(expected, expected_proto) extractor = macro_extractor.MacroDocExtractor() extractor.parse_bzl(tf.name) proto = extractor.proto() os.remove(tf.name) self.assertEqual(expected_proto, proto)
def __init__(self): """Inits RuleDocExtractor with a new BuildLanguage proto""" self.__language = build_pb2.BuildLanguage() self.__extracted_rules = {} self.__load_symbols = []
def __init__(self): """Inits MacroDocExtractor with a new BuildLanguage proto""" self.__language = build_pb2.BuildLanguage() self.title = "" self.description = ""