def test_get_replica(self, mock_randint): mock_randint.return_value = 0 self.assertEqual(pindb.get_replica("default"), "default-0") mock_randint.return_value = 1 self.assertEqual(pindb.get_replica("default"), "default-0") # gets the master if there are no replicas self.assertEqual(pindb.get_replica("egg"), "egg") self.assertEqual(pindb.get_replica("frob"), "frob")
def test_get_replica(self, mock_randint): mock_randint.return_value = 0 self.assertEqual(pindb.get_replica("default"), "default") mock_randint.return_value = 1 self.assertEqual(pindb.get_replica("default"), "default") # gets the master if there are no replicas mock_randint.return_value = 0 self.assertEqual(pindb.get_replica("egg"), "egg-0") mock_randint.return_value = 1 self.assertEqual(pindb.get_replica("egg"), "egg-0") # nonexistent or unmanaged DATABASES should return the alias self.assertEqual(pindb.get_replica("frob"), "frob") self.assertEqual(pindb.get_replica("nope"), "nope")