def test_success(self, init, time_):
inst = ZkSharedCache("zk", "path", "handler")
inst._path = "/path"
inst._kazoo = create_mock(["get"])
inst._kazoo.get.return_value = ("data", "meta")
inst._entries = create_mock(["setdefault"])
# Call
ntools.eq_(inst._get("name"), "data")
# Tests
inst._kazoo.get.assert_called_once_with("/path/name")
inst._entries.setdefault.assert_called_once_with(
"name", time_.return_value)
def test(self, init):
inst = ZkSharedCache("zk", "path", "handler")
entry_names = ["entry0", "entry1", "entry2", "entry3"]
inst._get = create_mock()
inst._get.side_effect = [
"data0", ZkNoNodeError, "data2", ZkNoConnection
]
inst._path = "/path"
inst._handler = create_mock()
# Call
ntools.eq_(inst._handle_entries(entry_names), 2)
# Tests
assert_these_calls(inst._get, ([call(i) for i in entry_names]))
inst._handler.assert_called_once_with(["data0", "data2"])