diff --git a/tests/videodinges/unit/models/__init__.py b/tests/videodinges/unit/models/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/videodinges/unit/models/test_get_quality_by_name.py b/tests/videodinges/unit/models/test_get_quality_by_name.py new file mode 100644 index 0000000..6184035 --- /dev/null +++ b/tests/videodinges/unit/models/test_get_quality_by_name.py @@ -0,0 +1,15 @@ +from django.test import TestCase +from videodinges.models import get_quality_by_name + +class GetQualityByNameTestCase(TestCase): + + def test_returns_quality_if_listed(self): + result = get_quality_by_name('480p') + self.assertEqual(result.name, '480p') + self.assertEqual(result.width, 853) + self.assertEqual(result.height, 480) + self.assertEqual(result.priority, 2) + + def test_returns_none_if_not_listed(self): + result = get_quality_by_name('non-existend') + self.assertIsNone(result)