diff --git a/hosting/test_models.py b/hosting/test_models.py index 0e80ce31..9c9cd71c 100644 --- a/hosting/test_models.py +++ b/hosting/test_models.py @@ -1,6 +1,5 @@ from django.test import TestCase -from model_mommy import mommy from django.core.management import call_command @@ -20,6 +19,7 @@ class VirtualMachineTypeModelTest(TestCase): self.HETZNER_RAID6_EXPECTED_PRICE = 216 self.HETZNER_GLUSTERFS_EXPECTED_PRICE = 252 self.BERN_EXPECTED_PRICE = 202 + call_command('create_vm_types') def test_calculate_price(self): diff --git a/requirements.txt b/requirements.txt index f3f2836c..13ea5675 100644 --- a/requirements.txt +++ b/requirements.txt @@ -16,6 +16,7 @@ psycopg2 django-mptt easy_thumbnails django-polymorphic +model-mommy #PLUGINS djangocms_flash