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
