def test_process_request_positive(self): """ Tests positive case of EzSecurityMiddleware with right headers. """ # setup m_request = self.__set_headers(mock.MagicMock(spec=HttpRequest)) middleware = EzSecurityMiddleware() middleware.ez_security_client = mock.Mock() middleware.ez_security_client.validateCurrentRequest.return_value = \ True response = middleware.process_request(m_request) # test self.assertIsNone(response)
def test_process_request_negative(self): """ Tests negative case of EzSecurityMiddleware with wrong headers. """ # setup # settings = mock.Mock() with mock.patch('ezbake.support.django.middleware.HttpResponseForbidden') as m_HttpResponseForbidden: m_request = self.__set_headers(mock.MagicMock(spec=HttpRequest)) middleware = EzSecurityMiddleware() middleware.ez_security_client = mock.Mock() middleware.ez_security_client.validate_current_request.return_value = \ False response = middleware.process_request(m_request) # test self.assertEqual(response, m_HttpResponseForbidden())