From 477219415249f06d8749f208f610f5e9a13f23da Mon Sep 17 00:00:00 2001
From: Ross Lawley <ross.lawley@gmail.com>
Date: Tue, 29 Apr 2025 12:35:46 +0100
Subject: [PATCH] Update ClientEncryptionCustomEndpointTest

DRIVERS-3181
---
 .../AbstractClientEncryptionCustomEndpointTest.java | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/driver-sync/src/test/functional/com/mongodb/client/AbstractClientEncryptionCustomEndpointTest.java b/driver-sync/src/test/functional/com/mongodb/client/AbstractClientEncryptionCustomEndpointTest.java
index 9826c592190..f16d1f59855 100644
--- a/driver-sync/src/test/functional/com/mongodb/client/AbstractClientEncryptionCustomEndpointTest.java
+++ b/driver-sync/src/test/functional/com/mongodb/client/AbstractClientEncryptionCustomEndpointTest.java
@@ -120,7 +120,7 @@ public void setUp() {
                 put("endpoint", "doesnotexist.invalid:443");
             }});
             put("kmip",  new HashMap<String, Object>() {{
-                put("endpoint", "doesnotexist.local:5698");
+                put("endpoint", "doesnotexist.invalid:5698");
             }});
         }};
 
@@ -189,12 +189,11 @@ public static Collection<Object[]> data() {
                         + "  endpoint: \"kms.us-east-1.amazonaws.com:443\""
                         + "}"),
                 false, null, null, null});
-        data.add(new Object[]{"4. [aws] invalid amazon region in endpoint",
-                "aws",
+        data.add(new Object[]{"4. [kmip] Call client_encryption.createDataKey() and the following masterKey",
+                "kmip",
                 BsonDocument.parse("{\n"
-                        + "  region: \"us-east-1\",\n"
-                        + "  key: \"arn:aws:kms:us-east-1:579766882180:key/89fcc2c4-08b0-4bd9-9f25-e30687b580d0\",\n"
-                        + "  endpoint: \"kms.us-east-1.amazonaws.com:12345\"\n"
+                        + "  \"keyId\": \"1\",\n"
+                        + "  \"endpoint\": \"localhost:12345\""
                         + "}"),
                 false, MongoClientException.class, ConnectException.class, "Connection refused"});
         data.add(new Object[]{"5. [aws] invalid endpoint host",
@@ -260,7 +259,7 @@ public static Collection<Object[]> data() {
                 "kmip",
                 BsonDocument.parse("{\n"
                         + "  \"keyId\": \"1\",\n"
-                        + "  \"endpoint\": \"doesnotexist.local:5698\"\n"
+                        + "  \"endpoint\": \"doesnotexist.invalid:5698\"\n"
                         + "}"),
                 false, MongoClientException.class, UnknownHostException.class, null});
         return data;