def test_process_line_with_id_in_API(self):
test_processor = CSV_Processor("data-test-formatted.csv", "output.csv", False)
test_processor._dest = cStringIO.StringIO()
line = "lexcorp,Lex,2011-01-12,12345"
columns = {'Account ID': 3, 'Created On': 2}
test_processor._process_line(line, columns)
self.assertEqual(test_processor._dest.getvalue(), "12345,2011-01-12,good,2011-01-12\n")
def test_process_line_with_id_not_in_API(self):
test_processor = CSV_Processor("data-test-formatted.csv", "output.csv", False)
test_processor._dest = cStringIO.StringIO()
line = "vanhalen,Eddie,1984-01-12,OU812"
columns = {'Account ID': 3, 'Created On': 2}
sys.stderr = cStringIO.StringIO()
test_processor._process_line(line, columns)
self.assertEqual(sys.stderr.getvalue(), "ID OU812 not found in API; skipped...\n")
def test_process_file_without_account_id_column(self):
"""
No idea why this fails.
"""
test_processor = CSV_Processor("data-test-no-account-id.csv", "output.csv", False)
test_processor._dest = cStringIO.StringIO()
test_processor.process_file()
self.assertRaises(ValueError)
def test_process_file_without_created_on_column(self):
"""
This also barfs on the SystemExit call.
"""
test_processor = CSV_Processor("data-test-no-created-on.csv", "output.csv", False)
test_processor._dest = cStringIO.StringIO()
test_processor.process_file()
self.assertRaises(ValueError)
def test_process_correctly_formatted_file(self):
test_processor = CSV_Processor("data-test-formatted.csv", "output.csv", False)
test_processor._dest = cStringIO.StringIO()
test_processor.process_file()
self.assertEqual(test_processor._dest.getvalue(), "Account ID,Created On,Status,Status Set On\n12345,2011-01-12,good,2011-01-12\n")
