def test_compile_referenced(self):
test_files = [('test.smd', '''\
struct Struct1
Struct2 b
struct Struct2
struct Struct3
struct Struct4
''')]
with create_test_files(test_files) as input_dir:
input_path = os.path.join(input_dir, 'test.smd')
output_path = os.path.join(input_dir, 'test.json')
argv = [
'python3 -m schema_markdown', 'compile', input_path, '-o',
output_path, '--referenced', 'Struct1', '--referenced',
'Struct3'
]
with unittest_mock.patch('sys.stdout', new=StringIO()) as stdout, \
unittest_mock.patch('sys.stderr', new=StringIO()) as stderr, \
unittest_mock.patch('sys.argv', argv):
main()
self.assertEqual(stdout.getvalue(), '')
self.assertEqual(stderr.getvalue(), '')
with open(output_path, 'r', encoding='utf-8') as output_file:
self.assertEqual(
output_file.read(), '''\
{
"title": "Index",
"types": {
"Struct1": {
"struct": {
"members": [
{
"name": "b",
"type": {
"user": "******"
}
}
],
"name": "Struct1"
}
},
"Struct2": {
"struct": {
"name": "Struct2"
}
},
"Struct3": {
"struct": {
"name": "Struct3"
}
}
}
}''')
def test_compile_stdin_stdout(self):
argv = ['python3 -m schema_markdown', 'compile']
with unittest_mock.patch('sys.stdin', new=StringIO(TEST_SCHEMA_MARKDOWN)), \
unittest_mock.patch('sys.stdout', new=StringIO()) as stdout, \
unittest_mock.patch('sys.stderr', new=StringIO()) as stderr, \
unittest_mock.patch('sys.argv', argv):
main()
self.assertEqual(stderr.getvalue(), '')
self.assertEqual(stdout.getvalue(), TEST_MODEL)
def test_compile_multiple(self):
test_files = [('test.smd', TEST_SCHEMA_MARKDOWN),
('test2.smd', 'struct MyStruct2 (MyStruct)')]
with create_test_files(test_files) as input_dir:
input_path = os.path.join(input_dir, 'test.smd')
input_path2 = os.path.join(input_dir, 'test2.smd')
output_path = os.path.join(input_dir, 'test.json')
argv = [
'python3 -m schema_markdown', 'compile', input_path2,
input_path, '-o', output_path
]
with unittest_mock.patch('sys.stdout', new=StringIO()) as stdout, \
unittest_mock.patch('sys.stderr', new=StringIO()) as stderr, \
unittest_mock.patch('sys.argv', argv):
main()
self.assertEqual(stdout.getvalue(), '')
self.assertEqual(stderr.getvalue(), '')
with open(output_path, 'r', encoding='utf-8') as output_file:
self.assertEqual(
output_file.read(), '''\
{
"title": "Index",
"types": {
"MyStruct": {
"struct": {
"members": [
{
"name": "a",
"type": {
"builtin": "int"
}
},
{
"name": "b",
"optional": true,
"type": {
"builtin": "bool"
}
}
],
"name": "MyStruct"
}
},
"MyStruct2": {
"struct": {
"bases": [
"MyStruct"
],
"name": "MyStruct2"
}
}
}
}''')
def test_validate_no_type(self):
argv = ['python3 -m schema_markdown', 'validate']
with unittest_mock.patch('sys.stdout', new=StringIO()) as stdout, \
unittest_mock.patch('sys.stderr', new=StringIO()) as stderr, \
unittest_mock.patch('sys.argv', argv):
with self.assertRaises(SystemExit):
main()
self.assertEqual(stdout.getvalue(), '')
self.assertEqual(
stderr.getvalue().splitlines()[-1],
'schema-markdown validate: error: the following arguments are required: -s, -t'
)
def test_compile_error_stdin(self):
argv = ['python3 -m schema_markdown', 'compile']
with unittest_mock.patch('sys.stdin', new=StringIO('asdf')), \
unittest_mock.patch('sys.stdout', new=StringIO()) as stdout, \
unittest_mock.patch('sys.stderr', new=StringIO()) as stderr, \
unittest_mock.patch('sys.argv', argv):
with self.assertRaises(SystemExit) as cm_exc:
main()
self.assertEqual(cm_exc.exception.code, 1)
self.assertEqual(stderr.getvalue(), '''\
:1: error: Syntax error''')
self.assertEqual(stdout.getvalue(), '')
def test_no_command(self):
argv = ['python3 -m schema_markdown']
with unittest_mock.patch('sys.stdout', new=StringIO()) as stdout, \
unittest_mock.patch('sys.stderr', new=StringIO()) as stderr, \
unittest_mock.patch('sys.argv', argv):
with self.assertRaises(SystemExit):
main()
self.assertEqual(stdout.getvalue(), '')
self.assertEqual(
stderr.getvalue(), '''\
usage: schema-markdown [-h] {compile,validate} ...
schema-markdown: error: the following arguments are required: command
''')
def test_validate_stdin_value_error(self):
test_files = [('test.smd', TEST_SCHEMA_MARKDOWN)]
with create_test_files(test_files) as input_dir:
schema_path = os.path.join(input_dir, 'test.smd')
argv = [
'python3 -m schema_markdown', 'validate', '-s', schema_path,
'-t', 'MyStruct'
]
with unittest_mock.patch('sys.stdin', new=StringIO('{}')), \
unittest_mock.patch('sys.stdout', new=StringIO()), \
unittest_mock.patch('sys.stderr', new=StringIO()), \
unittest_mock.patch('sys.argv', argv):
with self.assertRaises(ValidationError):
main()
def test_compile_stdin(self):
with create_test_files([]) as input_dir:
output_path = os.path.join(input_dir, 'test.json')
argv = ['python3 -m schema_markdown', 'compile', '-o', output_path]
with unittest_mock.patch('sys.stdin', new=StringIO(TEST_SCHEMA_MARKDOWN)), \
unittest_mock.patch('sys.stdout', new=StringIO()) as stdout, \
unittest_mock.patch('sys.stderr', new=StringIO()) as stderr, \
unittest_mock.patch('sys.argv', argv):
main()
self.assertEqual(stderr.getvalue(), '')
self.assertEqual(stdout.getvalue(), '')
with open(output_path, 'r', encoding='utf-8') as output_file:
self.assertEqual(output_file.read(), TEST_MODEL)
def test_validate_stdin(self):
test_files = [('test.smd', TEST_SCHEMA_MARKDOWN)]
with create_test_files(test_files) as input_dir:
schema_path = os.path.join(input_dir, 'test.smd')
argv = [
'python3 -m schema_markdown', 'validate', '-s', schema_path,
'-t', 'MyStruct'
]
with unittest_mock.patch('sys.stdin', new=StringIO(TEST_VALUE)), \
unittest_mock.patch('sys.stdout', new=StringIO()) as stdout, \
unittest_mock.patch('sys.stderr', new=StringIO()) as stderr, \
unittest_mock.patch('sys.argv', argv):
main()
self.assertEqual(stdout.getvalue(), '')
self.assertEqual(stderr.getvalue(), '')
def test_compile_error(self):
test_files = [('test.smd', 'asdf')]
with create_test_files(test_files) as input_dir:
input_path = os.path.join(input_dir, 'test.smd')
argv = ['python3 -m schema_markdown', 'compile', input_path]
with unittest_mock.patch('sys.stdout', new=StringIO()) as stdout, \
unittest_mock.patch('sys.stderr', new=StringIO()) as stderr, \
unittest_mock.patch('sys.argv', argv):
with self.assertRaises(SystemExit) as cm_exc:
main()
self.assertEqual(cm_exc.exception.code, 1)
self.assertEqual(stderr.getvalue().replace(input_dir, '<input_dir>'),
'''\
<input_dir>/test.smd:1: error: Syntax error''')
self.assertEqual(stdout.getvalue(), '')
def test_compile_compact(self):
test_files = [('test.smd', TEST_SCHEMA_MARKDOWN)]
with create_test_files(test_files) as input_dir:
input_path = os.path.join(input_dir, 'test.smd')
output_path = os.path.join(input_dir, 'test.json')
argv = [
'python3 -m schema_markdown', 'compile', input_path, '-o',
output_path, '--compact'
]
with unittest_mock.patch('sys.stdout', new=StringIO()) as stdout, \
unittest_mock.patch('sys.stderr', new=StringIO()) as stderr, \
unittest_mock.patch('sys.argv', argv):
main()
self.assertEqual(stdout.getvalue(), '')
self.assertEqual(stderr.getvalue(), '')
with open(output_path, 'r', encoding='utf-8') as output_file:
self.assertEqual(output_file.read(),
json.dumps(json.loads(TEST_MODEL)))
def test_compile_title(self):
test_files = [('test.smd', TEST_SCHEMA_MARKDOWN)]
with create_test_files(test_files) as input_dir:
input_path = os.path.join(input_dir, 'test.smd')
output_path = os.path.join(input_dir, 'test.json')
argv = [
'python3 -m schema_markdown', 'compile', input_path, '-o',
output_path, '-t', 'My Type Model'
]
with unittest_mock.patch('sys.stdout', new=StringIO()) as stdout, \
unittest_mock.patch('sys.stderr', new=StringIO()) as stderr, \
unittest_mock.patch('sys.argv', argv):
main()
self.assertEqual(stdout.getvalue(), '')
self.assertEqual(stderr.getvalue(), '')
with open(output_path, 'r', encoding='utf-8') as output_file:
self.assertEqual(
output_file.read(),
TEST_MODEL.replace('"title": "Index"',
'"title": "My Type Model"'))
