class ReplicationMiddlewareHttpSafeMethodsTest(TestCase): SETTINGS = dict( DATABASE_ROUTERS=['django_replicated2.routers.ReplicationRouter'], ) def setUp(self): self.middleware = ReplicationMiddleware() self.request = RequestFactory() self.router = flexmock().should_receive('enable_slaves').with_args().once flexmock(self.middleware).should_receive('get_router').and_return(self.router.mock) @override_settings(**SETTINGS) def test_get(self): self.middleware.process_request(self.request.get('/')) @override_settings(**SETTINGS) def test_head(self): self.middleware.process_request(self.request.head('/')) @override_settings(**SETTINGS) def test_options(self): self.middleware.process_request(self.request.options('/')) @override_settings(**SETTINGS) def test_trace(self): request = self.request.get('/') request.method = 'TRACE' self.middleware.process_request(request)
class ReplicationMiddlewareRecentlyUpdatedTest(TestCase): SETTINGS = dict( DATABASE_ROUTERS=['django_replicated2.routers.ReplicationRouter'], ) def setUp(self): self.middleware = ReplicationMiddleware() self.request = RequestFactory() @override_settings(**SETTINGS) def test_process_request(self): self.request.cookies[ReplicationMiddleware.COOKIE_NAME] = 'any-value' router = flexmock().should_receive('disable_slaves').with_args().once flexmock(self.middleware).should_receive('get_router').and_return(router.mock) self.middleware.process_request(self.request.get('/')) @override_settings(**SETTINGS) def test_process_response_updateable_methods(self): request = self.request.post('/') response = HttpResponseRedirect('/') flexmock(ReplicationRouter).should_receive('is_db_recently_updated').with_args().and_return(True) self.middleware.process_response(request, response) self.assertIn(ReplicationMiddleware.COOKIE_NAME, response.cookies) @override_settings(**SETTINGS) def test_process_response_safe_methods(self): request = self.request.get('/') response = HttpResponseRedirect('/') flexmock(ReplicationRouter).should_receive('is_db_recently_updated').with_args().and_return(True) self.middleware.process_response(request, response) self.assertNotIn(ReplicationMiddleware.COOKIE_NAME, response.cookies)
class ReplicationMiddlewareHttpMethodsTest(TestCase): SETTINGS = dict( DATABASE_ROUTERS=['django_replicated2.routers.ReplicationRouter'], ) def setUp(self): self.middleware = ReplicationMiddleware() self.request = RequestFactory() self.router = flexmock().should_receive('disable_slaves').with_args().once flexmock(self.middleware).should_receive('get_router').and_return(self.router.mock) @override_settings(**SETTINGS) def test_post(self): self.middleware.process_request(self.request.post('/')) @override_settings(**SETTINGS) def test_delete(self): self.middleware.process_request(self.request.delete('/')) @override_settings(**SETTINGS) def test_put(self): self.middleware.process_request(self.request.put('/'))
def setUp(self): self.middleware = ReplicationMiddleware() self.request = RequestFactory()
def setUp(self): self.middleware = ReplicationMiddleware() self.request = RequestFactory() self.router = flexmock().should_receive('disable_slaves').with_args().once flexmock(self.middleware).should_receive('get_router').and_return(self.router.mock)