mirror of
https://github.com/dragonflydb/dragonfly.git
synced 2024-12-15 17:51:06 +00:00
241 lines
8.2 KiB
Python
241 lines
8.2 KiB
Python
import pathlib
|
|
import subprocess
|
|
from typing import Awaitable
|
|
from redis import asyncio as aioredis
|
|
import pytest
|
|
import time
|
|
import asyncio
|
|
from datetime import datetime
|
|
from sys import stderr
|
|
import logging
|
|
|
|
|
|
# Helper function to parse some sentinel cli commands output as key value dictionaries.
|
|
# Output is expected be of even number of lines where each pair of consecutive lines results in a single key value pair.
|
|
# If new_dict_key is not empty, encountering it in the output will start a new dictionary, this let us return multiple
|
|
# dictionaries, for example in the 'slaves' command, one dictionary for each slave.
|
|
def stdout_as_list_of_dicts(cp: subprocess.CompletedProcess, new_dict_key=""):
|
|
lines = cp.stdout.splitlines()
|
|
res = []
|
|
d = None
|
|
if new_dict_key == "":
|
|
d = dict()
|
|
res.append(d)
|
|
for i in range(0, len(lines), 2):
|
|
if (lines[i]) == new_dict_key: # assumes output never has '' as a key
|
|
d = dict()
|
|
res.append(d)
|
|
d[lines[i]] = lines[i + 1]
|
|
return res
|
|
|
|
|
|
def wait_for(func, pred, timeout_sec, timeout_msg=""):
|
|
while not pred(func()):
|
|
assert timeout_sec > 0, timeout_msg
|
|
timeout_sec = timeout_sec - 1
|
|
time.sleep(1)
|
|
|
|
|
|
async def await_for(func, pred, timeout_sec, timeout_msg=""):
|
|
done = False
|
|
while not done:
|
|
val = func()
|
|
if isinstance(val, Awaitable):
|
|
val = await val
|
|
done = pred(val)
|
|
assert timeout_sec > 0, timeout_msg
|
|
timeout_sec = timeout_sec - 1
|
|
await asyncio.sleep(1)
|
|
|
|
|
|
class Sentinel:
|
|
def __init__(self, port, master_port, config_dir) -> None:
|
|
self.config_file = pathlib.Path(config_dir).joinpath("sentinel.conf")
|
|
self.port = port
|
|
self.image = "bitnami/redis-sentinel:latest"
|
|
self.container_name = "sentinel_test_py_sentinel"
|
|
self.default_deployment = "my_deployment"
|
|
self.initial_master_port = master_port
|
|
self.proc = None
|
|
|
|
def start(self):
|
|
config = [
|
|
f"port {self.port}",
|
|
f"sentinel monitor {self.default_deployment} 127.0.0.1 {self.initial_master_port} 1",
|
|
f"sentinel down-after-milliseconds {self.default_deployment} 3000",
|
|
]
|
|
self.config_file.write_text("\n".join(config))
|
|
|
|
logging.info(self.config_file.read_text())
|
|
|
|
self.proc = subprocess.Popen(
|
|
["redis-server", f"{self.config_file.absolute()}", "--sentinel"]
|
|
)
|
|
|
|
def stop(self):
|
|
self.proc.terminate()
|
|
self.proc.wait(timeout=10)
|
|
|
|
def run_cmd(
|
|
self, args, sentinel_cmd=True, capture_output=False, assert_ok=True
|
|
) -> subprocess.CompletedProcess:
|
|
run_args = ["redis-cli", "-p", f"{self.port}"]
|
|
if sentinel_cmd:
|
|
run_args = run_args + ["sentinel"]
|
|
run_args = run_args + args
|
|
cp = subprocess.run(run_args, capture_output=capture_output, text=True)
|
|
if assert_ok:
|
|
assert cp.returncode == 0, f"Command failed: {run_args}"
|
|
return cp
|
|
|
|
def wait_ready(self):
|
|
wait_for(
|
|
lambda: self.run_cmd(["ping"], sentinel_cmd=False, assert_ok=False),
|
|
lambda cp: cp.returncode == 0,
|
|
timeout_sec=10,
|
|
timeout_msg="Timeout waiting for sentinel to become ready.",
|
|
)
|
|
|
|
def master(self, deployment="") -> dict:
|
|
if deployment == "":
|
|
deployment = self.default_deployment
|
|
cp = self.run_cmd(["master", deployment], capture_output=True)
|
|
return stdout_as_list_of_dicts(cp)[0]
|
|
|
|
def slaves(self, deployment="") -> dict:
|
|
if deployment == "":
|
|
deployment = self.default_deployment
|
|
cp = self.run_cmd(["slaves", deployment], capture_output=True)
|
|
return stdout_as_list_of_dicts(cp)
|
|
|
|
def live_master_port(self, deployment=""):
|
|
if deployment == "":
|
|
deployment = self.default_deployment
|
|
cp = self.run_cmd(["get-master-addr-by-name", deployment], capture_output=True)
|
|
return int(cp.stdout.splitlines()[1])
|
|
|
|
def failover(self, deployment=""):
|
|
if deployment == "":
|
|
deployment = self.default_deployment
|
|
self.run_cmd(
|
|
[
|
|
"failover",
|
|
deployment,
|
|
]
|
|
)
|
|
|
|
|
|
@pytest.fixture(
|
|
scope="function"
|
|
) # Sentinel has state which we don't want carried over form test to test.
|
|
def sentinel(tmp_dir, port_picker) -> Sentinel:
|
|
s = Sentinel(port_picker.get_available_port(), port_picker.get_available_port(), tmp_dir)
|
|
s.start()
|
|
s.wait_ready()
|
|
yield s
|
|
s.stop()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.slow
|
|
async def test_failover(df_local_factory, sentinel, port_picker):
|
|
master = df_local_factory.create(port=sentinel.initial_master_port)
|
|
replica = df_local_factory.create(port=port_picker.get_available_port())
|
|
|
|
master.start()
|
|
replica.start()
|
|
|
|
master_client = aioredis.Redis(port=master.port)
|
|
replica_client = aioredis.Redis(port=replica.port)
|
|
logging.info("master: " + str(master.port) + " replica: " + str(replica.port))
|
|
|
|
await replica_client.execute_command("REPLICAOF localhost " + str(master.port))
|
|
|
|
assert sentinel.live_master_port() == master.port
|
|
|
|
# Verify sentinel picked up replica.
|
|
await await_for(
|
|
lambda: sentinel.master(),
|
|
lambda m: m["num-slaves"] == "1",
|
|
timeout_sec=15,
|
|
timeout_msg="Timeout waiting for sentinel to pick up replica.",
|
|
)
|
|
|
|
sentinel.failover()
|
|
|
|
# Verify sentinel switched.
|
|
await await_for(
|
|
lambda: sentinel.live_master_port(),
|
|
lambda p: p == replica.port,
|
|
timeout_sec=10,
|
|
timeout_msg="Timeout waiting for sentinel to report replica as master.",
|
|
)
|
|
assert sentinel.slaves()[0]["port"] == str(master.port)
|
|
|
|
# Verify we can now write to replica and read replicated value from master.
|
|
assert await replica_client.set("key", "value"), "Failed to set key on promoted replica."
|
|
logging.info("key was set on promoted replica, awaiting get on promoted replica.")
|
|
try:
|
|
await await_for(
|
|
lambda: master_client.get("key"),
|
|
lambda val: val == b"value",
|
|
15,
|
|
"Timeout waiting for key to exist in replica.",
|
|
)
|
|
except AssertionError:
|
|
syncid, r_offset = await master_client.execute_command("DEBUG REPLICA OFFSET")
|
|
replicaoffset_cmd = "DFLY REPLICAOFFSET " + syncid.decode()
|
|
m_offset = await replica_client.execute_command(replicaoffset_cmd)
|
|
logging.info(f"{syncid.decode()} {r_offset} {m_offset}")
|
|
logging.info("replica client role:")
|
|
logging.info(await replica_client.execute_command("role"))
|
|
logging.info("master client role:")
|
|
logging.info(await master_client.execute_command("role"))
|
|
logging.info("replica client info:")
|
|
logging.info(await replica_client.info())
|
|
logging.info("master client info:")
|
|
logging.info(await master_client.info())
|
|
replica_val = await replica_client.get("key")
|
|
master_val = await master_client.get("key")
|
|
logging.info(f"replica val: {replica_val}")
|
|
logging.info(f"master val: {master_val}")
|
|
raise
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.slow
|
|
async def test_master_failure(df_local_factory, sentinel, port_picker):
|
|
master = df_local_factory.create(port=sentinel.initial_master_port)
|
|
replica = df_local_factory.create(port=port_picker.get_available_port())
|
|
|
|
master.start()
|
|
replica.start()
|
|
|
|
replica_client = aioredis.Redis(port=replica.port)
|
|
|
|
await replica_client.execute_command("REPLICAOF localhost " + str(master.port))
|
|
|
|
assert sentinel.live_master_port() == master.port
|
|
|
|
# Verify sentinel picked up replica.
|
|
await await_for(
|
|
lambda: sentinel.master(),
|
|
lambda m: m["num-slaves"] == "1",
|
|
timeout_sec=15,
|
|
timeout_msg="Timeout waiting for sentinel to pick up replica.",
|
|
)
|
|
|
|
# Simulate master failure.
|
|
master.stop()
|
|
|
|
# Verify replica pormoted.
|
|
await await_for(
|
|
lambda: sentinel.live_master_port(),
|
|
lambda p: p == replica.port,
|
|
timeout_sec=300,
|
|
timeout_msg="Timeout waiting for sentinel to report replica as master.",
|
|
)
|
|
|
|
# Verify we can now write to replica.
|
|
await replica_client.set("key", "value")
|
|
assert await replica_client.get("key") == b"value"
|