# from django.test import TestCase
#
# from django.core.management import call_command
#
#
# #from .models import VirtualMachineType
#
#
# class VirtualMachineTypeModelTest(TestCase):
#
#     def setUp(self):
#         self.HETZNER_NUG_NAME = 'hetzner_nug'
#         self.HETZNER_NAME = 'hetzner'
#         self.HETZNER_RAID6_NAME = 'hetzner_raid6'
#         self.HETZNER_GLUSTERFS_NAME = 'hetzner_glusterfs'
#         self.BERN_NAME = 'bern'
#         self.HETZNER_NUG_EXPECTED_PRICE = 79
#         self.HETZNER_EXPECTED_PRICE = 180
#         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):
#
#         # hetzner_nug
#         # specifications = {
#         #     'cores': 2,
#         #     'memory': 10,
#         #     'disk_size': 100
#         # }
#         # vm_type = VirtualMachineType.objects.get(hosting_company=self.HETZNER_NUG_NAME)
#         # calculated_price = vm_type.calculate_price(specifications)
#         # self.assertEqual(calculated_price, self.HETZNER_NUG_EXPECTED_PRICE)
#
#         # hetzner
#         specifications = {
#             'cores': 2,
#             'memory': 10,
#             'disk_size': 100
#         }
#         vm_type = VirtualMachineType.objects.get(hosting_company=self.HETZNER_NAME)
#         calculated_price = vm_type.calculate_price(specifications)
#         self.assertEqual(calculated_price, self.HETZNER_EXPECTED_PRICE)
#
#         # hetzner_raid6
#         # specifications = {
#         #     'cores': 2,
#         #     'memory': 10,
#         #     'disk_size': 100
#         # }
#         # vm_type = VirtualMachineType.objects.get(hosting_company=self.HETZNER_RAID6_NAME)
#         # calculated_price = vm_type.calculate_price(specifications)
#         # self.assertEqual(calculated_price, self.HETZNER_RAID6_EXPECTED_PRICE)
#
#         # hetzner_glusterfs
#         # specifications = {
#         #     'cores': 2,
#         #     'memory': 10,
#         #     'disk_size': 100
#         # }
#         # vm_type = VirtualMachineType.objects.get(hosting_company=self.HETZNER_GLUSTERFS_NAME)
#         # calculated_price = vm_type.calculate_price(specifications)
#         # self.assertEqual(calculated_price, self.HETZNER_GLUSTERFS_EXPECTED_PRICE)
#
#         # bern
#         specifications = {
#             'cores': 2,
#             'memory': 10,
#             'disk_size': 100
#         }
#         vm_type = VirtualMachineType.objects.get(hosting_company=self.BERN_NAME)
#         calculated_price = vm_type.calculate_price(specifications)
#         self.assertEqual(calculated_price, self.BERN_EXPECTED_PRICE)