def test_get_agent_heartbeat_idle_interval_sec(self):
netutil = NetUtil.NetUtil(MagicMock())
heartbeat_interval = netutil.get_agent_heartbeat_idle_interval_sec(
1, 10, 32)
self.assertEqual(heartbeat_interval, 3)
def test_get_agent_heartbeat_idle_interval_sec_max(self):
netutil = NetUtil.NetUtil(MagicMock())
heartbeat_interval = netutil.get_agent_heartbeat_idle_interval_sec(
1, 10, 1500)
self.assertEqual(heartbeat_interval,
netutil.HEARTBEAT_IDLE_INTERVAL_DEFAULT_MAX_SEC)
def test_checkURL(self, httpsConMock, parseMock):
NetUtil.logger = MagicMock()
parseMock.return_value = [1, 2]
ca_connection = MagicMock()
response = MagicMock()
response.status = 200
ca_connection.getresponse.return_value = response
httpsConMock.return_value = ca_connection
# test 200
netutil = NetUtil.NetUtil()
self.assertTrue(netutil.checkURL("url")[0])
# test fail
response.status = 404
self.assertFalse(netutil.checkURL("url")[0])
# test Exception
response.status = 200
httpsConMock.side_effect = Exception("test")
self.assertFalse(netutil.checkURL("url")[0])
def test_try_to_connect(self, event_mock,
sleepMock):
event_mock.return_value = False
netutil = NetUtil.NetUtil()
checkURL = MagicMock(name="checkURL")
checkURL.return_value = True, "test"
netutil.checkURL = checkURL
# one successful get
self.assertEqual((0, True, False), netutil.try_to_connect("url", 10))
# got successful after N retries
gets = [[True, ""], [False, ""], [False, ""]]
def side_effect(*args):
return gets.pop()
checkURL.side_effect = side_effect
self.assertEqual((2, True, False), netutil.try_to_connect("url", 10))
# max retries
checkURL.side_effect = None
checkURL.return_value = False, "test"
self.assertEqual((5, False, False), netutil.try_to_connect("url", 5))
def test_try_to_connect(self, sleepMock):
netutil = NetUtil.NetUtil()
checkURL = MagicMock(name="checkURL")
checkURL.return_value = True
netutil.checkURL = checkURL
l = MagicMock()
# one successful get
self.assertEqual(0, netutil.try_to_connect("url", 10))
# got successful after N retries
gets = [True, False, False]
def side_effect(*args):
return gets.pop()
checkURL.side_effect = side_effect
self.assertEqual(2, netutil.try_to_connect("url", 10))
# max retries
checkURL.side_effect = None
checkURL.return_value = False
self.assertEqual(5, netutil.try_to_connect("url", 5))