def test_restartAgent(self, _clean_mock, execMock): ProcessHelper.logger = MagicMock() ProcessHelper.restartAgent() self.assertTrue(_clean_mock.called) self.assertTrue(execMock.called) self.assertEqual(2, len(execMock.call_args_list[0]))
def test_clean(self, getTempFilesMock): tf1 = tempfile.NamedTemporaryFile(delete=False) tf2 = tempfile.NamedTemporaryFile(delete=False) tf3 = tempfile.NamedTemporaryFile(delete=False) getTempFilesMock.return_value = [tf2.name, tf3.name] ProcessHelper.pidfile = tf1.name ProcessHelper.logger = MagicMock() ProcessHelper._clean() self.assertFalse(os.path.exists(tf1.name)) self.assertFalse(os.path.exists(tf2.name)) self.assertFalse(os.path.exists(tf3.name))
def test_stopAgent(self, _clean_mock, exitMock): ProcessHelper.stopAgent() self.assertTrue(_clean_mock.called) self.assertTrue(exitMock.called)