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_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'), )