def test_notifies_watchers_when_children_are_added(self): mock_zk = _mock_zk() mock_callback = Mock(name='monitor callback') def first_callback(*args, **kwargs): group.monitor(frozenset([Membership(1)]), mock_callback) group = ActiveKazooGroup(mock_zk, DEFAULT_PATH) group.monitor(callback=first_callback) members, _ = _happy_path(group, mock_zk, 1) mock_callback.assert_called_with(members)
def test_notifies_watchers_when_children_are_added(self): mock_zk = _mock_zk() mock_callback = Mock(name="monitor callback") def first_callback(*args, **kwargs): group.monitor(frozenset([Membership(1)]), mock_callback) group = ActiveKazooGroup(mock_zk, DEFAULT_PATH) group.monitor(callback=first_callback) members, _ = _happy_path(group, mock_zk, 1) mock_callback.assert_called_with(members)
def test_notifies_watchers_when_children_are_removed(self): mock_zk = _mock_zk() mock_callback = Mock(name='monitor callback') def devnull(*args, **kwargs): pass group = ActiveKazooGroup(mock_zk, DEFAULT_PATH) group.monitor(frozenset([]), mock_callback) members, completion_callback = _happy_path(group, mock_zk, 2) member_remaining, member_removed = list(members) znode_member_id = ActiveKazooGroup.id_to_znode(member_remaining.id) mock_get_children_result = Mock(name='mock get children async result') mock_get_children_result.get = Mock(return_value=[znode_member_id]) completion_callback(mock_get_children_result) assert member_removed not in group._members mock_callback.assert_called_with(set([member_remaining]))
def test_notifies_watchers_when_children_are_removed(self): mock_zk = _mock_zk() mock_callback = Mock(name="monitor callback") def devnull(*args, **kwargs): pass group = ActiveKazooGroup(mock_zk, DEFAULT_PATH) group.monitor(frozenset([]), mock_callback) members, completion_callback = _happy_path(group, mock_zk, 2) member_remaining, member_removed = list(members) znode_member_id = ActiveKazooGroup.id_to_znode(member_remaining.id) mock_get_children_result = Mock(name="mock get children async result") mock_get_children_result.get = Mock(return_value=[znode_member_id]) completion_callback(mock_get_children_result) assert member_removed not in group._members mock_callback.assert_called_with(set([member_remaining]))