def test_take_date(self):
now = timezone.now()
self.assertEqual(take_date('today', now), now - timedelta(hours=24))
self.assertEqual(take_date('week', now), now - timedelta(days=7))
self.assertEqual(take_date('month', now), now - timedelta(days=32))
self.assertEqual(take_date('year', now), now - timedelta(days=365))
self.assertEqual(
take_date('beginning', now),
now - timedelta(days=1000)
)
def test_create_report(self):
now = timezone.now()
user = User()
user.created = now
user.save()
category = Category()
category.name = 'Test category'
category.description = ''
category.user = user
category.save()
subcategory = Subcategory()
subcategory.name = ''
subcategory.category = category
subcategory.save()
payment = Payment()
payment.value = 10
payment.currency = 'EUR'
payment.category = category
payment.subcategory = subcategory
payment.comment = ''
payment.date_time = now
payment.user = user
payment.save()
report = Report()
report.user = user
report.created = now.strftime('%Y-%m-%d %H:%M:%S')
report.report_type = 'today'
report.report_date = now.strftime('%Y-%m-%d %H:%M:%S')
report.start_date = take_date('today')
report.end_date = now.strftime('%Y-%m-%d %H:%M:%S')
report.currency = 'BGN'
report.is_active = 1
report.save()
report.url = 'app_cashtracker/reports/{}.pdf'.format(report)
report.save()
self.assertEqual(report, get_object_or_404(Report, id=report.id))
self.assertEqual(Report.generate_report_pdf(report), report)
self.assertEqual(report.category_names(), 'All')
self.assertEqual(report.add_category(category), None)
self.assertEqual(report.add_payment(payment), None)
self.assertEqual(report.fetch_payments(), [payment])
self.assertEqual(Report.fetch_reports(user), [report])
self.assertEqual(
str(report),
'Report_from_today_in_BGN_({})'.format(report.id)
)
def test_create_report(self):
now = timezone.now()
user = User()
user.created = now
user.save()
category = Category()
category.name = 'Test category'
category.description = ''
category.user = user
category.save()
subcategory = Subcategory()
subcategory.name = ''
subcategory.category = category
subcategory.save()
payment = Payment()
payment.value = 10
payment.currency = 'EUR'
payment.category = category
payment.subcategory = subcategory
payment.comment = ''
payment.date_time = now
payment.user = user
payment.save()
report = Report()
report.user = user
report.created = now.strftime('%Y-%m-%d %H:%M:%S')
report.report_type = 'today'
report.report_date = now.strftime('%Y-%m-%d %H:%M:%S')
report.start_date = take_date('today')
report.end_date = now.strftime('%Y-%m-%d %H:%M:%S')
report.currency = 'BGN'
report.is_active = 1
report.save()
report.url = 'app_cashtracker/reports/{}.pdf'.format(report)
report.save()
report_pdf = ReportPDF(report)
self.assertEqual(len(report_pdf.elements), 0)
report_pdf.generate_header()
self.assertEqual(len(report_pdf.elements), 5)
report_pdf.generate_statistics_data_and_table()
self.assertEqual(len(report_pdf.elements), 8)
report_pdf.generate_pie_charts()
self.assertEqual(len(report_pdf.elements), 9)
if len(report_pdf.lc_data) != 0:
report_pdf.generate_line_charts()
self.assertEqual(len(report_pdf.elements), 10)
report_pdf.build_and_save()
self.assertEqual(ReportPDF.payments_table_labels, [
'Date',
'Name',
'Category',
'Subcategory',
'Comment',
'Value'
])
