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_pinned_reads(self): """Test PinningMasterSlaveRouter.db_for_read() when pinned and when not.""" router = PinningMasterSlaveRouter() eq_(router.db_for_read(TestModel), get_slave()) pin_this_thread() eq_(router.db_for_read(TestModel), MASTER_DATABASE)
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())
def test_db_write_decorator(self): def read_view(req): eq_(router.db_for_read(TestModel), get_slave()) return HttpResponse() @db_write def write_view(req): eq_(router.db_for_read(TestModel), MASTER_DATABASE) return HttpResponse() router = PinningMasterSlaveRouter() eq_(router.db_for_read(TestModel), get_slave()) write_view(HttpRequest()) read_view(HttpRequest())