def setUp(self): sql_url = "https://raw.githubusercontent.com/asyml/stave/master" \ "/simple-backend/example_db.sql" self.datapack_table: str = StaveMultiDocSqlReader.default_configs( )['datapack_table'] self.multipack_table: str = StaveMultiDocSqlReader.default_configs( )['multipack_table'] self.project_table: str = StaveDataPackSqlReader.default_configs( )['project_table'] self.temp_dir = tempfile.TemporaryDirectory() maybe_download(sql_url, self.temp_dir.name, 'example_db.sql') sql_script: str = os.path.join(self.temp_dir.name, 'example_db.sql') self.sql_db: str = os.path.join(self.temp_dir.name, 'db.sqlite3') pack_count: int mp_count: int with open(sql_script) as q_file: # Build the example database by executing the sample sql script. q = q_file.read() conn = sqlite3.connect(self.sql_db) c = conn.cursor() c.executescript(q) conn.commit()
def setUp(self): self.datapack_table: str = StaveMultiDocSqlReader.default_configs()[ 'datapack_table'] self.multipack_table: str = StaveMultiDocSqlReader.default_configs()[ 'multipack_table'] self.project_table: str = StaveDataPackSqlReader.default_configs()[ 'project_table' ] # This path correspond to the .travis.yml. self.sql_db: str = os.path.abspath(os.path.join( os.path.dirname(os.path.realpath(__file__)), *([os.path.pardir] * 4), 'db.sqlite3')) self.assertTrue(os.path.exists(self.sql_db))