def test_listen_to_pubsub(self, mc_retask_queue, mc_grc):
mc_rc = MagicMock()
mc_grc.return_value = mc_rc
self.jg = CoprJobGrab(self.opts)
assert not mc_rc.pubsub.called
self.jg.listen_to_pubsub()
assert mc_rc.pubsub.called
assert mc_rc.pubsub.return_value.subscribe.called
def test_connect_queues(self, mc_retask_queue):
self.jg = CoprJobGrab(self.opts, self.queue, self.lock)
assert len(self.jg.task_queues_by_arch) == 0
self.jg.connect_queues()
# created retask queue
expected = [call(u'copr-be-0'), call(u'copr-be-1')]
assert mc_retask_queue.call_args_list == expected
# connected to them
for obj in self.jg.task_queues_by_arch.values():
assert obj.connect.called
def test_connect_queues(self, mc_retask_queue, mc_grc):
mc_rc = MagicMock()
mc_grc.return_value = mc_rc
self.jg = CoprJobGrab(self.opts, self.frontend_client)
assert len(self.jg.task_queues_by_arch) == 0
self.jg.connect_queues()
# created retask queue
expected = [call(u'copr-be-0'), call(u'copr-be-1')]
assert mc_retask_queue.call_args_list == expected
# connected to them
for obj in self.jg.task_queues_by_arch.values():
assert obj.connect.called
assert not mc_rc.pubsub.called
def init_jg(self, mc_retask_queue, mc_grc):
self.jg = CoprJobGrab(self.opts)
self.jg.init_internal_structures()
self.jg.jg_control = MagicMock()
def init_jg(self, mc_retask_queue):
self.jg = CoprJobGrab(self.opts, self.queue, self.lock)
self.jg.connect_queues()
def init_jg(self, mc_retask_queue, mc_grc):
self.jg = CoprJobGrab(self.opts, self.frontend_client)
self.jg.connect_queues()
self.jg.vm_manager = MagicMock()
def main():
opts = get_backend_opts()
grabber = CoprJobGrab(opts)
grabber.run()
def main():
opts = get_backend_opts()
fc = FrontendClient(opts)
grabber = CoprJobGrab(opts, frontend_client=fc)
grabber.run()
