encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-07-28 00:49:49 +03:00
Code Issues Packages Projects Releases Wiki Activity

add tests for generateschema command

Browse Source
This commit is contained in:
Dhaval Mehta 2020-05-26 02:30:20 +05:30
parent ce925970b4
commit f092e55ae1
1 changed files with 25 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tests/schemas/test_managementcommand.py
View File

@ -149,3 +149,28 @@ class GenerateSchemaTests(TestCase):
'--format=corejson', '--format=corejson',
stdout=self.out) stdout=self.out)
self.assertIn(expected_out, self.out.getvalue()) self.assertIn(expected_out, self.out.getvalue())
def test_accepts_tag_objects(self):
tag_objects = '[{"name": "pet", "description": "Pets operations"}, {"name": "store", "description": "Store ' \
'operations"}]'
call_command('generateschema',
'--tag_objects={}'.format(tag_objects),
stdout=self.out)
out_json = yaml.safe_load(self.out.getvalue())
assert out_json['tags'] == [
{
"name": "pet",
"description": "Pets operations"
},
{
"name": "store",
"description": "Store operations"
}
]
def test_rejects_invalid_tag_objects(self):
tag_objects = '[{"name": "pet", "description": "Pets operations"}'
with pytest.raises(SyntaxError):
call_command('generateschema',
'--tag_objects={}'.format(tag_objects),
stdout=self.out)