From fa627e21f1c3469e6ce6385ae74b136e8e8398aa Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gonzalo=20Mart=C3=ADnez=20Lema?= Date: Thu, 10 Mar 2022 11:56:44 +0100 Subject: [PATCH] Add tests --- invisible_cities/database/download_test.py | 26 +++++++++++++++++++++- 1 file changed, 25 insertions(+), 1 deletion(-) diff --git a/invisible_cities/database/download_test.py b/invisible_cities/database/download_test.py index bf4bdc071..9c45c7ac7 100644 --- a/invisible_cities/database/download_test.py +++ b/invisible_cities/database/download_test.py @@ -21,10 +21,34 @@ def test_create_table_sqlite(dbname, output_tmpdir): connSqlite = sqlite3.connect(dbfile) connMySql = pymysql.connect(host="neutrinos1.ific.uv.es", - user='nextreader',passwd='readonly', db=dbname) + user='nextreader',passwd='readonly', db=dbname) cursorMySql = connMySql .cursor() cursorSqlite = connSqlite.cursor() for table in db.tables: db.create_table_sqlite(cursorSqlite, cursorMySql, table) + + +@mark.parametrize('dbname', db.dbnames) +def test_table_assignment(dbname): + for name in db.common_tables: + assert name in db.table_dict[dbname] + + for name in db.extended.get(dbname, ()): + assert name in db.table_dict[dbname] + + +@mark.parametrize('dbname', db.dbnames) +def test_tables_exist(dbname): + connMySql = pymysql.connect(host="neutrinos1.ific.uv.es", + user='nextreader',passwd='readonly', db=dbname) + + cursor = connMySql.cursor() + cursor.execute("Show tables;") + available = cursor.fetchall() + + for name in db.table_dict[dbname]: + assert (name,) in available + +