minor fix

Author: qingyang-hu

internal/integration/client_side_encryption_prose_test.go

@@ -1385,6 +1385,9 @@ func TestClientSideEncryptionProse(t *testing.T) {
 	// running. See specification for port numbers and necessary arguments:
 	// https://github.com/mongodb/specifications/blob/master/source/client-side-encryption/tests/README.md#10-kms-tls-tests
 	mt.RunOpts("10. kms tls tests", noClientOpts, func(mt *mtest.T) {
+		if os.Getenv("KMS_MOCK_SERVERS_RUNNING") == "" {
+			mt.Skipf("Skipping test as KMS_MOCK_SERVERS_RUNNING is not set")
+		}
 		testcases := []struct {
 			name       string
 			port       int
@@ -1407,7 +1410,7 @@ func TestClientSideEncryptionProse(t *testing.T) {
 				ceo := options.ClientEncryption().
 					SetKmsProviders(fullKmsProvidersMap).
 					SetKeyVaultNamespace(kvNamespace)
-				cpt := setup(mt, nil, nil, ceo)
+				cpt := setup(mt, nil, defaultKvClientOptions, ceo)
 				defer cpt.teardown(mt)
 
 				_, err := cpt.clientEnc.CreateDataKey(context.Background(), "aws", options.DataKey().SetMasterKey(
@@ -1417,9 +1420,7 @@ func TestClientSideEncryptionProse(t *testing.T) {
 						{"endpoint", fmt.Sprintf("127.0.0.1:%d", tc.port)},
 					},
 				))
-				assert.NotNil(mt, err, "expected CreateDataKey error, got nil")
-				assert.True(mt, strings.Contains(err.Error(), tc.errMessage),
-					"expected CreateDataKey error to contain %v, got %v", tc.errMessage, err.Error())
+				assert.ErrorContains(mt, err, tc.errMessage)
 			})
 		}
 	})