def test_genAgentCrtReq(self, communicate_mock, popen_mock):
man = CertificateManager(self.config)
p = MagicMock(spec=subprocess.Popen)
p.communicate = communicate_mock
popen_mock.return_value = p
man.genAgentCrtReq()
self.assertTrue(popen_mock.called)
self.assertTrue(communicate_mock.called)
def test_genAgentCrtReq(self, communicate_mock, popen_mock): man = CertificateManager(self.config) p = MagicMock(spec=subprocess.Popen) p.communicate = communicate_mock popen_mock.return_value = p man.genAgentCrtReq() self.assertTrue(popen_mock.called) self.assertTrue(communicate_mock.called)
def test_genAgentCrtReq(self, chmod_mock, communicate_mock, popen_mock):
man = CertificateManager(self.config, "active_server")
p = MagicMock(spec=subprocess32.Popen)
p.communicate = communicate_mock
popen_mock.return_value = p
man.genAgentCrtReq('/dummy-keysdir/hostname.key')
self.assertTrue(chmod_mock.called)
self.assertTrue(popen_mock.called)
self.assertTrue(communicate_mock.called)
def test_init_warn(self, socketListenMock, socketBindMock, popen_mock):
procObj = MagicMock()
procObj.communicate = MagicMock()
procObj.communicate.return_value = {"mine.py": 0, "log": "log"}
popen_mock.return_value = procObj
PingPortListener.logger.reset_mock()
try:
PingPortListener.PingPortListener(self.config)
self.fail("Should throw exception")
except Exception as fe:
# Expected
self.assertTrue("port already used" in str(fe))
pass
def test_init_warn(self,socketListenMock,socketBindMock,popen_mock):
procObj = MagicMock()
procObj.communicate = MagicMock()
procObj.communicate.return_value = {"mine.py": 0, "log": "log"}
popen_mock.return_value = procObj
PingPortListener.logger.reset_mock()
try:
PingPortListener.PingPortListener(self.config)
self.fail("Should throw exception")
except Exception as fe:
# Expected
self.assertEqual(1, procObj.communicate.call_count)
pass
def test_init_success(self,socketMock,popen_mock):
procObj = MagicMock()
procObj.communicate = MagicMock()
procObj.communicate.return_value = {"": 0, "log": "log"}
popen_mock.return_value = procObj
PingPortListener.logger.reset_mock()
popen_mock.reset_mock()
allive_daemon = PingPortListener.PingPortListener(self.config)
self.assertTrue(popen_mock.called)
self.assertFalse(PingPortListener.logger.warn.called)
self.assertTrue(socketMock.call_args_list[0][0][0] == socket.AF_INET)
self.assertTrue(socketMock.call_args_list[0][0][1] == socket.SOCK_STREAM)
self.assertTrue(allive_daemon.socket.bind.call_args_list[0][0][0] == ('0.0.0.0',55000))
self.assertTrue(allive_daemon.socket.listen.call_args_list[0][0][0] == 1)
self.assertTrue(allive_daemon.config.set.call_args_list[0][0][0] == 'agent')
self.assertTrue(allive_daemon.config.set.call_args_list[0][0][1] == 'current_ping_port')
def test_init_success(self, socketMock, popen_mock):
procObj = MagicMock()
procObj.communicate = MagicMock()
procObj.communicate.return_value = {"": 0, "log": "log"}
popen_mock.return_value = procObj
PingPortListener.logger.reset_mock()
popen_mock.reset_mock()
allive_daemon = PingPortListener.PingPortListener(self.config)
self.assertTrue(popen_mock.called)
self.assertFalse(PingPortListener.logger.warn.called)
self.assertTrue(socketMock.call_args_list[0][0][0] == socket.AF_INET)
self.assertTrue(
socketMock.call_args_list[0][0][1] == socket.SOCK_STREAM)
self.assertTrue(
allive_daemon.socket.bind.call_args_list[0][0][0] == ('0.0.0.0',
55000))
self.assertTrue(
allive_daemon.socket.listen.call_args_list[0][0][0] == 1)
self.assertTrue(
allive_daemon.config.set.call_args_list[0][0][0] == 'agent')
self.assertTrue(allive_daemon.config.set.call_args_list[0][0][1] ==
'current_ping_port')
