def test_no_cache(self, init):
inst = ZkSharedCache("zk", "path", "handler")
inst._path = "/path"
inst._kazoo = create_mock(["get_children"])
inst._kazoo.get_children.side_effect = NoNodeError
# Call
ntools.eq_(inst._list_entries(), set())
def test_sucesss(self, init):
inst = ZkSharedCache("zk", "path", "handler")
inst._path = "/path"
inst._kazoo = create_mock(["get_children"])
inst._kazoo.get_children.return_value = [
"node0", "node1", "node2", "node3"
]
# Call
ntools.eq_(inst._list_entries(), {"node0", "node1", "node2", "node3"})
def test_full(self, init):
inst = ZkSharedCache("zk", "path", "handler")
inst._zk = create_mock(["conn_epoch", "is_connected"])
inst._incoming_entries = create_mock(["__bool__", "popleft"])
inst._incoming_entries.__bool__.side_effect = True, True, False
inst._incoming_entries.popleft.side_effect = ("inc0", 1), ("inc1", 0)
inst._entries = {"inc0": 0, "old0": 0}
inst._list_entries = create_mock()
inst._list_entries.return_value = "inc0", "inc1", "new0"
inst._handle_entries = create_mock()
inst._path = "/path"
# Call
inst.process()
# Tests
ntools.eq_(inst._entries, {"inc0": 0, "inc1": 0})
inst._handle_entries.assert_called_once_with({"new0"})