Add tests for GroupingLatestValueCache

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

tests/test_grouping_latest_value_cache_integration.py

@@ -0,0 +1,70 @@
+# License: MIT
+# Copyright © 2024 Frequenz Energy-as-a-Service GmbH
+
+"""Tests for the LatestValueCache implementation."""
+
+import asyncio
+
+import pytest
+
+from frequenz.channels import Broadcast
+from frequenz.channels.experimental import GroupingLatestValueCache
+
+
+@pytest.mark.integration
+async def test_latest_value_cache_key() -> None:
+    """Ensure LatestValueCache works with keys."""
+    channel = Broadcast[tuple[int, str]](name="lvc_test")
+
+    cache: GroupingLatestValueCache[tuple[int, str], int] = GroupingLatestValueCache(
+        channel.new_receiver(), key=lambda x: x[0]
+    )
+    sender = channel.new_sender()
+
+    assert not cache.has_value(5)
+    with pytest.raises(ValueError, match="No value received for key: 0"):
+        cache.get(0)
+
+    assert cache.keys() == set()
+
+    await sender.send((5, "a"))
+    await sender.send((6, "b"))
+    await sender.send((5, "c"))
+    await asyncio.sleep(0)
+
+    assert cache.has_value(5)
+    assert cache.has_value(6)
+    assert not cache.has_value(7)
+
+    assert cache.get(5) == (5, "c")
+    assert cache.get(6) == (6, "b")
+
+    assert cache.keys() == {5, 6}
+
+    with pytest.raises(ValueError, match="No value received for key: 7"):
+        cache.get(7)
+
+    await sender.send((12, "d"))
+    await asyncio.sleep(0)
+
+    assert cache.get(12) == (12, "d")
+    assert cache.get(12) == (12, "d")
+    assert cache.get(5) == (5, "c")
+    assert cache.get(6) == (6, "b")
+
+    await sender.send((6, "e"))
+    await sender.send((6, "f"))
+    await sender.send((6, "g"))
+    await asyncio.sleep(0)
+
+    assert cache.get(6) == (6, "g")
+
+    assert cache.keys() == {5, 6, 12}
+
+    cache.clear(5)
+    assert not cache.has_value(5)
+    assert cache.has_value(6)
+
+    with pytest.raises(ValueError, match="No value received for key: 5"):
+        assert cache.get(5)
+    assert cache.keys() == {6, 12}