Python Command._broadcast_kill_requests 예제들

예제 #1

파일: test_broadcast_queue.py 프로젝트: brocaar/job-runner

    def test__broadcast_kill_requests(self):
        """
        Test :meth:`.Command._broadcast_kill_requests`.
        """
        worker = Worker.objects.get(pk=1)
        run = Run.objects.get(pk=1)
        run.start_dts = timezone.now()
        run.pid = 1234
        run.worker = worker
        run.save()
        KillRequest.objects.create(
            run=run,
            schedule_dts=timezone.now()
        )

        command = Command()

        command.publisher = Mock()
        command._broadcast_kill_requests()

        self.assertEqual([
            call([
                'master.broadcast.worker1',
                '{"action": "kill", "kill_request_id": 1}'
            ])
        ], command.publisher.send_multipart.call_args_list)

예제 #2

파일: test_broadcast_queue.py 프로젝트: lomignet/job-runner

    def test__broadcast_kill_requests(self):
        """
        Test :meth:`.Command._broadcast_kill_requests`.
        """
        worker = Worker.objects.get(pk=1)
        run = Run.objects.get(pk=1)
        run.start_dts = timezone.now()
        run.pid = 1234
        run.worker = worker
        run.save()
        KillRequest.objects.create(run=run, schedule_dts=timezone.now())

        command = Command()

        command.publisher = Mock()
        command._broadcast_kill_requests()

        self.assertEqual([
            call([
                'master.broadcast.worker1',
                '{"action": "kill", "kill_request_id": 1}'
            ])
        ], command.publisher.send_multipart.call_args_list)