def test_ErrorDBNotExists(tmpdir):
'''Test the class ErrorDBNotExists (error when a database doesn't exist)'''
# This file is not created, just a tmp path
tmpfile = os.path.join(tmpdir.strpath, "tmpfile")
# Test instantiation
test_error = NCBImetaErrors.ErrorDBNotExists(tmpfile)
# Test str representation (error message)
error_output = str(test_error)
error_expect = ("\n\nDatabase does not exist." + "\n" + tmpfile)
assert error_output == error_expect
args = vars(parser.parse_args())
db_name = args['dbName']
output_dir = args['outputDir']
#-----------------------------------------------------------------------#
# Argument Checking #
#-----------------------------------------------------------------------#
# Check if database exists
if os.path.exists(db_name):
conn = sqlite3.connect(db_name)
print('\nOpening database: ' + db_name, flush=True)
else:
raise NCBImetaErrors.ErrorDBNotExists(db_name)
# Check if output dir exists
if not os.path.exists(output_dir):
os.makedirs(CONFIG_OUTPUT_DIR)
# no errors were raised, safe to connect to db
cur = conn.cursor()
#-----------------------------------------------------------------------#
# Process Database #
#-----------------------------------------------------------------------#
# Get a list of tables
cur.execute("SELECT name FROM sqlite_master WHERE type='table';")
table_list = cur.fetchall()