class AssetDevInfoPostSaveTest(TestCase):
def setUp(self):
# core side
self.dc_model = DeviceModelFactory(
name="DC", type=DeviceType.data_center,
)
self.rack_model = DeviceModelFactory(
name="Rack", type=DeviceType.rack,
)
self.dc_1 = DeviceFactory(name='DC1', sn='DC1', model=self.dc_model)
self.dc_2 = DeviceFactory(name='DC2', sn='DC2', model=self.dc_model)
self.rack_1_1 = DeviceFactory(
name='Rack 1 DC1', sn='Rack 1 DC1', model=self.rack_model,
parent=self.dc_1,
)
self.rack_1_2 = DeviceFactory(
name='Rack 2 DC1', sn='Rack 2 DC1', model=self.rack_model,
parent=self.dc_1,
)
self.rack_2_1 = DeviceFactory(
name='Rack 1 DC2', sn='Rack 1 DC2', model=self.rack_model,
parent=self.dc_1,
)
self.rack_2_2 = DeviceFactory(
name='Rack 2 DC2', sn='Rack 2 DC2', model=self.rack_model,
parent=self.dc_2,
)
self.dev_1 = DeviceFactory(name="h101.dc1", parent=self.rack_1_1)
self.dev_2 = DeviceFactory(name="h201.dc1", parent=self.rack_1_2)
self.dev_3 = DeviceFactory(name="h101.dc2", parent=self.rack_2_1)
self.dev_4 = DeviceFactory(name="h201.dc2", parent=self.rack_2_2)
self.dev_5 = DeviceFactory(name="h201-1.dc2", parent=self.dev_4)
# assets side
self.assets_dc_1 = DataCenterFactory(
name='DC1', deprecated_ralph_dc_id=self.dc_1.id,
)
self.assets_dc_2 = DataCenterFactory(
name='DC2', deprecated_ralph_dc_id=self.dc_2.id,
)
self.assets_sr_1 = ServerRoomFactory(
name="DC1_1", data_center=self.assets_dc_1,
)
self.assets_sr_2 = ServerRoomFactory(
name="DC2_1", data_center=self.assets_dc_2,
)
self.assets_rack_1_1 = RackFactory(
name="Rack 1 DC1", deprecated_ralph_rack_id=self.rack_1_1.id,
data_center=self.assets_dc_1,
)
self.assets_rack_1_2 = RackFactory(
name="Rack 2 DC1", deprecated_ralph_rack_id=self.rack_1_2.id,
data_center=self.assets_dc_1,
)
self.assets_rack_2_1 = RackFactory(
name="Rack 1 DC2", deprecated_ralph_rack_id=self.rack_2_1.id,
data_center=self.assets_dc_2,
)
self.assets_rack_2_2 = RackFactory(
name="Rack 2 DC2", deprecated_ralph_rack_id=self.rack_2_2.id,
data_center=self.assets_dc_2,
)
self.assets_dev_1 = DCAssetFactory(
device_info=DeviceInfoFactory(
ralph_device_id=self.dev_1.id, data_center=self.assets_dc_1,
server_room=self.assets_sr_1, rack=self.assets_rack_1_1,
),
)
self.assets_dev_2 = DCAssetFactory(
device_info=DeviceInfoFactory(
ralph_device_id=self.dev_2.id, data_center=self.assets_dc_1,
server_room=self.assets_sr_1, rack=self.assets_rack_1_2,
)
)
self.assets_dev_3 = DCAssetFactory(
device_info=DeviceInfoFactory(
ralph_device_id=self.dev_3.id, data_center=self.assets_dc_2,
server_room=self.assets_sr_2, rack=self.assets_rack_2_1,
)
)
self.assets_dev_4 = DCAssetFactory(
device_info=DeviceInfoFactory(
ralph_device_id=self.dev_4.id, data_center=self.assets_dc_2,
server_room=self.assets_sr_2, rack=self.assets_rack_2_2,
position=10, orientation=Orientation.front,
)
)
self.assets_dev_5 = DCAssetFactory(
device_info=DeviceInfoFactory(
ralph_device_id=self.dev_5.id, data_center=self.assets_dc_2,
server_room=self.assets_sr_2, rack=self.assets_rack_2_2,
position=10, orientation=Orientation.front, slot_no=1,
)
)
def test_update_cached_localization(self):
_update_cached_localization(
device=self.dev_1, asset_dev_info=self.assets_dev_1.device_info,
)
self.assertEqual(self.dev_1.rack, 'Rack 1 DC1')
self.assertEqual(self.dev_1.dc, 'DC1')
_update_cached_localization(
device=self.dev_3, asset_dev_info=self.assets_dev_3.device_info,
)
self.assertEqual(self.dev_3.rack, 'Rack 1 DC2')
self.assertEqual(self.dev_3.dc, 'DC2')
def test_update_localization(self):
# case: device_info without deprecated_ralph_rack
rack = RackFactory(
name="Rack 4 DC2", data_center=self.assets_dc_2,
)
old_device_info = self.assets_dev_2.device_info
self.assets_dev_2.device_info = None
self.assets_dev_2.save()
old_device_info.delete()
device_info = DeviceInfoFactory(
ralph_device_id=self.dev_2.id, data_center=self.assets_dc_2,
server_room=self.assets_sr_2, rack=rack,
)
self.assets_dev_2.device_info = device_info
self.assets_dev_2.save()
_update_localization(device=self.dev_2, asset_dev_info=device_info)
self.assertEqual(self.dev_2.parent_id, self.rack_1_2.id)
# case: rack and dc changed
rack = RackFactory(
name="Rack 5 DC2", data_center=self.assets_dc_2,
deprecated_ralph_rack_id=self.rack_2_2.id
)
old_device_info = self.assets_dev_2.device_info
self.assets_dev_2.device_info = None
self.assets_dev_2.save()
old_device_info.delete()
device_info = DeviceInfoFactory(
ralph_device_id=self.dev_2.id, data_center=self.assets_dc_2,
server_room=self.assets_sr_2, rack=rack,
)
self.assets_dev_2.device_info = device_info
self.assets_dev_2.save()
_update_localization(device=self.dev_2, asset_dev_info=device_info)
self.assertEqual(self.dev_2.parent_id, self.rack_2_2.id)
self.assertEqual(self.dev_2.parent.parent_id, self.dc_2.id)
def test_update_localization_details(self):
_update_localization_details(
device=self.dev_5, asset_dev_info=self.assets_dev_5.device_info,
)
self.assertEqual(self.dev_5.chassis_position, 10)
self.assertEqual(self.dev_5.position, 1)
def test_get_core_parent(self):
# normal device
parent, is_blade_system = _get_core_parent(
self.assets_dev_4.device_info,
)
self.assertEqual(parent.id, self.rack_2_2.id)
self.assertFalse(is_blade_system)
# blade server
self.assets_dev_5.model.category.is_blade = True
self.assets_dev_5.model.category.save()
parent, is_blade_system = _get_core_parent(
self.assets_dev_5.device_info,
)
self.assertEqual(parent.id, self.dev_4.id)
self.assertTrue(is_blade_system)
# asset without rack
self.assets_dev_2.device_info.rack = None
self.assets_dev_2.device_info.save()
self.assertEqual(
_get_core_parent(self.assets_dev_2.device_info),
(None, False),
)
@patch('ralph_assets.models_signals._update_localization_details')
@patch('ralph_assets.models_signals._update_localization')
@patch('ralph_assets.models_signals._update_cached_localization')
def test_update_core_localization(
self, mock_update_cached_localization, mock_update_localization,
mock_update_localization_details,
):
# case: ralph device doesn't exist
self.dev_1.delete()
update_core_localization(
asset_dev_info=self.assets_dev_1.device_info,
)
self.assertFalse(mock_update_cached_localization.called)
self.assertFalse(mock_update_localization.called)
self.assertFalse(mock_update_localization_details.called)
# case: ralph device exists
update_core_localization(
asset_dev_info=self.assets_dev_2.device_info,
)
self.assertTrue(mock_update_cached_localization.called)
self.assertTrue(mock_update_localization.called)
self.assertTrue(mock_update_localization_details.called)
@patch('ralph_assets.models_signals.update_core_localization')
def test_asset_device_info_post_save(self, mock):
asset_device_info_post_save(
sender=DeviceInfo, instance=self.assets_dev_2.device_info,
)
mock.assert_called_with(asset_dev_info=self.assets_dev_2.device_info)
class TestPricingAssetModelsCommand(TestCase):
def setUp(self):
self.asset = DCAssetFactory()
self.ralph_device = self.asset.get_ralph_device()
def test_get_device_name_from_asset_when_device_info_is_null(self):
self.asset.device_info = None
self.assertEqual(
Command().get_device_name_from_asset(self.asset, {1: 'test0'}),
None,
)
def test_get_device_name_from_asset_when_ralph_id_is_null(self):
self.asset.device_info.ralph_device_id = None
self.assertEqual(
Command().get_device_name_from_asset(self.asset, {1: 'test0'}),
None,
)
def test_get_device_name_from_asset(self):
self.assertEqual(
Command().get_device_name_from_asset(
self.asset, {self.asset.device_info.ralph_device_id: 'test0'}),
'test0',
)
def test_get_device_ids_when_device_info_is_null(self):
self.asset.device_info = None
self.assertEqual(Command()._get_device_ids([self.asset]), [])
def test_get_device_ids(self):
self.assertEqual(Command()._get_device_ids([self.asset]),
[self.ralph_device.id])
def test_get_ralph_devices(self):
self.assertEqual(
repr(Command()._get_ralph_devices([self.asset])),
repr([self.ralph_device]),
)
def test_get_device_ids_and_names(self):
self.assertEqual(Command().get_device_ids_and_names([self.asset]),
{self.ralph_device.id: self.ralph_device.name})
def test_get_deprecated_date_when_force_deprecation(self):
self.asset.force_deprecation = True
self.assertEqual(Command().get_deprecated_date(self.asset),
'Deprecated')
def test_get_deprecated_date_when_no_invoice_date(self):
self.asset.invoice_date = None
self.asset.save()
self.assertEqual(Command().get_deprecated_date(self.asset),
'No invoice date')
def test_get_deprecated_date_when_deprecation_end_date(self):
self.asset.invoice_date = datetime.date.today()
self.asset.deprecation_rate = 25
self.asset.deprecation_end_date = datetime.date.today()
self.asset.save()
self.assertEqual(Command().get_deprecated_date(self.asset),
datetime.date.today())
def test_get_deprecated_date(self):
self.asset.invoice_date = datetime.date.today()
self.asset.deprecation_end_date = None
self.asset.deprecation_rate = 25
self.asset.save()
self.assertEqual(Command().get_deprecated_date(self.asset),
datetime.date.today() + relativedelta(years=4))
def test_get_data(self):
self.assertEqual(Command().get_data(), [[
self.asset.id,
self.asset.sn,
self.asset.barcode,
self.asset.venture,
self.asset.get_ralph_device().name,
self.asset.deprecation_end_date,
]])
def test_dc_asset_without_linked_device(self):
asset = DCAssetFactory(device_info=None)
asset.save()
self.assertEqual(asset.linked_device, None)
def test_dc_asset_without_linked_device(self):
asset = DCAssetFactory(device_info=None)
asset.save()
self.assertEqual(asset.linked_device, None)
class TestPricingAssetModelsCommand(TestCase):
def setUp(self):
self.asset = DCAssetFactory()
self.ralph_device = self.asset.get_ralph_device()
def test_get_device_name_from_asset_when_device_info_is_null(self):
self.asset.device_info = None
self.assertEqual(
Command().get_device_name_from_asset(
self.asset,
{1: 'test0'}
),
None,
)
def test_get_device_name_from_asset_when_ralph_id_is_null(self):
self.asset.device_info.ralph_device_id = None
self.assertEqual(
Command().get_device_name_from_asset(
self.asset,
{1: 'test0'}
),
None,
)
def test_get_device_name_from_asset(self):
self.assertEqual(
Command().get_device_name_from_asset(
self.asset,
{self.asset.device_info.ralph_device_id: 'test0'}
),
'test0',
)
def test_get_device_ids_when_device_info_is_null(self):
self.asset.device_info = None
self.assertEqual(Command()._get_device_ids([self.asset]), [])
def test_get_device_ids(self):
self.assertEqual(
Command()._get_device_ids([self.asset]),
[self.ralph_device.id]
)
def test_get_ralph_devices(self):
self.assertEqual(
repr(Command()._get_ralph_devices([self.asset])),
repr([self.ralph_device]),
)
def test_get_device_ids_and_names(self):
self.assertEqual(
Command().get_device_ids_and_names([self.asset]),
{self.ralph_device.id: self.ralph_device.name}
)
def test_get_deprecated_date_when_force_deprecation(self):
self.asset.force_deprecation = True
self.assertEqual(
Command().get_deprecated_date(self.asset),
'Deprecated'
)
def test_get_deprecated_date_when_no_invoice_date(self):
self.asset.invoice_date = None
self.asset.save()
self.assertEqual(
Command().get_deprecated_date(self.asset),
'No invoice date'
)
def test_get_deprecated_date_when_deprecation_end_date(self):
self.asset.invoice_date = datetime.date.today()
self.asset.deprecation_rate = 25
self.asset.deprecation_end_date = datetime.date.today()
self.asset.save()
self.assertEqual(
Command().get_deprecated_date(self.asset),
datetime.date.today()
)
def test_get_deprecated_date(self):
self.asset.invoice_date = datetime.date.today()
self.asset.deprecation_end_date = None
self.asset.deprecation_rate = 25
self.asset.save()
self.assertEqual(
Command().get_deprecated_date(self.asset),
datetime.date.today() + relativedelta(years=4)
)
def test_get_data(self):
self.assertEqual(
Command().get_data(),
[[
self.asset.id,
self.asset.sn,
self.asset.barcode,
self.asset.venture,
self.asset.get_ralph_device().name,
self.asset.deprecation_end_date,
]]
)