def test_context_manager(monkeypatch): """Asserts context manager properly opens/closes""" channel_ssh_auth_called = False def _channel_ssh_auth(cls, auth_password, auth_private_key_passphrase): nonlocal channel_ssh_auth_called channel_ssh_auth_called = True monkeypatch.setattr( "scrapli.transport.plugins.system.transport.SystemTransport.open", lambda x: None ) monkeypatch.setattr( "scrapli.channel.sync_channel.Channel.channel_authenticate_ssh", _channel_ssh_auth ) with Driver(host="localhost") as conn: pass assert channel_ssh_auth_called is True
def test_async_transport_exception(): """Assert we raise ScrapliValueError if an async transport is provided to the sync driver""" with pytest.raises(ScrapliValueError): Driver(host="localhost", transport="asynctelnet")
def sync_driver_telnet(): """Fixture to provide sync driver instance""" sync_driver_telnet = Driver(host="localhost", auth_username="******", transport="telnet") return sync_driver_telnet
def sync_driver(): """Fixture to provide sync driver instance""" sync_driver = Driver(host="localhost", auth_username="******") return sync_driver