diff --git a/swh/core/tests/db_testing.py b/swh/core/tests/db_testing.py --- a/swh/core/tests/db_testing.py +++ b/swh/core/tests/db_testing.py @@ -3,6 +3,7 @@ # License: GNU General Public License version 3, or any later version # See top-level LICENSE file for more information +import os import psycopg2 import subprocess @@ -244,14 +245,18 @@ TEST_DB_NAME = 'softwareheritage-test' TEST_DB_DUMP = None - TEST_DB_DUMP_TYPE = 'pg_dump' + TEST_DB_DUMP_TYPE = None + DB_DUMP_TYPES = {'.sql': 'psql', '.dump': 'pg_dump'} @classmethod def setUpClass(cls): cls.dbname = cls.TEST_DB_NAME + dump_type = (cls.TEST_DB_DUMP_TYPE or + cls.DB_DUMP_TYPES[os.path.splitext(cls.TEST_DB_DUMP)[-1]]) + cls.add_db(name=cls.TEST_DB_NAME, dump=cls.TEST_DB_DUMP, - dump_type=cls.TEST_DB_DUMP_TYPE) + dump_type=dump_type) super().setUpClass() def setUp(self):