def setUp(self):
self.user = UserFactory()
self.user_pl = UserFactory()
self.user_pl.profile.country = Country.pl
self.user_pl.profile.save()
self.cat = AssetCategoryFactory()
self.cat1 = AssetCategoryFactory()
self.cat2 = AssetCategoryFactory()
self.cat3 = AssetCategoryFactory()
self.asset1 = BOAssetFactory()
self.asset2 = BOAssetFactory()
def setUp(self):
self.client = login_as_su()
self.category = AssetCategoryFactory()
self.asset = AssetFactory()
self.asset1 = AssetFactory()
self.model = AssetModelFactory(category=self.category)
self.model1 = AssetModelFactory(category=self.category)
self.user = UserFactory()
self.warehouse = WarehouseFactory()
self.assetOwner = AssetOwnerFactory()
self.asset_service = ServiceFactory()
self.common_asset_data = { # DC & BO common data
'barcode': 'barcode',
'deprecation_rate': '25',
'invoice_date': '2011-11-14',
'invoice_no': 'invoice_no',
'model': self.model,
'order_no': 'order_no',
'owner': self.user,
'price': '100',
'property_of': self.assetOwner,
'service_name': self.asset_service,
'source': models_assets.AssetSource.shipment,
'status': models_assets.AssetStatus.in_progress,
'task_url': 'www.test.com',
'user': self.user,
'warehouse': self.warehouse,
}
def test_get_asset_with_empty_asset_source(self):
"""Getting an asset with empty 'source' field should also succeed."""
category = AssetCategoryFactory()
model = AssetModelFactory(category=category)
asset = DCAssetFactory(model=model, source=None)
asset_data = get_asset(asset.device_info.ralph_device_id)
self.assertEqual(asset_data['source'], None)
def test_generate_next_hostname_out_of_range(self):
category = AssetCategoryFactory(code='PC')
model = AssetModelFactory(category=category)
asset = BOAssetFactory(model=model, owner=self.user_pl, hostname='')
models_assets.AssetLastHostname.objects.create(prefix='POLPC',
counter=99999)
template_vars = {
'code': asset.model.category.code,
'country_code': asset.country_code,
}
asset.generate_hostname(template_vars=template_vars)
self.assertEqual(asset.hostname, 'POLPC100000')
def test_generate_next_hostname(self):
category = AssetCategoryFactory(code='PC')
model = AssetModelFactory(category=category)
asset = BOAssetFactory(model=model, owner=self.user_pl, hostname='')
BOAssetFactory(owner=self.user_pl, hostname='POLSW00003')
models_assets.AssetLastHostname.increment_hostname(prefix='POLPC')
models_assets.AssetLastHostname.increment_hostname(prefix='POLPC')
template_vars = {
'code': asset.model.category.code,
'country_code': asset.country_code,
}
asset.generate_hostname(template_vars=template_vars)
self.assertEqual(asset.hostname, 'POLPC00003')
def test_generate_first_hostname(self):
"""Scenario:
- none of assets has hostname
- after generate first of asset have XXXYY00001 in hostname field
"""
category = AssetCategoryFactory(code='PC')
model = AssetModelFactory(category=category)
asset = BOAssetFactory(model=model, owner=self.user_pl, hostname='')
template_vars = {
'code': asset.model.category.code,
'country_code': asset.country_code,
}
asset.generate_hostname(template_vars=template_vars)
self.assertEqual(asset.hostname, 'POLPC00001')
def test_get_asset_by_sn_or_barcode(self):
category = AssetCategoryFactory()
model = AssetModelFactory(category=category)
asset = DCAssetFactory(model=model, )
# by sn
asset_data = get_asset_by_sn_or_barcode(asset.sn)
self.assertEqual(asset_data['sn'], asset.sn)
self.assertEqual(asset_data['barcode'], asset.barcode)
# by barcode
asset_data = get_asset_by_sn_or_barcode(asset.barcode)
self.assertEqual(asset_data['sn'], asset.sn)
self.assertEqual(asset_data['barcode'], asset.barcode)
# not exists
self.assertEqual(get_asset_by_sn_or_barcode('foo_ziew_123'), None)
def setUp(self):
self.client = login_as_su()
self.category = AssetCategoryFactory()
self.model = AssetModelFactory(category=self.category)
self.warehouse = WarehouseFactory()
self.first_asset = AssetFactory(model=self.model, )
self.second_asset = AssetFactory(model=self.model, )
self.asset_with_duplicated_sn = AssetFactory(model=self.model, )
# Prepare required fields (formset_name, field_name)
self.required_fields = [
('asset_form', 'model'),
('asset_form', 'warehouse'),
]
self.model1 = AssetModelFactory()
def test_get_asset(self):
"""Test get asset information by ralph_device_id."""
support1 = DCSupportFactory()
support2 = DCSupportFactory()
category = AssetCategoryFactory()
model = AssetModelFactory(category=category)
asset = DCAssetFactory(
model=model,
supports=[support1, support2],
)
asset_data = get_asset(asset.device_info.ralph_device_id)
self.assertEqual(asset_data['sn'], asset.sn)
self.assertEqual(asset_data['barcode'], asset.barcode)
self.assertEqual(asset_data['supports'][0]['name'], support1.name)
self.assertEqual(asset_data['supports'][0]['url'], support1.url)
self.assertEqual(asset_data['supports'][1]['name'], support2.name)
self.assertEqual(asset_data['supports'][1]['url'], support2.url)
self.assertEqual(
asset_data['required_support'],
asset.required_support,
)
def setUp(self):
self.client = login_as_su()
self.warehouse = WarehouseFactory()
self.category = AssetCategoryFactory()
self.model = AssetModelFactory(category=self.category)
self.addform = '/assets/dc/add/device/'
ci_relation = CIRelationFactory()
self.common_test_data = dict(
asset=True,
deprecation_rate=0,
device_environment=ci_relation.child.id,
invoice_date='2001-01-02',
model=self.model.id,
price='10',
production_year=2011,
ralph_device_id='',
service=ci_relation.parent.id,
size=1,
slots=1,
source=1,
status=AssetStatus.new.id,
type=AssetType.data_center.id,
warehouse=self.warehouse.id,
)
def test_can_generate_hostname_for_model_with_hostname(self):
category = AssetCategoryFactory(code='PC')
model = AssetModelFactory(category=category)
asset = BOAssetFactory(model=model, owner=self.user_pl)
self._check_hostname_is_generated(asset)