def test_load_balance(self): abcd = ['a', 'b', 'c', 'd'] key = 'my-load-balancing-test-key' for i in range(3): self.assertEqual(load_balance(key, abcd), 'a') self.assertEqual(load_balance(key, abcd), 'b') self.assertEqual(load_balance(key, abcd), 'c') self.assertEqual(load_balance(key, abcd), 'd')
def test_reset_key(self): key = 'my-load-balancing-reset-test-key' client = get_redis_client().client.get_client() client.set(key, 999998) load_balance(key, [1]) self.assertEqual(int(client.get(key)), 999999) load_balance(key, [1]) self.assertIsNone(client.get(key)) load_balance(key, [1]) self.assertEqual(int(client.get(key)), 1)