繁体   English   中英

在视图中测试和模拟线程函数

[英]Testing and mocking a threaded function within a view

我试图对在视图中运行线程的功能进行单元测试。 每当我尝试模拟它时,它总是转到原始函数,而不是模拟函数。

我正在从视图模块测试的代码:

def restart_process(request):
    batch_name = request.POST.get("batch_name", "")
    if batch_name:
        try:
            batch = models.Batch.objects.get(num=batch_name)
        except models.Batch.DoesNotExist:
            logger.warning("Trying to restart a batch that does not exist: " + batch_name)
            return HttpResponse(404)
        else:
            logger.info(batch_name + " restarted")
            try:
                t = threading.Thread(target=restart_from_last_completed_state, args=(batch,))
                t.daemon = True
                t.start()
            except RuntimeError:
                return HttpResponse(500, "Threading error")
            return HttpResponse(200)
    else:
        return HttpResponse(400)

测试功能:

class ThreadTestCases(TransactionTestCase):

def test_restart_process(self):
    client = Client()
    mock_restart_from_last_completed_state = mock.Mock()
    with mock.patch("processapp.views.restart_from_last_completed_state", mock_restart_from_last_completed_state):
        response = client.post('/batch/restart/', {"batch_name": "BATCH555"})
        self.assertEqual(response.status_code, 200)
    mock_restart_from_last_completed_state.assert_called_once()

网址:

url(r'^batch/restart/$', views.restart_from_last_completed_state, name="restart_batch"),

我总是收到此错误:

ValueError: The view processapp.processing.process_runner.restart_from_last_completed_state didn't return an HttpResponse object. It returned None instead.

我在原始函数(restart_from_last_completed_state)中放置了一条打印命令,该命令始终运行,因此不会发生模拟。

该错误似乎将函数视为视图,尽管事实并非如此。

我不确定错误在哪里,线程,测试等等。

URL变量错误。 应该是views.restart_process而不是views.restart_from_last_completed_state

多次复制/粘贴错误...

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM