Python get_or_create_dbの例

コード例 #1

 def test_get_or_create_db(self):
     get_or_create_db("test_db", "sqlite:///superset.db")
     database = db.session.query(Database).filter_by(database_name="test_db").one()
     self.assertIsNotNone(database)
     self.assertEqual(database.sqlalchemy_uri, "sqlite:///superset.db")
     self.assertIsNotNone(
         security_manager.find_permission_view_menu("database_access", database.perm)
     )
     # Test change URI
     get_or_create_db("test_db", "sqlite:///changed.db")
     database = db.session.query(Database).filter_by(database_name="test_db").one()
     self.assertEqual(database.sqlalchemy_uri, "sqlite:///changed.db")
     db.session.delete(database)
     db.session.commit()

コード例 #2

ファイル: utils_tests.py プロジェクト: EBoisseauSierra/superset

 def test_get_or_create_db_invalid_uri(self):
     with self.assertRaises(DatabaseInvalidError):
         get_or_create_db("test_db", "yoursql:superset.db/()")

コード例 #3

ファイル: config_tests.py プロジェクト: airbnb/superset-fork

 def setUp(self) -> None:
     self.login(username="******")
     self._test_db_id = get_or_create_db(
         "column_test_db", app.config["SQLALCHEMY_DATABASE_URI"]).id
     self._old_sqla_table_mutator = app.config["SQLA_TABLE_MUTATOR"]

コード例 #4

ファイル: update.py プロジェクト: opus-42/incubator-superset

def set_database_uri(database_name: str, uri: str, skip_create: bool) -> None:
    """Updates a database connection URI"""
    database_utils.get_or_create_db(database_name, uri, not skip_create)