def test_model_asset_type_back_office_shall_pass(self): back_office_model = DataCenterAssetModelFactory( type=ObjectModelType.from_name('back_office')) self.data.update({'model': back_office_model.pk}) response = self.client.post(self.asset.get_absolute_url(), self.data) self.asset.refresh_from_db() self.assertEqual(response.status_code, 302) self.assertEqual(self.asset.model, back_office_model)
def test_model_asset_type_data_center_asset_shall_not_pass(self): back_office_model = DataCenterAssetModelFactory( type=ObjectModelType.from_name('data_center')) self.data.update({'model': back_office_model.pk}) response = self.client.post(self.asset.get_absolute_url(), self.data) self.asset.refresh_from_db() self.assertIn('Model must be of', response.content.decode('utf-8')) self.assertNotEqual(self.asset.model, back_office_model) self.assertEqual(response.status_code, 200)
def test_model_asset_type_data_center_shall_pass(self): data_center_model = DataCenterAssetModelFactory( type=ObjectModelType.from_name('data_center') ) data = self._get_initial_data() data.update({ 'model': data_center_model.pk }) response = self.client.post( self.dca.get_absolute_url(), data ) self.dca.refresh_from_db() self.assertEqual(response.status_code, 302) self.assertEqual(self.dca.model, data_center_model)