def test_ping(proxy_minion_config): """ check netmiko_proxy ping method """ proxy_minion_config["skip_connect_on_init"] = True netmiko_proxy.init(proxy_minion_config) result = netmiko_proxy.ping() assert result is True
def test_ping(self): """ check netmiko_proxy ping method """ opts = salt.config.proxy_config(RUNTIME_VARS.TMP_PROXY_CONF_DIR + "/proxy", minion_id="proxytest") opts["skip_connect_on_init"] = True netmiko_proxy.init(opts) result = netmiko_proxy.ping() self.assertEqual(result, True)
def test_make_con(proxy_minion_config): """ check netmiko_proxy make_con method """ proxy_minion_config["skip_connect_on_init"] = True netmiko_proxy.init(proxy_minion_config) mock_connection = MockNetmikoConnection with patch.object(netmiko_proxy, "ConnectHandler", mock_connection, create=True): result = netmiko_proxy.make_con() assert result is not None
def test_init_skip_connect_on_init_false(self): """ check netmiko_proxy init method """ opts = salt.config.proxy_config(RUNTIME_VARS.TMP_PROXY_CONF_DIR + "/proxy", minion_id="proxytest") opts["skip_connect_on_init"] = False mock_make_con = MagicMock() with patch.object(netmiko_proxy, "make_con", mock_make_con): result = netmiko_proxy.init(opts) self.assertIn("netmiko_device", netmiko_proxy.__context__) self.assertIn("args", netmiko_proxy.__context__["netmiko_device"]) self.assertIn("initialized", netmiko_proxy.__context__["netmiko_device"]) self.assertTrue( netmiko_proxy.__context__["netmiko_device"]["initialized"]) self.assertIn("up", netmiko_proxy.__context__["netmiko_device"]) self.assertTrue(netmiko_proxy.__context__["netmiko_device"]["up"]) self.assertIn("always_alive", netmiko_proxy.__context__["netmiko_device"]) self.assertNotIn("conneciton", netmiko_proxy.__context__["netmiko_device"])
def test_make_con(self): """ check netmiko_proxy make_con method """ opts = salt.config.proxy_config(RUNTIME_VARS.TMP_PROXY_CONF_DIR + "/proxy", minion_id="proxytest") opts["skip_connect_on_init"] = True netmiko_proxy.init(opts) mock_connection = MockNetmikoConnection with patch.object(netmiko_proxy, "ConnectHandler", mock_connection, create=True): result = netmiko_proxy.make_con() self.assertNotEqual(result, None)
def test_initialized(proxy_minion_config): """ check netmiko_proxy alive method """ proxy_minion_config["skip_connect_on_init"] = True netmiko_proxy.init(proxy_minion_config) result = netmiko_proxy.initialized() assert not result # Always alive True with skip_connect_on_init on False # should return alive as True proxy_minion_config["skip_connect_on_init"] = False mock_make_con = MagicMock() with patch.object(netmiko_proxy, "make_con", mock_make_con): netmiko_proxy.init(proxy_minion_config) result = netmiko_proxy.initialized() assert result
def test_alive(self): """ check netmiko_proxy alive method """ # Always alive False with skip_connect_on_init on True # should return alive as True opts = salt.config.proxy_config(RUNTIME_VARS.TMP_PROXY_CONF_DIR + "/proxy", minion_id="proxytest") opts["skip_connect_on_init"] = True opts["proxy_always_alive"] = False netmiko_proxy.init(opts) result = netmiko_proxy.alive(opts) self.assertTrue(result) # Always alive True with skip_connect_on_init on True # should return alive as False opts["proxy_always_alive"] = True netmiko_proxy.init(opts) result = netmiko_proxy.alive(opts) self.assertFalse(result) # Always alive True with skip_connect_on_init on False # should return alive as True opts["skip_connect_on_init"] = False opts["proxy_always_alive"] = True mock_make_con = MagicMock() with patch.object(netmiko_proxy, "make_con", mock_make_con): netmiko_proxy.init(opts) result = netmiko_proxy.alive(opts) self.assertTrue(result)
def test_make_con_raise_exception(proxy_minion_config): """ check netmiko_proxy make_con method """ class ExpectedException(Exception): """ Custom exception class expected to raise as a side_effect """ def raise_exception(*arg, **kwarg): raise ExpectedException("expected") proxy_minion_config["skip_connect_on_init"] = True netmiko_proxy.init(proxy_minion_config) with patch.object(netmiko_proxy, "DEFAULT_CONNECTION_TIMEOUT", 0), patch.object( netmiko_proxy, "ConnectHandler", raise_exception, create=True ): with pytest.raises(ExpectedException): netmiko_proxy.make_con(0)
def test_make_con_raise_exception(self): """ check netmiko_proxy make_con method """ def raise_exception(*arg, **kwarg): raise Exception("expected") opts = salt.config.proxy_config(RUNTIME_VARS.TMP_PROXY_CONF_DIR + "/proxy", minion_id="proxytest") opts["skip_connect_on_init"] = True netmiko_proxy.init(opts) mock_connection = MockNetmikoConnection with patch.object(netmiko_proxy, "DEFAULT_CONNECTION_TIMEOUT", 0), patch.object(netmiko_proxy, "ConnectHandler", raise_exception, create=True): result = None try: result = netmiko_proxy.make_con(0) except Exception: # pylint: disable=broad-except self.assertEqual(result, None)
def test_init_skip_connect_on_init_true(proxy_minion_config): """ check netmiko_proxy init method """ proxy_minion_config["skip_connect_on_init"] = True assert netmiko_proxy.init(proxy_minion_config) assert "netmiko_device" in netmiko_proxy.__context__ netmiko_device = netmiko_proxy.__context__["netmiko_device"] assert "args" in netmiko_device assert "initialized" in netmiko_device assert not netmiko_device["initialized"] assert "up" in netmiko_device assert netmiko_device["up"] assert "always_alive" in netmiko_device assert "connection" not in netmiko_device
def test_init_skip_connect_on_init_false(proxy_minion_config): """ check netmiko_proxy init method """ proxy_minion_config["skip_connect_on_init"] = False mock_make_con = MagicMock() with patch.object(netmiko_proxy, "make_con", mock_make_con): assert netmiko_proxy.init(proxy_minion_config) is None assert "netmiko_device" in netmiko_proxy.__context__ netmiko_device = netmiko_proxy.__context__["netmiko_device"] assert "args" in netmiko_device assert "initialized" in netmiko_device assert netmiko_device["initialized"] assert "up" in netmiko_device assert netmiko_device["up"] assert "always_alive" in netmiko_device assert "connection" in netmiko_device
def test_alive(proxy_minion_config, subtests): """ check netmiko_proxy alive method """ # Always alive False with skip_connect_on_init on True # should return alive as True with subtests.test("skip_connect_on_init=True, proxy_always_alive=False"): proxy_minion_config["skip_connect_on_init"] = True proxy_minion_config["proxy_always_alive"] = False netmiko_proxy.init(proxy_minion_config) result = netmiko_proxy.alive(proxy_minion_config) assert result # Always alive True with skip_connect_on_init on True # should return alive as False with subtests.test("skip_connect_on_init=True, proxy_always_alive=True"): proxy_minion_config["skip_connect_on_init"] = True proxy_minion_config["proxy_always_alive"] = True netmiko_proxy.init(proxy_minion_config) result = netmiko_proxy.alive(proxy_minion_config) assert not result # Always alive True with skip_connect_on_init on False # should return alive as True with subtests.test("skip_connect_on_init=False, proxy_always_alive=True"): proxy_minion_config["skip_connect_on_init"] = False proxy_minion_config["proxy_always_alive"] = True mock_make_con = MagicMock() with patch.object(netmiko_proxy, "make_con", mock_make_con): netmiko_proxy.init(proxy_minion_config) result = netmiko_proxy.alive(proxy_minion_config) assert result