mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-27 03:54:01 +03:00
29 lines
975 B
Python
29 lines
975 B
Python
from django.db import models
|
|
from django.test import TestCase
|
|
|
|
from rest_framework.models import resolve_model
|
|
from rest_framework.tests.models import BasicModel
|
|
|
|
|
|
class ResolveModelTests(TestCase):
|
|
"""
|
|
`resolve_model` should return a Django model class given the
|
|
provided argument is a Django model class itself, or a properly
|
|
formatted string representation of one.
|
|
"""
|
|
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_non_django_model(self):
|
|
with self.assertRaises(ValueError):
|
|
resolve_model(TestCase)
|
|
|
|
def test_resolve_with_improper_string_representation(self):
|
|
with self.assertRaises(ValueError):
|
|
resolve_model('BasicModel')
|