def setUp(self):
# fake settings needed to run plugin, even with mocked library.
settings.OPENSTACK_URL = '/'
settings.OPENSTACK_USER = '******'
settings.OPENSTACK_PASSWORD = '******'
# model group for sample costs calculations.
self.cmg = ComponentModelGroup()
self.cmg.price = 100.0
self.cmg.per_size = 1
self.cmg.size_modifier = 1
self.cmg.save()
self.mk = MarginKind()
self.mk.margin = 200
self.mk.remarks = 'Margin kind 200%'
self.mk.save()
def test_cost(self):
dev = Device.create(sn='device',
model_type=DeviceType.rack_server,
model_name='device')
dmg = DeviceModelGroup(name='DeviceModelGroup')
dmg.price = 100
dmg.save()
dev.model.group = dmg
dev.model.save()
dev.margin_kind = MarginKind(name='50%', margin=50)
dev.margin_kind.save()
dev.deprecation_kind = DeprecationKind(name='10 months', months=10)
dev.deprecation_kind.save()
dev.save()
pricing.device_update_cached(dev)
self.assertEqual(dev.cached_cost, 15)
def test_price_deprecation(self):
# Device after deprecation period should have raw price = 0
dev = Device.create(sn='device',
model_type=DeviceType.rack_server,
model_name='device')
dmg = DeviceModelGroup(name='DeviceModelGroup')
dmg.price = 100
dmg.save()
dev.model.group = dmg
dev.purchase_date = datetime.today() - timedelta(11 * (365 / 12))
dev.model.save()
dev.margin_kind = MarginKind(name='50%', margin=50)
dev.margin_kind.save()
dev.deprecation_kind = DeprecationKind(name='10 months', months=10)
dev.deprecation_kind.save()
dev.save()
pricing.device_update_cached(dev)
self.assertEqual(get_device_raw_price(dev), 0)
self.assertEqual(dev.cached_price, 0)
self.assertEqual(dev.cached_cost, 0)
def test_price_deprecation_in_progress(self):
# Device in deprecation period should have raw price >0 if price is set
dev = Device.create(sn='device', model_type=DeviceType.rack_server,
model_name='device')
dmg = DeviceModelGroup(name='DeviceModelGroup')
dmg.price = 100
dmg.save()
dev.model.group = dmg
# currently first month in deprecation period
dev.purchase_date = datetime.today() - timedelta(1 * (365 / 12))
dev.model.save()
mk = MarginKind(name='50%', margin=50)
mk.save()
dev.margin_kind = mk
dk = DeprecationKind(name='10 months', months=10)
dk.save()
dev.deprecation_kind = dk
dev.save()
pricing.device_update_cached(dev)
dev = Device.objects.get(id=dev.id)
self.assertEqual(get_device_raw_price(dev), 100)
self.assertEqual(dev.cached_cost, 15)
self.assertEqual(dev.cached_price, 100)
