def test_writing_sets_db_written_flag(self): router = DBRouter() router.db_for_read(Submission) self.assertFalse(was_db_written()) router.db_for_write(Submission) self.assertTrue(was_db_written()) # A read won't reset the flag router.db_for_read(Submission) self.assertTrue(was_db_written())
def test_readwrite(self): self.assertFalse(was_db_written()) set_thread_readwrite_db() master = 'my_master_db' with override_settings(MASTER_DATABASE=master): router = DBRouter() out = router.db_for_read(Submission) self.assertEqual(master, out) out = router.db_for_write(Submission) self.assertEqual(master, out) self.assertTrue(was_db_written())