def testProcessInfo(self, process_iter_mock): def side_effect_running(): return 'running' class Proc: def status(self): return 'some_status' p1 = Proc() p1.status = side_effect_running p2 = Proc() p2.status = side_effect_running p3 = Proc() p4 = Proc() processes = [p1, p2, p3, p4] process_iter_mock.return_value = processes hostinfo = HostInfo(MagicMock()) procs = hostinfo.get_process_info() self.assertEqual(procs['proc_run'], 2) self.assertEqual(procs['proc_total'], len(processes))
def testProcessInfo(self, process_iter_mock): def side_effect_running(): return 'running' class Proc: def status(self): return 'some_status' p1 = Proc() p1.status = side_effect_running p2 = Proc() p2.status = side_effect_running p3 = Proc() p4 = Proc() processes = [p1, p2, p3, p4] process_iter_mock.return_value = processes hostinfo = HostInfo(MagicMock()) procs = hostinfo.get_process_info() self.assertEqual(procs['proc_run'], 2) self.assertEqual(procs['proc_total'], len(processes))