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"])