Python DBRouter.db_for_write Exemples

Exemple #1

 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())

Exemple #2

 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())