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')