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)