converted test utils asserts to pytest

This commit is contained in:
Asif Saifuddin Auvi 2017-01-04 14:31:15 +06:00
parent 519804b948
commit 1b0c0a4660

View File

@ -65,85 +65,62 @@ class BreadcrumbTests(TestCase):
""" """
def test_root_breadcrumbs(self): def test_root_breadcrumbs(self):
url = '/' url = '/'
self.assertEqual( assert get_breadcrumbs(url) == [('Root', '/')]
get_breadcrumbs(url),
[('Root', '/')]
)
def test_resource_root_breadcrumbs(self): def test_resource_root_breadcrumbs(self):
url = '/resource/' url = '/resource/'
self.assertEqual( assert get_breadcrumbs(url) == [
get_breadcrumbs(url), ('Root', '/'), ('Resource Root', '/resource/')
[
('Root', '/'),
('Resource Root', '/resource/')
] ]
)
def test_resource_instance_breadcrumbs(self): def test_resource_instance_breadcrumbs(self):
url = '/resource/123' url = '/resource/123'
self.assertEqual( assert get_breadcrumbs(url) == [
get_breadcrumbs(url),
[
('Root', '/'), ('Root', '/'),
('Resource Root', '/resource/'), ('Resource Root', '/resource/'),
('Resource Instance', '/resource/123') ('Resource Instance', '/resource/123')
] ]
)
def test_resource_instance_customname_breadcrumbs(self): def test_resource_instance_customname_breadcrumbs(self):
url = '/resource/customname' url = '/resource/customname'
self.assertEqual( assert get_breadcrumbs(url) == [
get_breadcrumbs(url),
[
('Root', '/'), ('Root', '/'),
('Resource Root', '/resource/'), ('Resource Root', '/resource/'),
('Foo', '/resource/customname') ('Foo', '/resource/customname')
] ]
)
def test_nested_resource_breadcrumbs(self): def test_nested_resource_breadcrumbs(self):
url = '/resource/123/' url = '/resource/123/'
self.assertEqual( assert get_breadcrumbs(url) == [
get_breadcrumbs(url),
[
('Root', '/'), ('Root', '/'),
('Resource Root', '/resource/'), ('Resource Root', '/resource/'),
('Resource Instance', '/resource/123'), ('Resource Instance', '/resource/123'),
('Nested Resource Root', '/resource/123/') ('Nested Resource Root', '/resource/123/')
] ]
)
def test_nested_resource_instance_breadcrumbs(self): def test_nested_resource_instance_breadcrumbs(self):
url = '/resource/123/abc' url = '/resource/123/abc'
self.assertEqual( assert get_breadcrumbs(url) == [
get_breadcrumbs(url),
[
('Root', '/'), ('Root', '/'),
('Resource Root', '/resource/'), ('Resource Root', '/resource/'),
('Resource Instance', '/resource/123'), ('Resource Instance', '/resource/123'),
('Nested Resource Root', '/resource/123/'), ('Nested Resource Root', '/resource/123/'),
('Nested Resource Instance', '/resource/123/abc') ('Nested Resource Instance', '/resource/123/abc')
] ]
)
def test_broken_url_breadcrumbs_handled_gracefully(self): def test_broken_url_breadcrumbs_handled_gracefully(self):
url = '/foobar' url = '/foobar'
self.assertEqual( assert get_breadcrumbs(url) == [('Root', '/')]
get_breadcrumbs(url),
[('Root', '/')]
)
def test_modelviewset_resource_instance_breadcrumbs(self): def test_modelviewset_resource_instance_breadcrumbs(self):
url = '/resources/1/' url = '/resources/1/'
self.assertEqual( assert get_breadcrumbs(url) == [
get_breadcrumbs(url),
[
('Root', '/'), ('Root', '/'),
('Resource List', '/resources/'), ('Resource List', '/resources/'),
('Resource Instance', '/resources/1/') ('Resource Instance', '/resources/1/')
] ]
)
class ResolveModelTests(TestCase): class ResolveModelTests(TestCase):
@ -154,15 +131,15 @@ class ResolveModelTests(TestCase):
""" """
def test_resolve_django_model(self): def test_resolve_django_model(self):
resolved_model = _resolve_model(BasicModel) resolved_model = _resolve_model(BasicModel)
self.assertEqual(resolved_model, BasicModel) assert resolved_model == BasicModel
def test_resolve_string_representation(self): def test_resolve_string_representation(self):
resolved_model = _resolve_model('tests.BasicModel') resolved_model = _resolve_model('tests.BasicModel')
self.assertEqual(resolved_model, BasicModel) assert resolved_model == BasicModel
def test_resolve_unicode_representation(self): def test_resolve_unicode_representation(self):
resolved_model = _resolve_model(six.text_type('tests.BasicModel')) resolved_model = _resolve_model(six.text_type('tests.BasicModel'))
self.assertEqual(resolved_model, BasicModel) assert resolved_model == BasicModel
def test_resolve_non_django_model(self): def test_resolve_non_django_model(self):
with self.assertRaises(ValueError): with self.assertRaises(ValueError):