SERVER-76188 Integrate cleanup implementation with unsharded command invocation

1 files changed, 40 insertions, 0 deletions

diff --git a/jstests/noPassthrough/fle2_shardsvr_cleanup.js b/jstests/noPassthrough/fle2_shardsvr_cleanup.js
new file mode 100644
index 00000000000..ac9a1b57751
--- /dev/null
+++ b/jstests/noPassthrough/fle2_shardsvr_cleanup.js
@@ -0,0 +1,40 @@
+/**
+ * Cannot run cleanup against a shard server
+ *
+ * @tags: [
+ * requires_fcv_70,
+ * requires_sharding,
+ * featureFlagFLE2CleanupCommand
+ * ]
+ */
+load("jstests/fle2/libs/encrypted_client_util.js");
+
+(function() {
+'use strict';
+
+function runTest(mongosConn, shardConn) {
+    let dbName = 'testdb';
+
+    let clientMongoS = new EncryptedClient(mongosConn, dbName);
+
+    assert.commandWorked(clientMongoS.createEncryptionCollection("basic", {
+        encryptedFields: {
+            "fields":
+                [{"path": "first", "bsonType": "string", "queries": {"queryType": "equality"}}]
+        }
+    }));
+
+    let clientShard = new EncryptedClient(shardConn, dbName);
+
+    assert.commandFailedWithCode(clientShard.getDB().basic.cleanup(), 7618804);
+}
+
+jsTestLog("Sharding: Testing fle2 cleanup not allowed against a shard server");
+{
+    const st = new ShardingTest({shards: 1, mongos: 1, config: 1});
+
+    runTest(st.s, st.shard0);
+
+    st.stop();
+}
+}());