def test_pinned_reads(self):
"""Test PinningMasterSlaveRouter.db_for_read() when pinned and when
not."""
router = PinningMasterSlaveRouter()
eq_(router.db_for_read(None), get_slave())
pin_this_thread()
eq_(router.db_for_read(None), DEFAULT_DB_ALIAS)
def test_db_write_decorator(self):
def read_view(req):
eq_(router.db_for_read(None), get_slave())
return HttpResponse()
@db_write
def write_view(req):
eq_(router.db_for_read(None), DEFAULT_DB_ALIAS)
return HttpResponse()
router = PinningMasterSlaveRouter()
eq_(router.db_for_read(None), get_slave())
write_view(HttpRequest())
read_view(HttpRequest())