55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
"""RedisState against a real server, enabled by FLUKSIO_TEST_REDIS_HOST (and
|
|
optional PORT)."""
|
|
|
|
import os
|
|
import uuid
|
|
|
|
import pytest
|
|
|
|
from fluksio.flow.state import RedisState
|
|
|
|
|
|
@pytest.fixture
|
|
def state():
|
|
host = os.environ.get("FLUKSIO_TEST_REDIS_HOST")
|
|
if not host:
|
|
pytest.skip("Set FLUKSIO_TEST_REDIS_HOST to run Redis integration checks")
|
|
port = int(os.environ.get("FLUKSIO_TEST_REDIS_PORT", "6379"))
|
|
state = RedisState(host, port, namespace=f"test-state-{uuid.uuid4().hex}", ttl=600)
|
|
try:
|
|
yield state
|
|
finally:
|
|
state.clear()
|
|
state.close()
|
|
|
|
|
|
def test_a_ttl_expires_counters_and_the_names_set(state):
|
|
state.set("heat.temp", 21)
|
|
state.increment("__version__:heat.a")
|
|
state.increment_multi(["__version__:heat.b"])
|
|
state.record({"heat.v": 1}, {}, ["__version__:heat.c"], 0.0)
|
|
|
|
keys = state.keys()
|
|
assert "__names__" in keys and "__version__:heat.c" in keys
|
|
assert all(state._client.ttl(state._key(k)) > 0 for k in keys)
|
|
|
|
|
|
def test_a_compare_and_swap_names_what_it_wrote(state):
|
|
state.set("heat.a", 1)
|
|
assert state.compare_and_swap_multi({"heat.a": 1}, {"heat.b": 2})
|
|
assert "heat.b" in state.message_names()
|
|
|
|
|
|
def test_a_compare_and_swap_loses_to_a_write_after_its_read(state, monkeypatch):
|
|
state.set("heat.a", 1)
|
|
note = RedisState._note
|
|
|
|
def racing_note(self, pipe, keys):
|
|
# Another writer lands between the compare and EXEC.
|
|
self._client.set(self._key("heat.a"), b"2")
|
|
note(self, pipe, keys)
|
|
|
|
monkeypatch.setattr(RedisState, "_note", racing_note)
|
|
assert not state.compare_and_swap_multi({"heat.a": 1}, {"heat.a": 3})
|
|
assert state.get("heat.a") == 2
|