def setUp(self):
self.client = login_as_su()
self.deprecation_kind = DeprecationKind(months=24, remarks='Default')
self.deprecation_kind.save()
self.margin = MarginKind(margin=100, remarks='100%')
self.margin.save()
self.venture = Venture(name='VenureName')
self.venture.save()
self.role = VentureRole(venture=self.venture, name='VentureRole')
self.role.save()
self.device = Device.create(
sn=DEVICE['sn'],
barcode=DEVICE['barcode'],
remarks=DEVICE['remarks'],
model_name=DEVICE['model_name'],
model_type=DeviceType.unknown,
rack=DEVICE['rack'],
position=DEVICE['position'],
dc=DATACENTER,
)
self.device.name = DEVICE['name']
self.device.save()
self.device2 = Device.create(
sn=DEVICE2['sn'],
barcode=DEVICE2['barcode'],
remarks=DEVICE2['remarks'],
model_name=DEVICE2['model_name'],
model_type=DeviceType.unknown,
rack=DEVICE2['rack'],
position=DEVICE2['position'],
dc=DATACENTER,
)
self.device2.name = DEVICE2['name']
self.device2.save()
class BulkeditTest(TestCase):
""" Tests edit form
Scenario:
1. Changes data in single device + test history change
2. Changes data in more than one device
3. Send form without select edit fields
4. Send form with empty save comment field
5. If data was added, check depreciation_date
"""
def setUp(self):
self.client = login_as_su()
self.deprecation_kind = DeprecationKind(months=24, remarks='Default')
self.deprecation_kind.save()
self.margin = MarginKind(margin=100, remarks='100%')
self.margin.save()
self.venture = Venture(name='VenureName')
self.venture.save()
self.role = VentureRole(venture=self.venture, name='VentureRole')
self.role.save()
self.device = Device.create(
sn=DEVICE['sn'],
barcode=DEVICE['barcode'],
remarks=DEVICE['remarks'],
model_name=DEVICE['model_name'],
model_type=DeviceType.unknown,
rack=DEVICE['rack'],
position=DEVICE['position'],
dc=DATACENTER,
)
self.device.name = DEVICE['name']
self.device.save()
self.device2 = Device.create(
sn=DEVICE2['sn'],
barcode=DEVICE2['barcode'],
remarks=DEVICE2['remarks'],
model_name=DEVICE2['model_name'],
model_type=DeviceType.unknown,
rack=DEVICE2['rack'],
position=DEVICE2['position'],
dc=DATACENTER,
)
self.device2.name = DEVICE2['name']
self.device2.save()
def test_single_device_edit(self):
url = '/ui/search/bulkedit/'
text_fields = ('position', 'chassis_position', 'remarks', 'verified')
device_fields = []
device_fields.extend(text_fields)
post_data = {
'select': [self.device.id], # 1
'edit': device_fields,
'verified': True,
'position': '9',
'chassis_position': 10,
'remarks': 'Hello Ralph',
'save_comment': 'Everything has changed',
'save': '', # save form
}
self.client.post(url, post_data)
# Check if data from form is the same that data in database
device = Device.objects.get(id=self.device.id)
for field in device_fields:
db_data = getattr(device, field)
form_data = post_data[field]
msg = 'FIELD: %s, DB: %s FORM: %s' % (field, db_data, form_data)
self.assertEqual(unicode(db_data), unicode(form_data), msg)
# Check if change can see in History change
history_device = HistoryChange.objects.filter(
device=self.device,
old_value='Very important device',
new_value='Hello Ralph',
)
self.assertEqual(history_device.count(), 1)
self.assertEqual(
history_device[0].comment,
'Everything has changed'
)
def test_many_devices_edit(self):
url = '/ui/search/bulkedit/'
remarks = 'change remarks in 2 devices'
post_data = {
'select': [self.device.id, self.device2.id],
'edit': ['remarks'],
'remarks': remarks,
'save_comment': 'change remarks',
'save': '', # save form
}
self.client.post(url, post_data)
device, device2 = Device.objects.filter(remarks=remarks)
self.assertEqual = (device.remarks, remarks)
self.assertEqual = (device.remarks, device2.remarks)
def test_send_form_without_marking_edited_fields(self):
url = '/ui/search/bulkedit/'
data_post = {
'select': self.device.id,
'remarks': 'Here are some remarks without confirmation.',
'save': '',
}
response = self.client.post(url, data_post)
self.assertEqual(response.status_code, 200)
self.assertTrue(ERROR_MSG['no_mark_fields'] in response.content)
def test_send_form_with_empty_save_comment_field(self):
url = '/ui/search/bulkedit/?'
post_data = {
'select': [self.device.id],
'edit': ['remarks'],
'remarks': 'Hello World!',
'save_comment': '',
'save': '', # save form
}
response = self.client.post(url, post_data)
self.assertEqual(response.status_code, 200)
self.assertTrue(ERROR_MSG['empty_save_comment'] in response.content)
self.assertFormError(
response,
'form',
'save_comment',
ERROR_MSG['empty_save_comment_field']
)
class BulkeditTest(TestCase):
""" Tests edit form
Scenario:
1. Changes data in single device + test history change
2. Changes data in more than one device
3. Send form without select edit fields
4. Send form with empty save comment field
5. If data was added, check depreciation_date
"""
def setUp(self):
self.client = login_as_su()
self.deprecation_kind = DeprecationKind(months=24, remarks='Default')
self.deprecation_kind.save()
self.margin = MarginKind(margin=100, remarks='100%')
self.margin.save()
self.venture = Venture(name='VenureName')
self.venture.save()
self.role = VentureRole(venture=self.venture, name='VentureRole')
self.role.save()
self.device = Device.create(
sn=DEVICE['sn'],
barcode=DEVICE['barcode'],
remarks=DEVICE['remarks'],
model_name=DEVICE['model_name'],
model_type=DeviceType.unknown,
rack=DEVICE['rack'],
position=DEVICE['position'],
dc=DATACENTER,
)
self.device.name = DEVICE['name']
self.device.save()
self.device2 = Device.create(
sn=DEVICE2['sn'],
barcode=DEVICE2['barcode'],
remarks=DEVICE2['remarks'],
model_name=DEVICE2['model_name'],
model_type=DeviceType.unknown,
rack=DEVICE2['rack'],
position=DEVICE2['position'],
dc=DATACENTER,
)
self.device2.name = DEVICE2['name']
self.device2.save()
def test_single_device_edit(self):
url = '/ui/search/bulkedit/'
select_fields = (
'venture', 'venture_role', 'margin_kind', 'deprecation_kind',
)
date_fields = (
'purchase_date', 'warranty_expiration_date',
'support_expiration_date', 'support_kind',
)
text_fields = (
'barcode', 'position', 'chassis_position', 'remarks', 'price',
'sn', 'verified'
)
device_fields = []
for field_list in [select_fields, date_fields, text_fields]:
device_fields.extend(field_list)
post_data = {
'select': [self.device.id], # 1
'edit': device_fields,
'venture': self.venture.id, # 1
'venture_role': self.role.id, # 1
'verified': True,
'barcode': 'bc-2222-2222-2222-2222',
'position': '9',
'chassis_position': 10,
'remarks': 'Hello Ralph',
'margin_kind': self.margin.id, # 1
'deprecation_kind': self.deprecation_kind.id, # 1
'price': 100,
'sn': '2222-2222-2222-2222',
'purchase_date': datetime(2001, 1, 1, 0, 0),
'warranty_expiration_date': datetime(2001, 1, 2, 0, 0),
'support_expiration_date': datetime(2001, 1, 3, 0, 0),
'support_kind': datetime(2001, 1, 4, 0, 0),
'save_comment': 'Everything has changed',
'save': '', # save form
}
response = self.client.post(url, post_data)
# Check if data from form is the same that data in database
device = Device.objects.get(id=self.device.id)
for field in device_fields:
db_data = getattr(device, field)
form_data = post_data[field]
msg = 'FIELD: %s, DB: %s FORM: %s' % (field, db_data, form_data)
if field in select_fields:
self.assertEqual(db_data.id, form_data, msg)
else:
self.assertEqual(unicode(db_data), unicode(form_data), msg)
# Check if change can see in History change
history_device = HistoryChange.objects.filter(
device=self.device,
old_value='Very important device',
new_value='Hello Ralph',
)
self.assertEqual(history_device.count(), 1)
self.assertEqual(
history_device[0].comment,
'Everything has changed'
)
def test_many_devices_edit(self):
url = '/ui/search/bulkedit/'
remarks = 'change remakrs in 2 devices'
post_data = {
'select': [self.device.id, self.device2.id],
'edit': ['remarks'],
'remarks': remarks,
'save_comment': 'change remakrs',
'save': '', # save form
}
response = self.client.post(url, post_data)
device, device2 = Device.objects.filter(remarks=remarks)
self.assertEqual = (device.remarks, remarks)
self.assertEqual = (device.remarks, device2.remarks)
def test_send_form_without_selected_edit_fields(self):
url = '/ui/search/bulkedit/'
data_post = {
'select': self.device.id,
'bulk': '', # show form
}
response = self.client.post(url, data_post)
self.assertEqual(response.status_code, 200) # form false
response = self.client.post(url, {'select': self.device.id})
self.assertTrue(ERROR_MSG['no_mark_fields'] in response.content)
def test_send_form_with_empty_save_comment_field(self):
url = '/ui/search/bulkedit/?'
post_data = {
'select': [self.device.id],
'edit': ['remarks'],
'remarks': 'Hello World!',
'save_comment': '',
'save': '', # save form
}
response = self.client.post(url, post_data)
self.assertEqual(response.status_code, 200) # form false
self.assertTrue(ERROR_MSG['empty_save_comment'] in response.content)
self.assertFormError(
response,
'form',
'save_comment',
ERROR_MSG['empty_save_comment_field']
)
def test_calculation_depreciation_date(self):
device = Device.objects.get(id=self.device.id)
self.assertEqual(device.purchase_date, None)
self.assertEqual(device.deprecation_kind, None)
url = '/ui/search/bulkedit/?'
post_data = {
'select': [self.device.id], # 1
'edit': ['purchase_date', 'deprecation_kind'],
'purchase_date': datetime(2001, 1, 4, 0, 0),
'deprecation_kind': self.deprecation_kind.id, # 1
'save_comment': 'Updated: purchase date and deprecation kind',
'save': '', # save form
}
response = self.client.post(url, post_data)
updated_device = Device.objects.get(id=self.device.id)
self.assertEqual(
updated_device.purchase_date, datetime(2001, 1, 4, 0, 0)
)
self.assertEqual(updated_device.deprecation_kind.months, 24)
self.assertEqual(
updated_device.deprecation_date, datetime(2003, 1, 4, 0, 0)
)
# Check if purchase date,change deprecation date
post_data = {
'select': [self.device.id], # 1
'edit': ['purchase_date'],
'purchase_date': datetime(2002, 1, 1, 0, 0),
'save_comment': 'Updated: purchase date',
'save': '', # save form
}
response = self.client.post(url, post_data, follow=True)
updated_device2 = Device.objects.get(id=self.device.id)
self.assertEqual(
updated_device2.deprecation_date, datetime(2004, 1, 1, 0, 0)
)
