async def test_jumphosts_fail_connect(netcfgbu_envars, log_vcr, inventory,
mock_asyncssh_connect, monkeypatch):
monkeypatch.setattr(jumphosts, "get_logger", Mock(return_value=log_vcr))
jh_spec = config_model.JumphostSpec(proxy="[email protected]:8022",
exclude=["os_name=eos"])
jumphosts.init_jumphosts(jumphost_specs=[jh_spec], inventory=inventory)
mock_asyncssh_connect.side_effect = asyncio.TimeoutError()
ok = await jumphosts.connect_jumphosts()
assert ok is False
mock_asyncssh_connect.side_effect = asyncssh.Error(code=10,
reason="nooooope")
ok = await jumphosts.connect_jumphosts()
assert ok is False
jh: jumphosts.JumpHost = jumphosts.JumpHost.available[0]
with pytest.raises(RuntimeError) as excinfo:
_ = jh.tunnel
errmsg = excinfo.value.args[0]
assert "not connected" in errmsg
log_recs = log_vcr.handlers[0].records
assert (log_recs[-1].msg ==
"JUMPHOST: connect to [email protected]:8022 failed: nooooope")
def test_jumphosts_pass_exlallused(inventory):
# exclude all OS names will result in no jump hosts required
jh_spec = config_model.JumphostSpec(proxy="1.2.3.4",
exclude=["os_name=.*"])
jumphosts.init_jumphosts(jumphost_specs=[jh_spec], inventory=inventory)
assert len(jumphosts.JumpHost.available) == 0
def test_jumphosts_pass_noused(inventory):
# without an include or exclude, this jump host will not be used
# TODO: consider this an error in config-model validation?
jh_spec = config_model.JumphostSpec(proxy="1.2.3.4")
jumphosts.init_jumphosts(jumphost_specs=[jh_spec], inventory=inventory)
assert len(jumphosts.JumpHost.available) == 0
def test_jumphosts_pass_exlused(inventory):
# exclude EOS devices to be used in the jump host, this will result in the
# jump host being required
jh_spec = config_model.JumphostSpec(proxy="1.2.3.4",
exclude=["os_name=eos"])
jumphosts.init_jumphosts(jumphost_specs=[jh_spec], inventory=inventory)
assert len(jumphosts.JumpHost.available) == 1
jh_use_count = Counter(
getattr(jumphosts.get_jumphost(rec), "name", None)
for rec in inventory)
assert jh_use_count["1.2.3.4"] == 4
async def test_jumphosts_pass_connect(inventory, mock_asyncssh_connect,
monkeypatch):
jh_spec = config_model.JumphostSpec(proxy="[email protected]:8022",
exclude=["os_name=eos"])
jumphosts.init_jumphosts(jumphost_specs=[jh_spec], inventory=inventory)
ok = await jumphosts.connect_jumphosts()
assert ok
assert mock_asyncssh_connect.called
assert mock_asyncssh_connect.call_count == 1
called = mock_asyncssh_connect.mock_calls[0]
assert called.kwargs["host"] == "1.2.3.4"
assert called.kwargs["username"] == "dummy-user"
assert called.kwargs["port"] == 8022
jh: jumphosts.JumpHost = jumphosts.JumpHost.available[0]
assert jh.tunnel is not None