def test_lando_worker_client_and_router(self): """ Test to verify that messages sent from LandoWorkerClient through rabbit make it to LandoWorker """ queue_name = "lando_worker" lando_worker_client = LandoWorkerClient(self.config, queue_name) fake_lando_worker = FakeLandoWorker() router = MessageRouter.make_worker_router(self.config, fake_lando_worker, queue_name) fake_lando_worker.router = router lando_worker_client.stage_job(credentials=None, job_details=FakeJobDetails(1), input_files=[], vm_instance_name='test1') lando_worker_client.run_job(job_details=FakeJobDetails(2), workflow=FakeWorkflow(), vm_instance_name='test2') lando_worker_client.store_job_output(credentials=None, job_details=FakeJobDetails(3), vm_instance_name='test3') router.run()
def _make_router(self): work_queue_config = self.config.work_queue_config return MessageRouter.make_worker_router(self.config, self, work_queue_config.queue_name)