示例#1
0
 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"])
示例#2
0
 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"})