def setUp(self):
self.service_environment = ServiceEnvironmentFactory()
self.unknown_service_environment = ServiceEnvironmentFactory(
service__name=UNKNOWN_SERVICE_ENVIRONMENT[0],
environment__name=UNKNOWN_SERVICE_ENVIRONMENT[0],
)
self.asset_info = AssetInfoFactory()
self.today = datetime.date(2014, 7, 1)
self.model = PricingObjectModelFactory(
type_id=PRICING_OBJECT_TYPES.VIP,
name=VIP_TYPES[0],
)
self.load_balancer = AssetInfoFactory()
def test_get_daily_asset_info(self):
self.assertEqual(
asset.get_daily_asset_info(
AssetInfoFactory.create(),
self.date,
self.data,
),
DailyAssetInfo.objects.all()[:1].get())
def test_get_daily_asset_info(self):
self.assertEqual(
asset.get_daily_asset_info(
AssetInfoFactory.create(),
self.date,
self.data,
),
DailyAssetInfo.objects.all()[:1].get()
)
def test_virtual(self):
service_environment = ServiceEnvironmentFactory.create()
asset_info = AssetInfoFactory.create(device_id=1)
DailyAssetInfoFactory.create(asset_info=asset_info, date=self.today)
virtual.ServiceEnvironment.objects.get = (
lambda *args, **kwargs: service_environment)
self.assertEqual(
virtual.virtual(today=self.today),
(True, 'Virtual: None new, 1 updated, 1 total'),
)
def test_update_when_asset_info_does_not_exist_error(self):
asset_info = AssetInfoFactory.build()
self.data['mount_device_id'] = asset_info.device_id
self.assertRaises(
share.AssetInfoDoesNotExistError,
share.update,
usage_type=self.usage_type,
data=self.data,
date=self.date,
)
def test_update_when_asset_info_does_not_exist_error(self):
asset_info = AssetInfoFactory.build()
self.data['mount_device_id'] = asset_info.device_id
self.assertRaises(
share.AssetInfoDoesNotExistError,
share.update,
usage_type=self.usage_type,
data=self.data,
date=self.date,
)
def test_virtual(self):
service_environment = ServiceEnvironmentFactory.create()
asset_info = AssetInfoFactory.create(device_id=1)
DailyAssetInfoFactory.create(asset_info=asset_info, date=self.today)
virtual.ServiceEnvironment.objects.get = (
lambda *args, **kwargs: service_environment
)
self.assertEqual(
virtual.virtual(today=self.today),
(True, 'Virtual: None new, 1 updated, 1 total'),
)
def setUp(self):
self.date = date.today()
self.data = {
'size': 100,
'mount_device_id': 1,
'storage_device_id': 1,
'label': 'test share',
}
self.asset_info = AssetInfoFactory.create(
device_id=self.data['mount_device_id'])
self.usage_type = UsageTypeFactory.create()
def setUp(self):
self.date = date.today()
self.data = {
'size': 100,
'mount_device_id': 1,
'storage_device_id': 1,
'label': 'test share',
}
self.asset_info = AssetInfoFactory.create(
device_id=self.data['mount_device_id']
)
self.usage_type = UsageTypeFactory.create()
def test_get_asset_and_pricing_object_when_asset_info_exist(self):
AssetInfoFactory(
asset_id=self.data['asset_id'],
warehouse=self.warehouse,
)
self.assertEqual(
asset.get_asset_info(
self.service_environment,
self.warehouse,
self.data,
), (
AssetInfo.objects.all()[:1].get(),
False,
))
def _create_asset_info(self):
self.ai = AssetInfoFactory(
service_environment=self.se1,
barcode='123',
sn='987654321',
)
self.dai1 = DailyAssetInfoFactory(
service_environment=self.se1,
pricing_object=self.ai,
asset_info=self.ai,
date=self.today,
)
self.dai2 = DailyAssetInfoFactory(
service_environment=self.se2,
pricing_object=self.ai,
asset_info=self.ai,
date=self.yesterday,
)
def test_get_asset_info_integrity_error(self):
AssetInfoFactory(
asset_id=self.data['asset_id'],
warehouse=self.warehouse,
)
data = [self.data.copy(), self.data.copy()]
data[1]['asset_id'] = 2
data[1]['barcode'] = 'Barcode2'
for d in data:
asset.get_asset_info(self.service_environment, self.warehouse, d)
self.assertEqual(AssetInfo.objects.count(), 2)
asset1 = AssetInfo.objects.get(asset_id=1)
asset2 = AssetInfo.objects.get(asset_id=2)
self.assertEqual(asset1.barcode, data[0]['barcode'])
self.assertEqual(asset2.barcode, data[1]['barcode'])
self.assertEqual(asset2.sn, data[1]['sn'])
self.assertEqual(asset2.device_id, data[1]['device_id'])
self.assertIsNone(asset1.sn)
self.assertIsNone(asset1.device_id)
def test_virtual_when_daily_asset_info_does_not_exist(self):
AssetInfoFactory.create(device_id=1)
self.assertEqual(
virtual.virtual(today=self.today),
(True, 'Virtual: None new, 0 updated, 1 total'),
)
def test_virtual_when_daily_asset_info_does_not_exist(self):
AssetInfoFactory.create(device_id=1)
self.assertEqual(
virtual.virtual(today=self.today),
(True, 'Virtual: None new, 0 updated, 1 total'),
)