Python SnowflakeEngineSpec.cancel_query Exemples

Exemple #1

def test_cancel_query_failed(engine_mock: mock.Mock) -> None:
    from superset.db_engine_specs.snowflake import SnowflakeEngineSpec
    from superset.models.sql_lab import Query

    query = Query()
    cursor_mock = engine_mock.raiseError.side_effect = Exception()
    assert SnowflakeEngineSpec.cancel_query(cursor_mock, query, "123") is False

Exemple #2

def test_cancel_query(engine_mock: mock.Mock) -> None:
    from superset.db_engine_specs.snowflake import SnowflakeEngineSpec
    from superset.models.sql_lab import Query

    query = Query()
    cursor_mock = engine_mock.return_value.__enter__.return_value
    assert SnowflakeEngineSpec.cancel_query(cursor_mock, query, "123") is True

Exemple #3

Fichier : snowflake_tests.py Projet : yoanyombapro1234/incubator-superset

 def test_cancel_query(self, engine_mock):
     query = Query()
     cursor_mock = engine_mock.return_value.__enter__.return_value
     assert SnowflakeEngineSpec.cancel_query(cursor_mock, query, 123) is True

Exemple #4

Fichier : snowflake_tests.py Projet : yoanyombapro1234/incubator-superset

 def test_cancel_query_failed(self, engine_mock):
     query = Query()
     cursor_mock = engine_mock.raiseError.side_effect = Exception()
     assert SnowflakeEngineSpec.cancel_query(cursor_mock, query, 123) is False