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()