Python get_auth_config示例

示例#1

def test_onlyhostset_none(mocker):
    mocker.patch.dict(os.environ,{'DATABRICKS_HOST':'host'})
    mocker.patch.dict(os.environ,{'DATABRICKS_TOKEN':''})

    config = auth.get_auth_config()
    # Assert
    assert config == None

示例#2

def test_tokenhostset_okay(mocker):
    mocker.patch.dict(os.environ,{'DATABRICKS_HOST':'host'})
    mocker.patch.dict(os.environ,{'DATABRICKS_TOKEN':'token'})

    config = auth.get_auth_config()
    # Assert
    assert config != None
    assert config.host == 'host'
    assert config.token == 'token'