def test_booted(self): from job_dictator import JobDictator from job import Job from worker import Worker dictator = JobDictator() dictator.client = mock.MagicMock() dictator.client.keys.return_value = ["job-", "jm-"] job = Job("booted", "something") worker = Worker("job-", None) dictator.client.get.side_effect = [pickle.dumps(job), pickle.dumps(worker)] self.request_mock.get = mock.MagicMock() dictator.settings = mock.MagicMock() dictator.headers = mock.MagicMock() returner = mock.MagicMock() returner.content = "status:ok" self.request_mock.get.return_value = returner dictator.push = mock.MagicMock() dictator.aladeen() assert dictator.client.keys.call_count == 2 assert dictator.client.get.call_count == 2 assert dictator.push.call_count == 1
def test_booted(self): from job_dictator import JobDictator from job import Job from worker import Worker dictator = JobDictator() dictator.client = mock.MagicMock() dictator.client.keys.return_value = ['job-', 'jm-'] job = Job('booted', 'something') worker = Worker('job-', None) dictator.client.get.side_effect = [pickle.dumps(job), pickle.dumps(worker)] self.request_mock.get = mock.MagicMock() dictator.settings = mock.MagicMock() dictator.headers = mock.MagicMock() returner = mock.MagicMock() returner.content = 'status:ok' self.request_mock.get.return_value = returner dictator.push = mock.MagicMock() dictator.aladeen() assert dictator.client.keys.call_count == 2 assert dictator.client.get.call_count == 2 assert dictator.push.call_count == 1