def test_config_domain_cache(reset_config, monkeypatch, mocker):
dfs_mock = mocker.MagicMock()
dfs_mock.return_value = DOMAIN_REFERRAL
monkeypatch.setattr(pool, 'get_smb_tree', mocker.MagicMock())
monkeypatch.setattr(pool, 'dfs_request', dfs_mock)
config = pool.ClientConfig()
domain_referral = config.lookup_domain(DOMAIN_NAME)
assert domain_referral is None
config.domain_controller = DOMAIN_NAME
domain_referral = config.lookup_domain(DOMAIN_NAME)
assert domain_referral.domain_name == '\\%s' % DOMAIN_NAME
assert dfs_mock.call_count == 1
assert dfs_mock.call_args[0][1] == u''
def test_config_domain_cache_not_dfs_endpoint(reset_config, monkeypatch,
mocker):
dfs_mock = mocker.MagicMock()
dfs_mock.side_effect = InvalidParameter()
warning_mock = mocker.MagicMock()
monkeypatch.setattr(pool, 'get_smb_tree', mocker.MagicMock())
monkeypatch.setattr(pool, 'dfs_request', dfs_mock)
monkeypatch.setattr(pool.log, 'warning', warning_mock)
config = pool.ClientConfig()
config.domain_controller = DOMAIN_NAME
domain_referral = config.lookup_domain(DOMAIN_NAME)
assert domain_referral is None
assert warning_mock.call_count == 1
assert 'cannot use as DFS domain cache source' in warning_mock.call_args[
0][0]
def reset_config():
config = pool.ClientConfig()
client_guid = config.client_guid
username = config.username
password = config.password
domain_controller = config.domain_controller
skip_dfs = config.skip_dfs
yield
config.set(client_guid=client_guid,
username=username,
password=password,
skip_dfs=skip_dfs,
domain_controller=domain_controller)
config._domain_cache = []
config._referral_cache = []
def test_set_config_option(reset_config):
pool.ClientConfig(username='******')
assert pool.ClientConfig().username == 'test'
pool.ClientConfig(username=None)
assert not pool.ClientConfig().username
def test_client_config_private_key(reset_config):
with pytest.raises(
ValueError,
match='Cannot set private attribute _domain_controller'):
pool.ClientConfig().set(_domain_controller='test')