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"]
]))
