def test_parse_typedefs(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
parser = ThriftParser(data.TEST_THRIFT_PATH)
parser._references = data.TEST_THRIFT_REFERENCES
expected_typedefs = data.TEST_THRIFT_TYPEDEFS
typedefs = parser._parse_typedefs()
self.assertEqual(typedefs, expected_typedefs)
def test_parse_endpoints_from_service_definition(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
parser = ThriftParser(data.TEST_THRIFT_PATH)
parser._references = data.TEST_THRIFT_REFERENCES
endpoints = parser._parse_endpoints_from_service_definition(data.TEST_THRIFT_SERVICE_DEFINITION)
expected_endpoints = data.TEST_THRIFT_SERVICE_ENDPOINTS
self.assertEqual(endpoints, expected_endpoints)
def test_unalias_type_circular(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT_CIRCULAR_TYPEDEFS
parser = ThriftParser(data.TEST_THRIFT_PATH)
parse_result = parser.parse()
for alias in data.TEST_THRIFT_CIRCULAR_TYPEDEFS:
with self.assertRaises(ThriftCLIError):
parse_result.unalias_type(alias)
def test_parse_enums(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
parser = ThriftParser(data.TEST_THRIFT_PATH)
parser._references = data.TEST_THRIFT_REFERENCES
expected_enums = data.TEST_THRIFT_ENUMS
enums = parser._parse_enums()
self.assertEqual(enums, expected_enums)
def test_parse_structs(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
parser = ThriftParser(data.TEST_THRIFT_PATH)
parser._references = data.TEST_THRIFT_REFERENCES
expected_structs = data.TEST_THRIFT_STRUCTS
structs = parser._parse_structs()
self.assertDictEqual(structs, expected_structs)
def test_unalias_type(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
parser = ThriftParser(data.TEST_THRIFT_PATH)
parse_result = parser.parse()
expected_unaliased = data.TEST_THRIFT_UNALIASED_TYPES
unaliased = {alias: parse_result.unalias_type(alias) for alias in data.TEST_THRIFT_UNALIASED_TYPES}
self.assertDictEqual(unaliased, expected_unaliased)
def test_parse_services(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
parser = ThriftParser(data.TEST_THRIFT_PATH)
parser._references = data.TEST_THRIFT_REFERENCES
parser._result = ThriftParseResult()
expected_services = data.TEST_THRIFT_SERVICES
services = parser._parse_services()
self.assertDictEqual(services, expected_services)
def test_apply_namespace_on_primitives(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
parser = ThriftParser(data.TEST_THRIFT_PATH)
parser.parse()
self.assertEqual(parser._apply_namespace('i32'), 'i32')
self.assertEqual(parser._apply_namespace('bool'), 'bool')
self.assertEqual(parser._apply_namespace('string'), 'string')
self.assertEqual(parser._apply_namespace('binary'), 'binary')
def test_parse_including(self, mock_load_file, mock_is_file):
mock_load_file.side_effect = [data.TEST_THRIFT_INCLUDING_CONTENT, data.TEST_THRIFT_INCLUDED_CONTENT]
mock_is_file.side_effect = lambda path: path == data.TEST_THRIFT_INCLUDED_PATH
parser = ThriftParser(data.TEST_THRIFT_INCLUDING_PATH, [data.TEST_THRIFT_DIR_PATH])
expected_parse_result = data.TEST_THRIFT_INCLUDING_PARSE_RESULT
parse_result = parser.parse()
expected_call_args_list = [mock.call(data.TEST_THRIFT_INCLUDING_PATH),
mock.call(data.TEST_THRIFT_INCLUDED_PATH)]
self.assertEqual(mock_load_file.call_args_list, expected_call_args_list)
self.assertEqual(parse_result, expected_parse_result)
def test_parse_struct_definitions(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
parser = ThriftParser(data.TEST_THRIFT_PATH)
parser._references = data.TEST_THRIFT_REFERENCES
expected_struct_definitions = {
data.TEST_THRIFT_STRUCT_REFERENCE: data.TEST_THRIFT_STRUCT_DEFINITION,
data.TEST_THRIFT_STRUCT_REFERENCE2: data.TEST_THRIFT_STRUCT_DEFINITION2,
data.TEST_THRIFT_STRUCT_REFERENCE3: data.TEST_THRIFT_STRUCT_DEFINITION3
}
struct_definitions = parser._parse_struct_definitions()
self.assertEqual(struct_definitions, expected_struct_definitions)
def test_parse_service_definitions(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
parser = ThriftParser(data.TEST_THRIFT_PATH)
parser._references = data.TEST_THRIFT_REFERENCES
expected_service_definitions = [
(data.TEST_THRIFT_SERVICE_REFERENCE, (data.TEST_THRIFT_SERVICE_DEFINITION, None)),
(data.TEST_THRIFT_SERVICE_REFERENCE2, (data.TEST_THRIFT_SERVICE_DEFINITION2, None)),
(data.TEST_THRIFT_SERVICE_REFERENCE3, (data.TEST_THRIFT_SERVICE_DEFINITION3,
data.TEST_THRIFT_SERVICE_REFERENCE))
]
service_definitions = parser._parse_service_definitions()
self.assertEqual(service_definitions, expected_service_definitions)
def test_apply_namespace_on_data_structures(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
parser = ThriftParser(data.TEST_THRIFT_PATH)
parser.parse()
self.assertEqual(parser._apply_namespace('set<bool>'), 'set<bool>')
self.assertEqual(
parser._apply_namespace('map<string, map<string, list<set<binary>>>>'),
'map<string, map<string, list<set<binary>>>>')
self.assertEqual(
parser._apply_namespace('list<set<%s>>' % data.TEST_THRIFT_TYPEDEF_ALIAS_NAME),
'list<set<%s>>' % data.TEST_THRIFT_TYPEDEF_ALIAS_REFERENCE)
self.assertEqual(
parser._apply_namespace('map<%s, list<%s>>' % (data.TEST_THRIFT_STRUCT_NAME, data.TEST_THRIFT_ENUM_NAME)),
'map<%s, list<%s>>' % (data.TEST_THRIFT_STRUCT_REFERENCE, data.TEST_THRIFT_ENUM_REFERENCE))
def test_apply_namespace_on_references(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
parser = ThriftParser(data.TEST_THRIFT_PATH)
parser.parse()
self.assertEqual(parser._apply_namespace(data.TEST_THRIFT_ENUM_NAME), data.TEST_THRIFT_ENUM_REFERENCE)
self.assertEqual(parser._apply_namespace(data.TEST_THRIFT_ENUM_NAME2), data.TEST_THRIFT_ENUM_REFERENCE2)
self.assertEqual(parser._apply_namespace(data.TEST_THRIFT_STRUCT_NAME), data.TEST_THRIFT_STRUCT_REFERENCE)
self.assertEqual(parser._apply_namespace(data.TEST_THRIFT_STRUCT_NAME2), data.TEST_THRIFT_STRUCT_REFERENCE2)
self.assertEqual(parser._apply_namespace(data.TEST_THRIFT_STRUCT_NAME3), data.TEST_THRIFT_STRUCT_REFERENCE3)
self.assertEqual(parser._apply_namespace(data.TEST_THRIFT_SERVICE_NAME), data.TEST_THRIFT_SERVICE_REFERENCE)
self.assertEqual(parser._apply_namespace(data.TEST_THRIFT_SERVICE_NAME2), data.TEST_THRIFT_SERVICE_REFERENCE2)
self.assertEqual(parser._apply_namespace(data.TEST_THRIFT_SERVICE_NAME3), data.TEST_THRIFT_SERVICE_REFERENCE3)
self.assertEqual(
parser._apply_namespace(data.TEST_THRIFT_TYPEDEF_ALIAS_NAME),
data.TEST_THRIFT_TYPEDEF_ALIAS_REFERENCE)
self.assertEqual(
parser._apply_namespace(data.TEST_THRIFT_TYPEDEF_ALIAS_NAME2),
data.TEST_THRIFT_TYPEDEF_ALIAS_REFERENCE2)
self.assertEqual(
parser._apply_namespace(data.TEST_THRIFT_TYPEDEF_ALIAS_NAME3),
data.TEST_THRIFT_TYPEDEF_ALIAS_REFERENCE3)
self.assertEqual(
parser._apply_namespace(data.TEST_THRIFT_TYPEDEF_ALIAS_NAME4),
data.TEST_THRIFT_TYPEDEF_ALIAS_REFERENCE4)
def test_parse(self, mock_load_file):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
parser = ThriftParser(data.TEST_THRIFT_PATH)
expected_parse_result = data.TEST_THRIFT_PARSE_RESULT
parse_result = parser.parse()
self.assertEqual(parse_result, expected_parse_result)