From 8a000fea33ea03da8989f3cb2199d4c76ddc1c99 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20DA=20ROCHA?= <sebastien@da-rocha.net>
Date: Tue, 14 Aug 2018 23:31:19 +0200
Subject: [PATCH] [enh] Add basic unit test with mock
 (cert.certification_confirmation)

---
 tests/test_unit_cert.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 tests/test_unit_cert.py

diff --git a/tests/test_unit_cert.py b/tests/test_unit_cert.py
new file mode 100644
index 00000000..172db7b5
--- /dev/null
+++ b/tests/test_unit_cert.py
@@ -0,0 +1,33 @@
+from unittest.mock import patch
+
+from silkaj.cert import certification_confirmation
+
+
+@patch('builtins.input')
+def test_certification_confirmation(mock_input):
+    id_to_certify = {"pubkey": "pubkeyid to certify"}
+    main_id_to_certify = {"uid": "id to certify"}
+    mock_input.return_value = "yes"
+
+    assert certification_confirmation(
+        "certifier id",
+        "certifier pubkey",
+        id_to_certify,
+        main_id_to_certify)
+
+    mock_input.assert_called_once()
+
+
+@patch('builtins.input')
+def test_certification_confirmation_no(mock_input):
+    id_to_certify = {"pubkey": "pubkeyid to certify"}
+    main_id_to_certify = {"uid": "id to certify"}
+    mock_input.return_value = "no"
+
+    assert certification_confirmation(
+        "certifier id",
+        "certifier pubkey",
+        id_to_certify,
+        main_id_to_certify) is None
+
+    mock_input.assert_called_once()
-- 
GitLab