You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@lucene.apache.org by ab...@apache.org on 2018/04/23 17:35:08 UTC

[04/40] lucene-solr:jira/solr-11833: SOLR-11924: Added CloudCollectionsListener to watch the list of collections in a cloud. This closes #313

SOLR-11924: Added CloudCollectionsListener to watch the list of collections in a cloud. This closes #313


Project: http://git-wip-us.apache.org/repos/asf/lucene-solr/repo
Commit: http://git-wip-us.apache.org/repos/asf/lucene-solr/commit/ae0190b6
Tree: http://git-wip-us.apache.org/repos/asf/lucene-solr/tree/ae0190b6
Diff: http://git-wip-us.apache.org/repos/asf/lucene-solr/diff/ae0190b6

Branch: refs/heads/jira/solr-11833
Commit: ae0190b696396bc2fc4d239a22d568c8438b8c4f
Parents: d904112
Author: Houston Putman <hp...@bloomberg.net>
Authored: Tue Apr 17 13:43:53 2018 +0000
Committer: Dennis Gove <dp...@gmail.com>
Committed: Tue Apr 17 18:57:04 2018 -0400

----------------------------------------------------------------------
 .../common/cloud/CloudCollectionsListener.java  |  40 +++
 .../apache/solr/common/cloud/ZkStateReader.java |  52 +++-
 .../cloud/TestCloudCollectionsListeners.java    | 311 +++++++++++++++++++
 3 files changed, 402 insertions(+), 1 deletion(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/lucene-solr/blob/ae0190b6/solr/solrj/src/java/org/apache/solr/common/cloud/CloudCollectionsListener.java
----------------------------------------------------------------------
diff --git a/solr/solrj/src/java/org/apache/solr/common/cloud/CloudCollectionsListener.java b/solr/solrj/src/java/org/apache/solr/common/cloud/CloudCollectionsListener.java
new file mode 100644
index 0000000..9920e59
--- /dev/null
+++ b/solr/solrj/src/java/org/apache/solr/common/cloud/CloudCollectionsListener.java
@@ -0,0 +1,40 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.solr.common.cloud;
+
+import java.util.Set;
+
+/**
+ * Callback registered with {@link ZkStateReader#registerCloudCollectionsListener(CloudCollectionsListener)}
+ * and called whenever the cloud's set of collections changes.
+ */
+public interface CloudCollectionsListener {
+
+  /**
+   * Called when a collection is created, a collection is deleted or a watched collection's state changes.
+   *
+   * Note that, due to the way Zookeeper watchers are implemented, a single call may be
+   * the result of multiple or no collection creation or deletions. Also, multiple calls to this method can be made
+   * with the same set of collections, ie. without any new updates.
+   *
+   * @param oldCollections       the previous set of collections
+   * @param newCollections       the new set of collections
+   */
+  void onChange(Set<String> oldCollections, Set<String> newCollections);
+
+}

http://git-wip-us.apache.org/repos/asf/lucene-solr/blob/ae0190b6/solr/solrj/src/java/org/apache/solr/common/cloud/ZkStateReader.java
----------------------------------------------------------------------
diff --git a/solr/solrj/src/java/org/apache/solr/common/cloud/ZkStateReader.java b/solr/solrj/src/java/org/apache/solr/common/cloud/ZkStateReader.java
index 7d5401d..a73e4c1 100644
--- a/solr/solrj/src/java/org/apache/solr/common/cloud/ZkStateReader.java
+++ b/solr/solrj/src/java/org/apache/solr/common/cloud/ZkStateReader.java
@@ -171,6 +171,8 @@ public class ZkStateReader implements Closeable {
 
   private ConcurrentHashMap<String, CollectionWatch<CollectionPropsWatcher>> collectionPropsWatches = new ConcurrentHashMap<>();
 
+  private Set<CloudCollectionsListener> cloudCollectionsListeners = ConcurrentHashMap.newKeySet();
+
   private final ExecutorService notifications = ExecutorUtil.newMDCAwareCachedThreadPool("watches");
   
   /** Used to submit notifications to Collection Properties watchers in order **/
@@ -545,6 +547,8 @@ public class ZkStateReader implements Closeable {
           clusterState.getCollectionStates());
     }
 
+    notifyCloudCollectionsListeners();
+
     for (String collection : changedCollections) {
       notifyStateWatchers(liveNodes, collection, clusterState.getCollectionOrNull(collection));
     }
@@ -650,6 +654,52 @@ public class ZkStateReader implements Closeable {
     }
   }
 
