Skip to content

Commit b15bdde

Browse files
committed
fix test leaks
1 parent c78802f commit b15bdde

File tree

1 file changed

+24
-17
lines changed

1 file changed

+24
-17
lines changed

src/libmongoc/tests/test-mongoc-client-side-encryption.c

Lines changed: 24 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -3233,16 +3233,6 @@ test_kms_tls_options_extra_rejected (void *unused)
32333233
mongoc_client_destroy (keyvault_client);
32343234
}
32353235

3236-
static mongoc_ssl_opt_t
3237-
make_csfle_ssl_opts (void)
3238-
{
3239-
/* The failpoint server is pretending to be a KMS server and uses the same certs */
3240-
mongoc_ssl_opt_t ssl_opts = {0};
3241-
ssl_opts.ca_file = test_framework_getenv_required ("MONGOC_TEST_CSFLE_TLS_CA_FILE");
3242-
ssl_opts.pem_file = test_framework_getenv_required ("MONGOC_TEST_CSFLE_TLS_CERTIFICATE_KEY_FILE");
3243-
return ssl_opts;
3244-
}
3245-
32463236
static void
32473237
set_retry_failpoint (mongoc_ssl_opt_t *ssl_opts, bool network)
32483238
{
@@ -6155,7 +6145,9 @@ test_kms_retry (void *unused)
61556145
bson_error_t error = {0};
61566146
bson_value_t keyid;
61576147
mongoc_client_encryption_datakey_opts_t *dkopts;
6158-
mongoc_ssl_opt_t ssl_opts = make_csfle_ssl_opts ();
6148+
char *ca_file = test_framework_getenv_required ("MONGOC_TEST_CSFLE_TLS_CA_FILE");
6149+
char *pem_file = test_framework_getenv_required ("MONGOC_TEST_CSFLE_TLS_CERTIFICATE_KEY_FILE");
6150+
mongoc_ssl_opt_t ssl_opts = {.ca_file = ca_file, .pem_file = pem_file};
61596151
bool res;
61606152

61616153
bson_value_t to_encrypt = {.value_type = BSON_TYPE_INT32, .value.v_int32 = 1};
@@ -6168,26 +6160,32 @@ test_kms_retry (void *unused)
61686160
mongoc_client_encryption_datakey_opts_set_masterkey (
61696161
dkopts, tmp_bson (BSON_STR ({"region" : "r", "key" : "k", "endpoint" : "127.0.0.1:9003"})));
61706162
res = mongoc_client_encryption_create_datakey (client_encryption, "aws", dkopts, &keyid, &error);
6171-
ASSERT (res);
6163+
ASSERT_OR_PRINT (res, error);
61726164

61736165
set_retry_failpoint (&ssl_opts, false);
61746166
set_retry_failpoint (&ssl_opts, true);
61756167
mongoc_client_encryption_encrypt_opts_set_keyid (encrypt_opts, &keyid);
61766168
res = mongoc_client_encryption_encrypt (client_encryption, &to_encrypt, encrypt_opts, &encrypted_field, &error);
6177-
ASSERT (res);
6169+
ASSERT_OR_PRINT (res, error);
6170+
bson_value_destroy (&keyid);
6171+
bson_value_destroy (&encrypted_field);
6172+
mongoc_client_encryption_datakey_opts_destroy (dkopts);
61786173

61796174
// Azure
61806175
dkopts = mongoc_client_encryption_datakey_opts_new ();
61816176
mongoc_client_encryption_datakey_opts_set_masterkey (
61826177
dkopts, tmp_bson (BSON_STR ({"keyVaultEndpoint" : "127.0.0.1:9003", "keyName" : "foo"})));
61836178
res = mongoc_client_encryption_create_datakey (client_encryption, "azure", dkopts, &keyid, &error);
6184-
ASSERT (res);
6179+
ASSERT_OR_PRINT (res, error);
61856180

61866181
set_retry_failpoint (&ssl_opts, false);
61876182
set_retry_failpoint (&ssl_opts, true);
61886183
mongoc_client_encryption_encrypt_opts_set_keyid (encrypt_opts, &keyid);
61896184
res = mongoc_client_encryption_encrypt (client_encryption, &to_encrypt, encrypt_opts, &encrypted_field, &error);
6190-
ASSERT (res);
6185+
ASSERT_OR_PRINT (res, error);
6186+
bson_value_destroy (&keyid);
6187+
bson_value_destroy (&encrypted_field);
6188+
mongoc_client_encryption_datakey_opts_destroy (dkopts);
61916189

61926190
// GCP
61936191
dkopts = mongoc_client_encryption_datakey_opts_new ();
@@ -6199,13 +6197,22 @@ test_kms_retry (void *unused)
61996197
"endpoint" : "127.0.0.1:9003"
62006198
})));
62016199
res = mongoc_client_encryption_create_datakey (client_encryption, "gcp", dkopts, &keyid, &error);
6202-
ASSERT (res);
6200+
ASSERT_OR_PRINT (res, error);
62036201

62046202
set_retry_failpoint (&ssl_opts, false);
62056203
set_retry_failpoint (&ssl_opts, true);
62066204
mongoc_client_encryption_encrypt_opts_set_keyid (encrypt_opts, &keyid);
62076205
res = mongoc_client_encryption_encrypt (client_encryption, &to_encrypt, encrypt_opts, &encrypted_field, &error);
6208-
ASSERT (res);
6206+
ASSERT_OR_PRINT (res, error);
6207+
bson_value_destroy (&keyid);
6208+
bson_value_destroy (&encrypted_field);
6209+
mongoc_client_encryption_datakey_opts_destroy (dkopts);
6210+
6211+
bson_free (ca_file);
6212+
bson_free (pem_file);
6213+
mongoc_client_encryption_encrypt_opts_destroy (encrypt_opts);
6214+
mongoc_client_encryption_destroy (client_encryption);
6215+
mongoc_client_destroy (keyvault_client);
62096216
}
62106217

62116218
void

0 commit comments

Comments
 (0)