1 files changed, 9 insertions, 3 deletions
diff --git a/tests/rsa.c b/tests/rsa.c
index d836690..9addcf4 100644
--- a/tests/rsa.c
+++ b/tests/rsa.c
@@ -53,9 +53,6 @@ main (void)
   size_t src_size, dest_size;
   CK_RV res;
 
-  res = ncr_open ();
-  assert (res == CKR_OK);
-
   /* Test the generic version as well? */
   res = ncr_public_key_create_rsa (&public, modulus, sizeof (modulus),
 				   public_exponent, sizeof (public_exponent));
@@ -98,8 +95,17 @@ main (void)
   res = ncr_public_key_destroy (public);
   assert (res == CKR_OK);
 
+  /* Test reference counting */
+  res = ncr_open ();
+  assert (res == CKR_OK);
+  res = ncr_close ();
+  assert (res == CKR_OK);
+  /* Close the implicit reference */
   res = ncr_close ();
   assert (res == CKR_OK);
+  /* Check further closes are invalid */
+  res = ncr_close ();
+  assert (res == CKR_GENERAL_ERROR);
 
   return EXIT_SUCCESS;
 }