+  // We don't get a Stat or track versions on getChildren() calls, so force linearization.
+  private final Object refreshCollectionsSetLock = new Object();
+  // Ensures that only the latest getChildren fetch gets applied.
+  private final AtomicReference<Set<String>> lastFetchedCollectionSet = new AtomicReference<>();
+
+  /**
+   * Register a CloudCollectionsListener to be called when the set of collections within a cloud changes.
+   */
+  public void registerCloudCollectionsListener(CloudCollectionsListener cloudCollectionsListener) {
+    cloudCollectionsListeners.add(cloudCollectionsListener);
+    notifyNewCloudCollectionsListener(cloudCollectionsListener);
+  }
+
+  /**
+   * Remove a registered CloudCollectionsListener.
+   */
+  public void removeCloudCollectionsListener(CloudCollectionsListener cloudCollectionsListener) {
+    cloudCollectionsListeners.remove(cloudCollectionsListener);
+  }
+
+  private void notifyNewCloudCollectionsListener(CloudCollectionsListener listener) {
+    listener.onChange(Collections.emptySet(), lastFetchedCollectionSet.get());
+  }
+
+  private void notifyCloudCollectionsListeners() {
+    notifyCloudCollectionsListeners(false);
+  }
+
+  private void notifyCloudCollectionsListeners(boolean notifyIfSame) {
+    synchronized (refreshCollectionsSetLock) {
+      final Set<String> newCollections = getCurrentCollections();
+      final Set<String> oldCollections = lastFetchedCollectionSet.getAndSet(newCollections);
+      if (!newCollections.equals(oldCollections) || notifyIfSame) {
+        cloudCollectionsListeners.forEach(listener -> listener.onChange(oldCollections, newCollections));
+      }
+    }
+  }
+
+  private Set<String> getCurrentCollections() {
+    Set<String> collections = new HashSet<>();
+    collections.addAll(legacyCollectionStates.keySet());
+    collections.addAll(watchedCollectionStates.keySet());
+    collections.addAll(lazyCollectionStates.keySet());
+    return collections;
+  }
+
   private class LazyCollectionRef extends ClusterState.CollectionRef {
     private final String collName;
     private long lastUpdateTime;
@@ -1364,7 +1414,7 @@ public class ZkStateReader implements Closeable {
     if (watchSet.get()) {
       new StateWatcher(collection).refreshAndWatch();
     }
-    
+
     DocCollection state = clusterState.getCollectionOrNull(collection);
     if (stateWatcher.onStateChanged(liveNodes, state) == true) {
       removeCollectionStateWatcher(collection, stateWatcher);

http://git-wip-us.apache.org/repos/asf/lucene-solr/blob/ae0190b6/solr/solrj/src/test/org/apache/solr/common/cloud/TestCloudCollectionsListeners.java
----------------------------------------------------------------------
diff --git a/solr/solrj/src/test/org/apache/solr/common/cloud/TestCloudCollectionsListeners.java b/solr/solrj/src/test/org/apache/solr/common/cloud/TestCloudCollectionsListeners.java
new file mode 100644
index 0000000..6d08180
--- /dev/null
+++ b/solr/solrj/src/test/org/apache/solr/common/cloud/TestCloudCollectionsListeners.java
@@ -0,0 +1,311 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.solr.common.cloud;
+
+import java.lang.invoke.MethodHandles;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Set;
+import java.util.concurrent.Callable;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Future;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
+
+import org.apache.solr.client.solrj.impl.CloudSolrClient;
+import org.apache.solr.client.solrj.request.CollectionAdminRequest;
+import org.apache.solr.cloud.SolrCloudTestCase;
+import org.apache.solr.common.util.ExecutorUtil;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class TestCloudCollectionsListeners extends SolrCloudTestCase {
+
+  private static final Logger log = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
+
+  private static final int CLUSTER_SIZE = 4;
+
+  private static final ExecutorService executor = ExecutorUtil.newMDCAwareCachedThreadPool("backgroundWatchers");
+
+  private static final int MAX_WAIT_TIMEOUT = 30;
+
+  @BeforeClass
+  public static void startCluster() throws Exception {
+    configureCluster(CLUSTER_SIZE)
+        .addConfig("config", getFile("solrj/solr/collection1/conf").toPath())
+        .configure();
+  }
+
+  @AfterClass
+  public static void shutdownBackgroundExecutors() {
+    executor.shutdown();
+  }
+
+  @Before
+  public void prepareCluster() throws Exception {
+    int missingServers = CLUSTER_SIZE - cluster.getJettySolrRunners().size();
+    for (int i = 0; i < missingServers; i++) {
+      cluster.startJettySolrRunner();
+    }
+    cluster.waitForAllNodes(30);
+  }
+
+  @Test
+  public void testSimpleCloudCollectionsListener() throws Exception {
+
+    CloudSolrClient client = cluster.getSolrClient();
+
+    Map<Integer, Set<String>> oldResults = new HashMap<>();
+    Map<Integer, Set<String>> newResults = new HashMap<>();
+
+    CloudCollectionsListener watcher1 = (oldCollections, newCollections) -> {
+      log.info("New set of collections: {}, {}", oldCollections, newCollections);
+      oldResults.put(1, oldCollections);
+      newResults.put(1, newCollections);
+    };
+    client.getZkStateReader().registerCloudCollectionsListener(watcher1);
+    CloudCollectionsListener watcher2 = (oldCollections, newCollections) -> {
+      log.info("New set of collections: {}, {}", oldCollections, newCollections);
+      oldResults.put(2, oldCollections);
+      newResults.put(2, newCollections);
+    };
+    client.getZkStateReader().registerCloudCollectionsListener(watcher2);
+
+    assertFalse("CloudCollectionsListener not triggered after registration", oldResults.get(1).contains("testcollection1"));
+    assertFalse("CloudCollectionsListener not triggered after registration", oldResults.get(2).contains("testcollection1"));
+
+    assertFalse("CloudCollectionsListener not triggered after registration", newResults.get(1).contains("testcollection1"));
+    assertFalse("CloudCollectionsListener not triggered after registration", newResults.get(2).contains("testcollection1"));
+
+    CollectionAdminRequest.createCollection("testcollection1", "config", 4, 1)
+        .processAndWait(client, MAX_WAIT_TIMEOUT);
+    client.waitForState("testcollection1", MAX_WAIT_TIMEOUT, TimeUnit.SECONDS,
+        (n, c) -> DocCollection.isFullyActive(n, c, 4, 1));
+
+    assertFalse("CloudCollectionsListener has new collection in old set of collections", oldResults.get(1).contains("testcollection1"));
+    assertFalse("CloudCollectionsListener has new collection in old set of collections", oldResults.get(2).contains("testcollection1"));
+
+    assertTrue("CloudCollectionsListener doesn't have new collection in new set of collections", newResults.get(1).contains("testcollection1"));
+    assertTrue("CloudCollectionsListener doesn't have new collection in new set of collections", newResults.get(2).contains("testcollection1"));
+
+    client.getZkStateReader().removeCloudCollectionsListener(watcher1);
+
+    CollectionAdminRequest.createCollection("testcollection2", "config", 4, 1)
+        .processAndWait(client, MAX_WAIT_TIMEOUT);
+    client.waitForState("testcollection2", MAX_WAIT_TIMEOUT, TimeUnit.SECONDS,
+        (n, c) -> DocCollection.isFullyActive(n, c, 4, 1));
+
+
+    assertFalse("CloudCollectionsListener notified after removal", oldResults.get(1).contains("testcollection1"));
+    assertTrue("CloudCollectionsListener does not contain old collection in list of old collections", oldResults.get(2).contains("testcollection1"));
+    assertFalse("CloudCollectionsListener contains new collection in old collection set", oldResults.get(1).contains("testcollection2"));
+    assertFalse("CloudCollectionsListener contains new collection in old collection set", oldResults.get(2).contains("testcollection2"));
+
+    assertFalse("CloudCollectionsListener notified after removal", newResults.get(1).contains("testcollection2"));
+    assertTrue("CloudCollectionsListener does not contain new collection in list of new collections", newResults.get(2).contains("testcollection2"));
+
+    CollectionAdminRequest.deleteCollection("testcollection1").processAndWait(client, MAX_WAIT_TIMEOUT);
+
+    CollectionAdminRequest.deleteCollection("testcollection2").processAndWait(client, MAX_WAIT_TIMEOUT);
+
+    client.getZkStateReader().removeCloudCollectionsListener(watcher2);
+  }
+
+  @Test
+  public void testCollectionDeletion() throws Exception {
+
+    CloudSolrClient client = cluster.getSolrClient();
+
+    CollectionAdminRequest.createCollection("testcollection1", "config", 4, 1)
+        .processAndWait(client, MAX_WAIT_TIMEOUT);
+    client.waitForState("testcollection1", MAX_WAIT_TIMEOUT, TimeUnit.SECONDS,
+        (n, c) -> DocCollection.isFullyActive(n, c, 4, 1));
+
+    CollectionAdminRequest.createCollection("testcollection2", "config", 4, 1)
+        .processAndWait(client, MAX_WAIT_TIMEOUT);
+    client.waitForState("testcollection2", MAX_WAIT_TIMEOUT, TimeUnit.SECONDS,
+        (n, c) -> DocCollection.isFullyActive(n, c, 4, 1));
+
+    Map<Integer, Set<String>> oldResults = new HashMap<>();
+    Map<Integer, Set<String>> newResults = new HashMap<>();
+
+    CloudCollectionsListener watcher1 = (oldCollections, newCollections) -> {
+      log.info("New set of collections: {}, {}", oldCollections, newCollections);
+      oldResults.put(1, oldCollections);
+      newResults.put(1, newCollections);
+    };
+    client.getZkStateReader().registerCloudCollectionsListener(watcher1);
+    CloudCollectionsListener watcher2 = (oldCollections, newCollections) -> {
+      log.info("New set of collections: {}, {}", oldCollections, newCollections);
+      oldResults.put(2, oldCollections);
+      newResults.put(2, newCollections);
+    };
+    client.getZkStateReader().registerCloudCollectionsListener(watcher2);
+
+
+    assertEquals("CloudCollectionsListener has old collection with size > 0 after registration", 0, oldResults.get(1).size());
+    assertEquals("CloudCollectionsListener has old collection with size > 0 after registration", 0, oldResults.get(2).size());
+
+    assertTrue("CloudCollectionsListener not notified of all collections after registration", newResults.get(1).contains("testcollection1"));
+    assertTrue("CloudCollectionsListener not notified of all collections after registration", newResults.get(1).contains("testcollection2"));
+    assertTrue("CloudCollectionsListener not notified of all collections after registration", newResults.get(2).contains("testcollection1"));
+    assertTrue("CloudCollectionsListener not notified of all collections after registration", newResults.get(2).contains("testcollection2"));
+
+    CollectionAdminRequest.deleteCollection("testcollection1").processAndWait(client, MAX_WAIT_TIMEOUT);
+
+    assertEquals("CloudCollectionsListener missing old collection after collection removal", 2, oldResults.get(1).size());
+    assertEquals("CloudCollectionsListener missing old collection after collection removal", 2, oldResults.get(2).size());
+
+    assertFalse("CloudCollectionsListener notifies with collection that no longer exists", newResults.get(1).contains("testcollection1"));
+    assertTrue("CloudCollectionsListener doesn't notify of collection that exists", newResults.get(1).contains("testcollection2"));
+    assertFalse("CloudCollectionsListener notifies with collection that no longer exists", newResults.get(2).contains("testcollection1"));
+    assertTrue("CloudCollectionsListener doesn't notify of collection that exists", newResults.get(2).contains("testcollection2"));
+
+    client.getZkStateReader().removeCloudCollectionsListener(watcher2);
+
+    CollectionAdminRequest.deleteCollection("testcollection2").processAndWait(client, MAX_WAIT_TIMEOUT);
+
+    assertEquals("CloudCollectionsListener has incorrect number of old collections", 1, oldResults.get(1).size());
+    assertTrue("CloudCollectionsListener has incorrect old collection after collection removal", oldResults.get(1).contains("testcollection2"));
+    assertEquals("CloudCollectionsListener called after removal", 2, oldResults.get(2).size());
+
+    assertFalse("CloudCollectionsListener shows live collection after removal", newResults.get(1).contains("testcollection1"));
+    assertFalse("CloudCollectionsListener shows live collection after removal", newResults.get(1).contains("testcollection2"));
+    assertFalse("CloudCollectionsListener called after removal", newResults.get(2).contains("testcollection1"));
+    assertTrue("CloudCollectionsListener called after removal", newResults.get(2).contains("testcollection2"));
+
+    client.getZkStateReader().removeCloudCollectionsListener(watcher1);
+  }
+
+  @Test
+  public void testWatchesWorkForBothStateFormats() throws Exception {
+    CloudSolrClient client = cluster.getSolrClient();
+
+    Map<Integer, Set<String>> oldResults = new HashMap<>();
+    Map<Integer, Set<String>> newResults = new HashMap<>();
+
+    CloudCollectionsListener watcher1 = (oldCollections, newCollections) -> {
+      log.info("New set of collections: {}, {}", oldCollections, newCollections);
+      oldResults.put(1, oldCollections);
+      newResults.put(1, newCollections);
+    };
+    client.getZkStateReader().registerCloudCollectionsListener(watcher1);
+    CloudCollectionsListener watcher2 = (oldCollections, newCollections) -> {
+      log.info("New set of collections: {}, {}", oldCollections, newCollections);
+      oldResults.put(2, oldCollections);
+      newResults.put(2, newCollections);
+    };
+    client.getZkStateReader().registerCloudCollectionsListener(watcher2);
+
+    assertEquals("CloudCollectionsListener has old collections with size > 0 after registration", 0, oldResults.get(1).size());
+    assertEquals("CloudCollectionsListener has old collections with size > 0 after registration", 0, oldResults.get(2).size());
+    assertEquals("CloudCollectionsListener has new collections with size > 0 after registration", 0, newResults.get(1).size());
+    assertEquals("CloudCollectionsListener has new collections with size > 0 after registration", 0, newResults.get(2).size());
+
+    // Creating old state format collection
+
+    CollectionAdminRequest.createCollection("testcollection1", "config", 4, 1)
+        .setStateFormat(1)
+        .processAndWait(client, MAX_WAIT_TIMEOUT);
+    client.waitForState("testcollection1", MAX_WAIT_TIMEOUT, TimeUnit.SECONDS,
+        (n, c) -> DocCollection.isFullyActive(n, c, 4, 1));
+
+    assertEquals("CloudCollectionsListener has old collections with size > 0 after collection created with old stateFormat", 0, oldResults.get(1).size());
+    assertEquals("CloudCollectionsListener has old collections with size > 0 after collection created with old stateFormat", 0, oldResults.get(2).size());
+    assertEquals("CloudCollectionsListener not updated with created collection with old stateFormat", 1, newResults.get(1).size());
+    assertTrue("CloudCollectionsListener not updated with created collection with old stateFormat", newResults.get(1).contains("testcollection1"));
+    assertEquals("CloudCollectionsListener not updated with created collection with old stateFormat", 1, newResults.get(2).size());
+    assertTrue("CloudCollectionsListener not updated with created collection with old stateFormat", newResults.get(2).contains("testcollection1"));
+
+    // Creating new state format collection
+
+    CollectionAdminRequest.createCollection("testcollection2", "config", 4, 1)
+        .processAndWait(client, MAX_WAIT_TIMEOUT);
+    client.waitForState("testcollection2", MAX_WAIT_TIMEOUT, TimeUnit.SECONDS,
+        (n, c) -> DocCollection.isFullyActive(n, c, 4, 1));
+
+    assertEquals("CloudCollectionsListener has incorrect old collections after collection created with new stateFormat", 1, oldResults.get(1).size());
+    assertEquals("CloudCollectionsListener has incorrect old collections after collection created with new stateFormat", 1, oldResults.get(2).size());
+    assertEquals("CloudCollectionsListener not updated with created collection with new stateFormat", 2, newResults.get(1).size());
+    assertTrue("CloudCollectionsListener not updated with created collection with new stateFormat", newResults.get(1).contains("testcollection2"));
+    assertEquals("CloudCollectionsListener not updated with created collection with new stateFormat", 2, newResults.get(2).size());
+    assertTrue("CloudCollectionsListener not updated with created collection with new stateFormat", newResults.get(2).contains("testcollection2"));
+
+    client.getZkStateReader().removeCloudCollectionsListener(watcher2);
+
+    // Creating old state format collection
+
+    CollectionAdminRequest.createCollection("testcollection3", "config", 4, 1)
+        .setStateFormat(1)
+        .processAndWait(client, MAX_WAIT_TIMEOUT);
+    client.waitForState("testcollection1", MAX_WAIT_TIMEOUT, TimeUnit.SECONDS,
+        (n, c) -> DocCollection.isFullyActive(n, c, 4, 1));
+
+    assertEquals("CloudCollectionsListener has incorrect old collections after collection created with old stateFormat", 2, oldResults.get(1).size());
+    assertEquals("CloudCollectionsListener updated after removal", 1, oldResults.get(2).size());
+    assertEquals("CloudCollectionsListener not updated with created collection with old stateFormat", 3, newResults.get(1).size());
+    assertTrue("CloudCollectionsListener not updated with created collection with old stateFormat", newResults.get(1).contains("testcollection3"));
+    assertEquals("CloudCollectionsListener updated after removal", 2, newResults.get(2).size());
+    assertFalse("CloudCollectionsListener updated after removal", newResults.get(2).contains("testcollection3"));
+
+    // Adding back listener
+    client.getZkStateReader().registerCloudCollectionsListener(watcher2);
+
+    assertEquals("CloudCollectionsListener has old collections after registration", 0, oldResults.get(2).size());
+    assertEquals("CloudCollectionsListener doesn't have all collections after registration", 3, newResults.get(2).size());
+
+    // Deleting old state format collection
+
+    CollectionAdminRequest.deleteCollection("testcollection1").processAndWait(client, MAX_WAIT_TIMEOUT);
+
+    assertEquals("CloudCollectionsListener doesn't have all old collections after collection removal", 3, oldResults.get(1).size());
+    assertEquals("CloudCollectionsListener doesn't have all old collections after collection removal", 3, oldResults.get(2).size());
+    assertEquals("CloudCollectionsListener doesn't have correct new collections after collection removal", 2, newResults.get(1).size());
+    assertEquals("CloudCollectionsListener doesn't have correct new collections after collection removal", 2, newResults.get(2).size());
+    assertFalse("CloudCollectionsListener not updated with deleted collection with old stateFormat", newResults.get(1).contains("testcollection1"));
+    assertFalse("CloudCollectionsListener not updated with deleted collection with old stateFormat", newResults.get(2).contains("testcollection1"));
+
+    CollectionAdminRequest.deleteCollection("testcollection2").processAndWait(client, MAX_WAIT_TIMEOUT);
+
+    assertEquals("CloudCollectionsListener doesn't have all old collections after collection removal", 2, oldResults.get(1).size());
+    assertEquals("CloudCollectionsListener doesn't have all old collections after collection removal", 2, oldResults.get(2).size());
+    assertEquals("CloudCollectionsListener doesn't have correct new collections after collection removal", 1, newResults.get(1).size());
+    assertEquals("CloudCollectionsListener doesn't have correct new collections after collection removal", 1, newResults.get(2).size());
+    assertFalse("CloudCollectionsListener not updated with deleted collection with new stateFormat", newResults.get(1).contains("testcollection2"));
+    assertFalse("CloudCollectionsListener not updated with deleted collection with new stateFormat", newResults.get(2).contains("testcollection2"));
+
+    client.getZkStateReader().removeCloudCollectionsListener(watcher1);
+
+    CollectionAdminRequest.deleteCollection("testcollection3").processAndWait(client, MAX_WAIT_TIMEOUT);
+
+    assertEquals("CloudCollectionsListener updated after removal", 2, oldResults.get(1).size());
+    assertEquals("CloudCollectionsListener doesn't have all old collections after collection removal", 1, oldResults.get(2).size());
+    assertEquals("CloudCollectionsListener updated after removal", 1, newResults.get(1).size());
+    assertEquals("CloudCollectionsListener doesn't have correct new collections after collection removal", 0, newResults.get(2).size());
+    assertTrue("CloudCollectionsListener updated after removal", newResults.get(1).contains("testcollection3"));
+    assertFalse("CloudCollectionsListener not updated with deleted collection with old stateFormat", newResults.get(2).contains("testcollection3"));
+
+    client.getZkStateReader().removeCloudCollectionsListener(watcher2);
+  }
+
+}