def test_to_string(self):
report = Report(report='my_report_class')
self.assertEqual(str(report), self.report_instance.name)
report.save()
self.assertEqual(str(report), '{} #{}'.format(self.report_instance.name, report.id))
def test_process_must_set_status_as_processed_after_report_process(self):
report = Report(report='my_report_class')
report.save()
report.process()
report = Report.objects.get(id=report.id)
self.assertEqual(report.status, Report.STATUS_PROCESSED)
def test_process_must_call_process_from_report_class(self):
report = Report(report='my_report_class')
report.save()
report.process()
self.my_import_mocked.assert_called_once_with(report.report)
self.assertTrue(self.report_instance.process.called)
def test_process_must_call_process_from_report_class(self):
report = Report(report='my_report_class')
report.save()
report.process()
self.my_import_mocked.assert_called_once_with(report.report)
self.assertTrue(self.report_instance.process.called)
def test_process_must_set_status_as_processed_after_report_process(self):
report = Report(report='my_report_class')
report.save()
report.process()
report = Report.objects.get(id=report.id)
self.assertEqual(report.status, Report.STATUS_PROCESSED)
def test_process_must_set_status_as_error_if_some_exception_is_raised(self):
self.report_instance.process.side_effect = Exception()
report = Report(report='my_report_class')
report.save()
self.assertRaises(Exception, report.process)
self.assertEqual(report.status, Report.STATUS_ERROR)
def test_to_string(self):
report = Report(report='my_report_class')
self.assertEqual(str(report), self.report_instance.name)
report.save()
self.assertEqual(str(report),
'{} #{}'.format(self.report_instance.name, report.id))
def test_process_must_set_status_as_error_if_some_exception_is_raised(
self):
self.report_instance.process.side_effect = Exception()
report = Report(report='my_report_class')
report.save()
self.assertRaises(Exception, report.process)
self.assertEqual(report.status, Report.STATUS_ERROR)
def test_process_must_set_process_time(self):
getcontext().prec = 5
start = Decimal(10.0000)
end = Decimal(15.1234)
self.patch('onmydesk.models.timer', side_effect=[start, end])
report = Report(report='my_report_class')
report.save()
report.process()
self.assertEqual(report.process_time, end - start)
def test_process_with_params_must_call_report_constructor_with_these_params(self):
report = Report(report='my_report_class')
params = {'type': 'whatever'}
report.set_params(params)
report.save()
report.process()
self.report_class.assert_called_once_with(params=params)
def test_process_must_set_process_time(self):
getcontext().prec = 5
start = Decimal(10.0000)
end = Decimal(15.1234)
self.patch('onmydesk.models.timer', side_effect=[start, end])
report = Report(report='my_report_class')
report.save()
report.process()
self.assertEqual(report.process_time, end - start)
def test_process_must_store_filepaths_result(self):
self.report_instance.output_filepaths = [
'/tmp/flunfa-2.tsv',
'/tmp/flunfa-3.tsv',
]
report = Report(report='my_report_class')
report.save()
report.process()
self.assertEqual(
report.results, ';'.join(self.report_instance.output_filepaths))
def test_process_with_params_must_call_report_constructor_with_these_params(
self):
report = Report(report='my_report_class')
params = {'type': 'whatever'}
report.set_params(params)
report.save()
report.process()
self.report_class.assert_called_once_with(params=params)
def test_process_must_store_filepaths_result(self):
self.report_instance.output_filepaths = [
'/tmp/flunfa-2.tsv',
'/tmp/flunfa-3.tsv',
]
report = Report(report='my_report_class')
report.save()
report.process()
self.assertEqual(report.results,
';'.join(self.report_instance.output_filepaths))
def test_process_must_set_status_as_processing_when_start(self):
self.patch('onmydesk.models.my_import', side_effect=Exception)
report = Report(report='my_report_class')
report.save()
self.assertEqual(report.status, Report.STATUS_PENDING)
try:
report.process()
except Exception:
pass
report = Report.objects.get(id=report.id)
self.assertEqual(report.status, Report.STATUS_PROCESSING)
def test_process_must_set_status_as_processing_when_start(self):
self.patch('onmydesk.models.my_import', side_effect=Exception)
report = Report(report='my_report_class')
report.save()
self.assertEqual(report.status, Report.STATUS_PENDING)
try:
report.process()
except Exception:
pass
report = Report.objects.get(id=report.id)
self.assertEqual(report.status, Report.STATUS_PROCESSING)
def test_result_links(self):
self.report_instance.output_filepaths = [
'/tmp/flunfa-2.tsv',
'/tmp/flunfa-3.tsv',
]
report = Report(report='my_report_class')
report.save()
report.process()
def my_handler(filepath):
url_example = 'http://someplace.com/somepath{}'
return url_example.format(filepath)
with mock.patch('onmydesk.models.my_import', return_value=my_handler):
links = report.result_links
self.assertEqual(links, [my_handler(i) for i in self.report_instance.output_filepaths])
def test_result_links(self):
self.report_instance.output_filepaths = [
'/tmp/flunfa-2.tsv',
'/tmp/flunfa-3.tsv',
]
report = Report(report='my_report_class')
report.save()
report.process()
def my_handler(filepath):
url_example = 'http://someplace.com/somepath{}'
return url_example.format(filepath)
with mock.patch('onmydesk.models.my_import', return_value=my_handler):
links = report.result_links
self.assertEqual(
links,
[my_handler(i) for i in self.report_instance.output_filepaths])
