def test_exit_code_ARGUMENT_ERROR(self): print("test_exit_code_ARGUMENT_ERROR") # Test that command raises SystemExit exception with self.assertRaises(SystemExit) as cm: csv2db.run([ "load", "-f", "../resources/201811-citibike-tripdata.csv.gz", "-t", "STAGING" ]) # Test that command threw SystemExit with status code 2 self.assertEqual(cm.exception.code, 2)
def test_exit_code_SUCCESS(self): print("test_exit_code_SUCCESS") self.assertEqual( f.ExitCodes.SUCCESS.value, csv2db.run([ "gen", "-f", "../resources/201811-citibike-tripdata.csv.gz", "-t", "STAGING" ]))
def test_loading(self): print("test_loading") self.assertEqual( f.ExitCodes.SUCCESS.value, csv2db.run([ "load", "-f", "../resources/201811-citibike-tripdata.csv", "-u", "test", "-p", "test", "-t", "STAGING" ]))
def test_exit_code_DATABASE_ERROR(self): print("test_exit_code_DATABASE_ERROR") self.assertEqual( f.ExitCodes.DATABASE_ERROR.value, csv2db.run([ "load", "-f", "../resources/201811-citibike-tripdata.csv", "-u", "INVALIDUSER", "-p", "test", "-t", "STAGING" ]))
def test_exit_code_GENERIC_ERROR(self): print("test_exit_code_GENERIC_ERROR") self.assertEqual( f.ExitCodes.GENERIC_ERROR.value, csv2db.run([ "gen", "-f", "../resources/bad/201811-citibike-tripdata-invalid.csv.zip" ]))
def test_exit_code_DATA_LOADING_ERROR(self): print("test_exit_code_DATA_LOADING_ERROR") self.assertEqual( f.ExitCodes.DATA_LOADING_ERROR.value, csv2db.run([ "load", "-f", "../resources/bad/201811-citibike-tripdata-bad-data.csv", "-u", "test", "-p", "test", "-t", "DOES_NOT_EXIST" ]))
def test_loading_Db2(self): print("test_loading_Db2") self.assertEqual( f.ExitCodes.SUCCESS.value, csv2db.run([ "load", "-f", "../resources/201811-citibike-tripdata.csv", "-o", "db2", "-u", login["db2_user"], "-p", login["password"], "-d", login["database"], "-t", login["table"] ]))
def test_unicode_file_Db2(self): print("test_unicode_file_Db2") self.assertEqual( f.ExitCodes.SUCCESS.value, csv2db.run([ "load", "-f", "../resources/allCountries.1000.txt.gz", "-o", "db2", "-u", login["db2_user"], "-p", login["password"], "-d", login["database"], "-t", "LOCATIONS", "-s", " " ]))
def test_exit_code_DATA_LOADING_ERROR(self): print("test_exit_code_DATA_LOADING_ERROR") self.assertEqual( f.ExitCodes.DATA_LOADING_ERROR.value, csv2db.run([ "load", "-f", "../resources/201811-citibike-tripdata.csv", "-u", login["user"], "-p", login["password"], "-d", login["database"], "-t", "DOES_NOT_EXIST", "--debug" ]))
def test_load_file_with_insufficient_columns(self): print("test_load_file_with_insufficient_columns") self.assertEqual( f.ExitCodes.SUCCESS.value, csv2db.run([ "load", "-f", "../resources/bad/201811-citibike-tripdata-not-enough-columns.csv", "-u", login["user"], "-p", login["password"], "-d", login["database"], "-t", login["table"], "--debug" ]))
def test_empty_file(self): print("test_empty_file") self.assertEqual( f.ExitCodes.SUCCESS.value, csv2db.run([ "load", "-f", "../resources/bad/201811-citibike-tripdata-empty.csv", "-u", login["user"], "-p", login["password"], "-d", login["database"], "-t", login["table"] ]))