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