def test_resolve_django_model(self): resolved_model = _resolve_model(BasicModel) self.assertEqual(resolved_model, BasicModel)
def test_resolve_string_representation(self): resolved_model = _resolve_model('tests.BasicModel') self.assertEqual(resolved_model, BasicModel)
def test_resolve_string_representation(self): resolved_model = _resolve_model('tests.BasicModel') assert resolved_model == BasicModel
def test_resolve_unicode_representation(self): resolved_model = _resolve_model(six.text_type('tests.BasicModel')) assert resolved_model == BasicModel
def test_resolve_django_model(self): resolved_model = _resolve_model(BasicModel) assert resolved_model == BasicModel
def test_resolve_non_django_model(self): with self.assertRaises(ValueError): _resolve_model(TestCase)
def test_resolve_improper_string_representation(self): with self.assertRaises(ValueError): _resolve_model('BasicModel')
def test_resolve_unicode_representation(self): resolved_model = _resolve_model(six.text_type('tests.BasicModel')) self.assertEqual(resolved_model, BasicModel)
def test_blows_up_if_model_does_not_resolve(self): with self.assertRaises(ImproperlyConfigured): _resolve_model('tests.BasicModel')
def get_related_model(field): # Backward compatibility patch for Django versions lower than 1.9.x # Function taken from DRF 3.6.x if DJANGO_VERSION < (1, 9): return _resolve_model(field.rel.to) return field.remote_field.model