import from kontor-flask

flask/tests/test_tysc_model.py

@@ -0,0 +1,53 @@
+"""
+This module cantains tests for the TYSC data model
+"""
+import unittest
+from . import TestBase
+from kontor.tysc import initialize_model
+from kontor.tysc.models import Sport, Position, Team, Player
+from kontor.tysc.models import Manufacturer, CardSet, ParallelSet, InsertSet, Card
+
+
+class TestTYSCModel(TestBase):
+    """
+    This TestCase contains tests for TYSC data model.
+    """
+    _initialize_db_ = True
+
+    def setUp(self):
+        if self._initialize_db_:
+            initialize_model()
+            self._initialize_db_ = False
+
+    def tearDown(self):
+        pass
+
+    def test_sport_model(self):
+        self.assertEqual(Sport.objects.all().count(), 4)
+
+    def test_position_model(self):
+        self.assertEqual(Position.objects.all().count(), 25)
+
+    def test_team_model(self):
+        self.assertEqual(Team.objects.all().count(), 122)
+
+    def test_manufacturer_model(self):
+        self.assertEqual(Manufacturer.objects.all().count(), 8)
+
+    def test_cardset_model(self):
+        self.assertEqual(CardSet.objects.all().count(), 11)
+
+    def test_parallelset_model(self):
+        self.assertEqual(ParallelSet.objects.all().count(), 3)
+
+    def test_insertset_model(self):
+        self.assertEqual(InsertSet.objects.all().count(), 1)
+
+    def test_playerset_model(self):
+        self.assertEqual(Player.objects.all().count(), 36)
+
+    def test_card_model(self):
+        self.assertEqual(Card.objects.all().count(), 36)
+
+if __name__ == '__main__':
+    unittest.main()