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)
