def test_assign_projects_from_ticket_filter(self):
project = ProjectFactory.create()
tickets = MaximoTicket.objects.all()[:5]
for ticket in tickets:
ticket.project = project
ticket.save()
TestExcel.create_time_registers(date(2015, 9, 1), date(2015, 9, 30))
updated = MaximoTimeRegister.objects.filter(date__range=(date(2015, 9, 1), date(2015, 9, 15))).assign_projects_from_ticket()
self.assertEqual(154, updated)
registers = MaximoTimeRegister.objects.filter(project__isnull=False)
self.assertEqual(79, len(registers))
def test_assign_projects_from_ticket(self):
project = ProjectFactory.create()
tickets = MaximoTicket.objects.all()[:5]
for ticket in tickets:
ticket.project = project
ticket.save()
TestExcel.create_time_registers(date(2015, 9, 1), date(2015, 9, 30))
updated = MaximoTimeRegister.objects.assign_projects_from_ticket()
self.assertEqual(294, updated)
registers = MaximoTimeRegister.objects.filter(project__isnull=False)
self.assertEqual(149, len(registers))
def test_export_time(self):
'''
Verifies the export for MaximoTimeRegister objects to an excel sheet. That contains the following columns:
Company Id Username Date Hours Pay Rate Ticket Type Ticket Number Ticket Name Memo Project Project Source
:return:
'''
TestExcel.create_time_registers(date(2015, 9, 1), date(2015, 9, 30))
out = StringIO()
filename = os.path.join(TEST_OUTPUT_PATH, '%s_%s.xlsx' % ('test_export_time', timezone.now().strftime('%Y%m%d_%H%M')))
call_command('maximo_data', filename, export_time=True)
self.assertTrue(os.path.exists(filename))
row_data, row_count = TestExcel.get_list_from_workbook(filename, 'Time')
self.assertEqual(row_count, MaximoTimeRegister.objects.count())
self.assertEqual(11, len(row_data[0]))
os.remove(filename)
self.assertFalse(os.path.exists(filename))
def test_sum_hours(self):
project = ProjectFactory.create()
tickets = MaximoTicket.objects.all()[:5]
for ticket in tickets:
ticket.project = project
ticket.save()
TestExcel.create_time_registers(date(2015, 9, 1), date(2015, 9, 3))
updated = MaximoTimeRegister.objects.assign_projects_from_ticket()
#self.assertEqual(294, updated)
total_hours = Decimal(0.0)
for tr in MaximoTimeRegister.objects.filter(project=project):
total_hours += tr.regular_hours
project_with_hours2 = Project.objects.filter(pk=project.pk).sum_regular_hours()[0]
project_with_hours = Project.objects.filter(pk=project.pk).sum_regular_hours()[0]
self.assertEqual(total_hours, project_with_hours.total_regular_hours)
self.assertEqual(120.0, project_with_hours.total_regular_hours)
def test_export_employees(self):
TestExcel.create_time_registers(date(2015, 9, 1), date(2015, 9, 30))
content = StringIO()
#filename = os.path.join(TEST_OUTPUT_PATH, filename_with_datetime('test_')
call_command('export_employees', 'TINO-NS', stdout=content)
content.seek(0)
lines = content.readlines()
filename = None
for line in lines:
regexp = r'^File\sexport\s([\w,\-,\\,:]+)(\.xlsx)'
pattern = re.compile(regexp)
match = pattern.match(line)
if match:
filename = match.group(1) + match.group(2)
#print(filename)
#print(line)
self.assertTrue(os.path.exists(filename))
def test_export_employees(self):
TestExcel.create_time_registers(date(2015, 9, 1), date(2015, 9, 30))
content = StringIO()
#filename = os.path.join(TEST_OUTPUT_PATH, filename_with_datetime('test_')
call_command('export_employees', 'TINO-NS', stdout=content)
content.seek(0)
lines = content.readlines()
filename = None
for line in lines:
regexp = r'^File\sexport\s([\w,\-,\\,:]+)(\.xlsx)'
pattern = re.compile(regexp)
match = pattern.match(line)
if match:
filename = match.group(1) + match.group(2)
#print(filename)
#print(line)
self.assertTrue(os.path.exists(filename))
def test_sum_hours(self):
project = ProjectFactory.create()
tickets = MaximoTicket.objects.all()[:5]
for ticket in tickets:
ticket.project = project
ticket.save()
TestExcel.create_time_registers(date(2015, 9, 1), date(2015, 9, 3))
updated = MaximoTimeRegister.objects.assign_projects_from_ticket()
#self.assertEqual(294, updated)
total_hours = Decimal(0.0)
for tr in MaximoTimeRegister.objects.filter(project=project):
total_hours += tr.regular_hours
project_with_hours2 = Project.objects.filter(
pk=project.pk).sum_regular_hours()[0]
project_with_hours = Project.objects.filter(
pk=project.pk).sum_regular_hours()[0]
self.assertEqual(total_hours, project_with_hours.total_regular_hours)
self.assertEqual(120.0, project_with_hours.total_regular_hours)
def test_export_time(self):
'''
Verifies the export for MaximoTimeRegister objects to an excel sheet. That contains the following columns:
Company Id Username Date Hours Pay Rate Ticket Type Ticket Number Ticket Name Memo Project Project Source
:return:
'''
TestExcel.create_time_registers(date(2015, 9, 1), date(2015, 9, 30))
out = StringIO()
filename = os.path.join(
TEST_OUTPUT_PATH, '%s_%s.xlsx' %
('test_export_time', timezone.now().strftime('%Y%m%d_%H%M')))
call_command('maximo_data', filename, export_time=True)
self.assertTrue(os.path.exists(filename))
row_data, row_count = TestExcel.get_list_from_workbook(
filename, 'Time')
self.assertEqual(row_count, MaximoTimeRegister.objects.count())
self.assertEqual(11, len(row_data[0]))
os.remove(filename)
self.assertFalse(os.path.exists(filename))