def test_incrementing_invoice_number(self):
client1, client2 = ClientFactory.create_batch(2)
with freeze_time('2015-10-02'):
invoice1 = InvoiceFactory.create(client=client1,
date=date(2015, 9, 30))
invoice1bis = InvoiceFactory.create(client=client1,
date=date(2015, 9, 30))
invoice1bis.delete()
invoice2 = InvoiceFactory.create(client=client2,
date=date(2015, 10, 31))
self.assertGreater(invoice2.pk - 1, invoice1.pk)
invoice1.generate_invoice_number()
invoice2.generate_invoice_number()
this_year = str(timezone.now().year)
self.assertEqual(invoice1.invoice_number,
'201500001'.format(this_year))
self.assertEqual(invoice2.invoice_number,
'201500002'.format(this_year))
with freeze_time('2016-02-23'):
invoice3 = InvoiceFactory.create(client=client2,
date=date(2016, 2, 15))
invoice3.generate_invoice_number()
self.assertEqual(invoice3.invoice_number,
'201600003'.format(this_year))
def test_multiple_generate_idempotent(self):
client = ClientFactory.create()
WorkEntryFactory.create(
project__client=client,
start=datetime(2015, 9, 14, 8, 0).replace(tzinfo=timezone.utc),
end=datetime(2015, 9, 14, 12, 0).replace(tzinfo=timezone.utc)
)
invoice = InvoiceFactory.create(client=client, date=date(2015, 9, 30))
invoice.generate()
items = InvoiceItem.objects.all()
self.assertEqual(items.count(), 1)
# second call may not generate new items
invoice.generate()
self.assertEqual(items.count(), 1)
def test_multiple_generate_idempotent(self):
client = ClientFactory.create()
WorkEntryFactory.create(project__client=client,
start=datetime(2015, 9, 14, 8,
0).replace(tzinfo=timezone.utc),
end=datetime(2015, 9, 14, 12,
0).replace(tzinfo=timezone.utc))
invoice = InvoiceFactory.create(client=client, date=date(2015, 9, 30))
invoice.generate()
items = InvoiceItem.objects.all()
self.assertEqual(items.count(), 1)
# second call may not generate new items
invoice.generate()
self.assertEqual(items.count(), 1)
def test_incrementing_invoice_number(self):
client1, client2 = ClientFactory.create_batch(2)
with freeze_time('2015-10-02'):
invoice1 = InvoiceFactory.create(client=client1, date=date(2015, 9, 30))
invoice1bis = InvoiceFactory.create(client=client1, date=date(2015, 9, 30))
invoice1bis.delete()
invoice2 = InvoiceFactory.create(client=client2, date=date(2015, 10, 31))
self.assertGreater(invoice2.pk-1, invoice1.pk)
invoice1.generate_invoice_number()
invoice2.generate_invoice_number()
this_year = str(timezone.now().year)
self.assertEqual(invoice1.invoice_number, '201500001'.format(this_year))
self.assertEqual(invoice2.invoice_number, '201500002'.format(this_year))
with freeze_time('2016-02-23'):
invoice3 = InvoiceFactory.create(client=client2, date=date(2016, 2, 15))
invoice3.generate_invoice_number()
self.assertEqual(invoice3.invoice_number, '201600003'.format(this_year))
def test_pickup_work_entries(self):
"""
Test that the work entries are correctly picked up for a given invoice date.
"""
client1, client2 = ClientFactory.create_batch(2)
project1, project2 = ProjectFactory.create_batch(2, client=client1)
project3 = ProjectFactory.create(client=client2)
# worked a bit on project 1
WorkEntryFactory.create(
project=project1,
start=datetime(2015, 9, 14, 8, 0).replace(tzinfo=timezone.utc),
end=datetime(2015, 9, 14, 12, 0).replace(tzinfo=timezone.utc)
)
# worked a bit on project 2
WorkEntryFactory.create(
project=project2,
start=datetime(2015, 9, 14, 13, 0).replace(tzinfo=timezone.utc),
end=datetime(2015, 9, 14, 15, 30).replace(tzinfo=timezone.utc)
)
WorkEntryFactory.create(
project=project2,
start=datetime(2015, 9, 30, 13, 0).replace(tzinfo=timezone.utc),
end=datetime(2015, 9, 30, 15, 30).replace(tzinfo=timezone.utc)
)
# future entry to be ignored
WorkEntryFactory.create(
project=project1,
start=datetime(2015, 10, 1, 8, 0).replace(tzinfo=timezone.utc),
end=datetime(2015, 10, 1, 11, 0).replace(tzinfo=timezone.utc)
)
# different client
WorkEntryFactory.create(
project=project3,
start=datetime(2015, 9, 21, 8, 0).replace(tzinfo=timezone.utc),
end=datetime(2015, 9, 21, 11, 0).replace(tzinfo=timezone.utc)
)
# create the invoices
invoice1 = InvoiceFactory.create(client=client1, date=date(2015, 9, 30))
invoice2 = InvoiceFactory.create(client=client1, date=date(2015, 10, 31))
# generate the invoices and check that the correct number of lines are created
invoice1.generate()
self.assertIsNotNone(invoice1.generated)
self.assertEqual(invoice1.invoice_number, '201500001')
self.assertEqual(InvoiceItem.objects.count(), 3)
# check the financial data
for item in InvoiceItem.objects.all():
self.assertEqual(item.rate, item.project.base_rate)
expected_vat_free_total = project1.base_rate * 4 + project2.base_rate * 5
self.assertEqual(invoice1.total_no_vat, expected_vat_free_total)
self.assertEqual(invoice1.total_with_vat, expected_vat_free_total * Decimal('1.21'))
invoice2.generate()
self.assertEqual(InvoiceItem.objects.count(), 4)
def test_pickup_work_entries(self):
"""
Test that the work entries are correctly picked up for a given invoice date.
"""
client1, client2 = ClientFactory.create_batch(2)
project1, project2 = ProjectFactory.create_batch(2, client=client1)
project3 = ProjectFactory.create(client=client2)
# worked a bit on project 1
WorkEntryFactory.create(project=project1,
start=datetime(2015, 9, 14, 8,
0).replace(tzinfo=timezone.utc),
end=datetime(2015, 9, 14, 12,
0).replace(tzinfo=timezone.utc))
# worked a bit on project 2
WorkEntryFactory.create(project=project2,
start=datetime(2015, 9, 14, 13,
0).replace(tzinfo=timezone.utc),
end=datetime(2015, 9, 14, 15,
30).replace(tzinfo=timezone.utc))
WorkEntryFactory.create(project=project2,
start=datetime(2015, 9, 30, 13,
0).replace(tzinfo=timezone.utc),
end=datetime(2015, 9, 30, 15,
30).replace(tzinfo=timezone.utc))
# future entry to be ignored
WorkEntryFactory.create(project=project1,
start=datetime(2015, 10, 1, 8,
0).replace(tzinfo=timezone.utc),
end=datetime(2015, 10, 1, 11,
0).replace(tzinfo=timezone.utc))
# different client
WorkEntryFactory.create(project=project3,
start=datetime(2015, 9, 21, 8,
0).replace(tzinfo=timezone.utc),
end=datetime(2015, 9, 21, 11,
0).replace(tzinfo=timezone.utc))
# create the invoices
invoice1 = InvoiceFactory.create(client=client1,
date=date(2015, 9, 30))
invoice2 = InvoiceFactory.create(client=client1,
date=date(2015, 10, 31))
# generate the invoices and check that the correct number of lines are created
invoice1.generate()
self.assertIsNotNone(invoice1.generated)
self.assertEqual(invoice1.invoice_number, '201500001')
self.assertEqual(InvoiceItem.objects.count(), 3)
# check the financial data
for item in InvoiceItem.objects.all():
self.assertEqual(item.rate, item.project.base_rate)
expected_vat_free_total = project1.base_rate * 4 + project2.base_rate * 5
self.assertEqual(invoice1.total_no_vat, expected_vat_free_total)
self.assertEqual(invoice1.total_with_vat,
expected_vat_free_total * Decimal('1.21'))
invoice2.generate()
self.assertEqual(InvoiceItem.objects.count(), 4)
