From f3732097c10c763c117c3cd0f75f92aecba12fde Mon Sep 17 00:00:00 2001
From: Naveen Tatikonda <navtat@amazon.com>
Date: Mon, 23 Jan 2023 19:38:27 -0600
Subject: [PATCH] Add BWC Tests for Clear Cache API

Signed-off-by: Naveen Tatikonda <navtat@amazon.com>
---
 .../org/opensearch/knn/bwc/ClearCacheIT.java  | 49 +++++++++++++++++
 .../org/opensearch/knn/bwc/ClearCacheIT.java  | 55 +++++++++++++++++++
 2 files changed, 104 insertions(+)
 create mode 100644 qa/restart-upgrade/src/test/java/org/opensearch/knn/bwc/ClearCacheIT.java
 create mode 100644 qa/rolling-upgrade/src/test/java/org/opensearch/knn/bwc/ClearCacheIT.java

diff --git a/qa/restart-upgrade/src/test/java/org/opensearch/knn/bwc/ClearCacheIT.java b/qa/restart-upgrade/src/test/java/org/opensearch/knn/bwc/ClearCacheIT.java
new file mode 100644
index 0000000000..045821e09e
--- /dev/null
+++ b/qa/restart-upgrade/src/test/java/org/opensearch/knn/bwc/ClearCacheIT.java
@@ -0,0 +1,49 @@
+/*
+ *  Copyright OpenSearch Contributors
+ *  SPDX-License-Identifier: Apache-2.0
+ */
+
+package org.opensearch.knn.bwc;
+
+import java.util.Collections;
+import static org.opensearch.knn.TestUtils.NODES_BWC_CLUSTER;
+
+public class ClearCacheIT extends AbstractRestartUpgradeTestCase {
+    private static final String TEST_FIELD = "test-field";
+    private static final int DIMENSIONS = 5;
+    private static int docId = 0;
+    private static final int NUM_DOCS = 10;
+    private static int queryCnt = 0;
+    private static final int K = 5;
+
+    // Restart Upgrade BWC Tests to validate Clear Cache API
+    public void testClearCache() throws Exception {
+        waitForClusterHealthGreen(NODES_BWC_CLUSTER);
+        if (isRunningAgainstOldCluster()) {
+            createKnnIndex(testIndex, getKNNDefaultIndexSettings(), createKnnIndexMapping(TEST_FIELD, DIMENSIONS));
+            addKNNDocs(testIndex, TEST_FIELD, DIMENSIONS, docId, NUM_DOCS);
+        } else {
+            queryCnt = NUM_DOCS;
+            validateClearCacheOnUpgrade(queryCnt);
+
+            docId = NUM_DOCS;
+            addKNNDocs(testIndex, TEST_FIELD, DIMENSIONS, docId, NUM_DOCS);
+
+            queryCnt = queryCnt + NUM_DOCS;
+            validateClearCacheOnUpgrade(queryCnt);
+            deleteKNNIndex(testIndex);
+        }
+    }
+
+    // validation steps for Clear Cache API after upgrading node to new version
+    private void validateClearCacheOnUpgrade(int queryCount) throws Exception {
+        int graphCount = getTotalGraphsInCache();
+        knnWarmup(Collections.singletonList(testIndex));
+        assertTrue(getTotalGraphsInCache() > graphCount);
+        validateKNNSearch(testIndex, TEST_FIELD, DIMENSIONS, queryCount, K);
+
+        clearCache(Collections.singletonList(testIndex));
+        assertEquals(0, getTotalGraphsInCache());
+        validateKNNSearch(testIndex, TEST_FIELD, DIMENSIONS, queryCount, K);
+    }
+}
diff --git a/qa/rolling-upgrade/src/test/java/org/opensearch/knn/bwc/ClearCacheIT.java b/qa/rolling-upgrade/src/test/java/org/opensearch/knn/bwc/ClearCacheIT.java
new file mode 100644
index 0000000000..24c674d0d2
--- /dev/null
+++ b/qa/rolling-upgrade/src/test/java/org/opensearch/knn/bwc/ClearCacheIT.java
@@ -0,0 +1,55 @@
+/*
+ *  Copyright OpenSearch Contributors
+ *  SPDX-License-Identifier: Apache-2.0
+ */
+
+package org.opensearch.knn.bwc;
+
+import java.util.Collections;
+
+import static org.opensearch.knn.TestUtils.NODES_BWC_CLUSTER;
+
+public class ClearCacheIT extends AbstractRollingUpgradeTestCase {
+    private static final String TEST_FIELD = "test-field";
+    private static final int DIMENSIONS = 5;
+    private static int docId = 0;
+    private static final int K = 5;
+    private static final int NUM_DOCS = 10;
+    private static int queryCnt = 0;
+
+    // Rolling Upgrade BWC Tests to validate Clear Cache API
+    public void testClearCache() throws Exception {
+        waitForClusterHealthGreen(NODES_BWC_CLUSTER);
+        switch (getClusterType()) {
+            case OLD:
+                createKnnIndex(testIndex, getKNNDefaultIndexSettings(), createKnnIndexMapping(TEST_FIELD, DIMENSIONS));
+                int docIdOld = 0;
+                addKNNDocs(testIndex, TEST_FIELD, DIMENSIONS, docIdOld, NUM_DOCS);
+                break;
+            case UPGRADED:
+                queryCnt = NUM_DOCS;
+                validateClearCacheOnUpgrade(queryCnt);
+
+                docId = NUM_DOCS;
+                addKNNDocs(testIndex, TEST_FIELD, DIMENSIONS, docId, NUM_DOCS);
+
+                queryCnt = queryCnt + NUM_DOCS;
+                validateClearCacheOnUpgrade(queryCnt);
+                deleteKNNIndex(testIndex);
+        }
+
+    }
+
+    // validation steps for Clear Cache API after upgrading all nodes from old version to new version
+    public void validateClearCacheOnUpgrade(int queryCount) throws Exception {
+        int graphCount = getTotalGraphsInCache();
+        knnWarmup(Collections.singletonList(testIndex));
+        assertTrue(getTotalGraphsInCache() > graphCount);
+        validateKNNSearch(testIndex, TEST_FIELD, DIMENSIONS, queryCount, K);
+
+        clearCache(Collections.singletonList(testIndex));
+        assertEquals(0, getTotalGraphsInCache());
+        validateKNNSearch(testIndex, TEST_FIELD, DIMENSIONS, queryCount, K);
+    }
+
+}