Ejemplo n.º 1
0
    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()
Ejemplo n.º 2
0
    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))