def test_process_request(self):
with patch('kobo.hub.middleware.get_worker',
return_value=DummyWorker()) as get_worker_mock:
req = DummyRequest()
req.user = PropertyMock(username='******')
middleware.WorkerMiddleware().process_request(req)
self.assertIsInstance(req.worker, DummyWorker)
get_worker_mock.assert_called_once_with(req)
def test_process_request_missing_user(self):
req = DummyRequest()
with self.assertRaises(AssertionError):
middleware.WorkerMiddleware().process_request(req)