Method change_database

tests/common/impala_test_suite.py:1191–1213

Change impala_client to point to approriate database under test and revert it to default database once it exit the with scope. Restore client configuration back to its default when getting in and getting out of 'with' scope. Test method should try to use fully qualified table name in the

(cls, impala_client, table_format=None, db_name=None,
                      scale_factor=None)

  @classmethod
  @contextlib.contextmanager
  def change_database(cls, impala_client, table_format=None, db_name=None,
                      scale_factor=None):
    """Change impala_client to point to approriate database under test and revert
    it to default database once it exit the with scope.
    Restore client configuration back to its default when getting in and getting out of
    'with' scope. Test method should try to use fully qualified table name in the test
    query as much as possible and only use this context manager function when it is
    not practical to do so.
    Sample usage:

      with ImpalaTestSuite.change_database(client, db_name='functional_parquet'):
        # client clear configuration and pointing to 'functional_parquet' database here.
        client.execute('show tables')
      # client clear configuration and pointing to 'default' database after it exit the
      # 'with' scope.
      client.execute('show tables')
    """
    try:
      cls.__change_client_database(impala_client, table_format=table_format,
                                   db_name=db_name, scale_factor=scale_factor)
      yield
    finally:
      cls.__change_client_database(impala_client, db_name='default')

  def execute_wrapper(function):
    """