1 files changed, 116 insertions, 0 deletions
diff --git a/pkcs11/wrap-layer/test-set-pin.c b/pkcs11/wrap-layer/test-set-pin.c
new file mode 100644
index 00000000..f53dc349
--- /dev/null
+++ b/pkcs11/wrap-layer/test-set-pin.c
@@ -0,0 +1,116 @@
+/*
+ * gnome-keyring
+ *
+ * Copyright (C) 2010 Stefan Walter
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this program; if not, see
+ * <http://www.gnu.org/licenses/>.
+ */
+
+#include "config.h"
+
+#include "wrap-layer/gkm-wrap-layer.h"
+#include "wrap-layer/gkm-wrap-prompt.h"
+
+#include "egg/egg-testing.h"
+
+#include "gkm/gkm-mock.h"
+#include "gkm/gkm-test.h"
+
+#include <gcr/gcr-base.h>
+
+#include <glib-object.h>
+
+typedef struct {
+	CK_FUNCTION_LIST functions;
+	CK_FUNCTION_LIST_PTR module;
+	CK_SESSION_HANDLE session;
+} Test;
+
+static void
+setup (Test *test, gconstpointer unused)
+{
+	CK_FUNCTION_LIST_PTR funcs;
+	CK_SLOT_ID slot_id;
+	CK_ULONG n_slots = 1;
+	const gchar *prompter;
+	CK_RV rv;
+
+	/* Always start off with test test->functions */
+	rv = gkm_mock_C_GetFunctionList (&funcs);
+	gkm_assert_cmprv (rv, ==, CKR_OK);
+	memcpy (&test->functions, funcs, sizeof (test->functions));
+
+	gkm_wrap_layer_reset_modules ();
+	gkm_wrap_layer_add_module (&test->functions);
+	test->module = gkm_wrap_layer_get_functions ();
+
+	prompter = gcr_mock_prompter_start ();
+	gkm_wrap_prompt_set_prompter_name (prompter);
+
+	/* Open a test->session */
+	rv = (test->module->C_Initialize) (NULL);
+	gkm_assert_cmprv (rv, ==, CKR_OK);
+
+	rv = (test->module->C_GetSlotList) (CK_TRUE, &slot_id, &n_slots);
+	gkm_assert_cmprv (rv, ==, CKR_OK);
+
+	rv = (test->module->C_OpenSession) (slot_id, CKF_SERIAL_SESSION, NULL, NULL, &test->session);
+	gkm_assert_cmprv (rv, ==, CKR_OK);
+}
+
+static void
+teardown (Test *test, gconstpointer unused)
+{
+	CK_RV rv;
+
+	g_assert (!gcr_mock_prompter_is_expecting ());
+	gcr_mock_prompter_stop ();
+
+	rv = (test->module->C_CloseSession) (test->session);
+	gkm_assert_cmprv (rv, ==, CKR_OK);
+	test->session = 0;
+
+	rv = (test->module->C_Finalize) (NULL);
+	gkm_assert_cmprv (rv, ==, CKR_OK);
+	test->module = NULL;
+}
+
+static void
+test_ok_passwords (Test *test, gconstpointer unused)
+{
+	CK_RV rv;
+
+	gcr_mock_prompter_expect_password_ok ("booo", NULL);
+	gcr_mock_prompter_expect_password_ok ("new", NULL);
+
+	rv = (test->module->C_SetPIN) (test->session, NULL, 0, NULL, 0);
+	gkm_assert_cmprv (rv, ==, CKR_OK);
+
+	rv = (test->module->C_Login) (test->session, CKU_USER, (guchar*)"new", 3);
+	gkm_assert_cmprv (rv, ==, CKR_OK);
+}
+
+int
+main (int argc, char **argv)
+{
+#if !GLIB_CHECK_VERSION(2,35,0)
+	g_type_init ();
+#endif
+	g_test_init (&argc, &argv, NULL);
+
+	g_test_add ("/wrap-layer/set-pin/ok_passwords", Test, NULL, setup, test_ok_passwords, teardown);
+
+	return egg_tests_run_in_thread_with_loop ();
+}