1 files changed, 128 insertions, 0 deletions
diff --git a/test/GpgCoreTestKeygen.cpp b/test/GpgCoreTestKeygen.cpp
new file mode 100644
index 00000000..b703ee40
--- /dev/null
+++ b/test/GpgCoreTestKeygen.cpp
@@ -0,0 +1,128 @@
+/**
+ * This file is part of GpgFrontend.
+ *
+ * GpgFrontend is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Foobar 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Foobar.  If not, see <https://www.gnu.org/licenses/>.
+ *
+ * The initial version of the source code is inherited from gpg4usb-team.
+ * Their source code version also complies with GNU General Public License.
+ *
+ * The source code version of this software was modified and released
+ * by Saturneric<[email protected]> starting on May 12, 2021.
+ *
+ */
+
+#include "GpgFrontendTest.h"
+#include "gpg/GpgGenKeyInfo.h"
+#include "gpg/function/GpgKeyGetter.h"
+#include "gpg/function/GpgKeyOpera.h"
+
+TEST_F(GpgCoreTest, GenerateKeyTest) {
+  auto& key_opera = GpgFrontend::GpgKeyOpera::GetInstance(default_channel);
+  auto keygen_info = std::make_unique<GpgFrontend::GenKeyInfo>();
+  keygen_info->setName("foo");
+  keygen_info->setEmail("[email protected]");
+  keygen_info->setComment("");
+  keygen_info->setKeySize(1024);
+  keygen_info->setAlgo("rsa");
+  keygen_info->setNonExpired(true);
+  keygen_info->setNonPassPhrase(true);
+
+  GpgFrontend::GpgGenKeyResult result = nullptr;
+  auto err = GpgFrontend::check_gpg_error_2_err_code(
+      key_opera.GenerateKey(keygen_info, result));
+  ASSERT_EQ(err, GPG_ERR_NO_ERROR);
+
+  auto fpr = result->fpr;
+  ASSERT_FALSE(fpr == nullptr);
+
+  auto key =
+      GpgFrontend::GpgKeyGetter::GetInstance(default_channel).GetKey(fpr);
+  ASSERT_TRUE(key.good());
+  key_opera.DeleteKey(fpr);
+}
+
+TEST_F(GpgCoreTest, GenerateKeyTest_1) {
+  auto& key_opera = GpgFrontend::GpgKeyOpera::GetInstance(default_channel);
+  auto keygen_info = std::make_unique<GpgFrontend::GenKeyInfo>();
+  keygen_info->setName("foo");
+  keygen_info->setEmail("[email protected]");
+  keygen_info->setComment("hello gpgfrontend");
+  keygen_info->setAlgo("rsa");
+  keygen_info->setKeySize(4096);
+  keygen_info->setNonExpired(false);
+  keygen_info->setExpired(boost::posix_time::second_clock::local_time() +
+                          boost::posix_time::hours(24));
+  keygen_info->setNonPassPhrase(false);
+
+  GpgFrontend::GpgGenKeyResult result = nullptr;
+  auto err = GpgFrontend::check_gpg_error_2_err_code(
+      key_opera.GenerateKey(keygen_info, result));
+  ASSERT_EQ(err, GPG_ERR_NO_ERROR);
+
+  auto fpr = result->fpr;
+  ASSERT_FALSE(fpr == nullptr);
+
+  auto key =
+      GpgFrontend::GpgKeyGetter::GetInstance(default_channel).GetKey(fpr);
+  ASSERT_TRUE(key.good());
+  key_opera.DeleteKey(fpr);
+}
+
+TEST_F(GpgCoreTest, GenerateKeyTest_4) {
+  auto& key_opera = GpgFrontend::GpgKeyOpera::GetInstance(default_channel);
+  auto keygen_info = std::make_unique<GpgFrontend::GenKeyInfo>();
+  keygen_info->setName("foo");
+  keygen_info->setEmail("[email protected]");
+  keygen_info->setComment("");
+  keygen_info->setAlgo("dsa");
+  keygen_info->setNonExpired(true);
+  keygen_info->setNonPassPhrase(false);
+
+  GpgFrontend::GpgGenKeyResult result = nullptr;
+  auto err = GpgFrontend::check_gpg_error_2_err_code(
+      key_opera.GenerateKey(keygen_info, result));
+  ASSERT_EQ(err, GPG_ERR_NO_ERROR);
+
+  auto fpr = result->fpr;
+  ASSERT_FALSE(fpr == nullptr);
+
+  auto key =
+      GpgFrontend::GpgKeyGetter::GetInstance(default_channel).GetKey(fpr);
+  ASSERT_TRUE(key.good());
+  key_opera.DeleteKey(fpr);
+}
+
+TEST_F(GpgCoreTest, GenerateKeyTest_5) {
+  auto& key_opera = GpgFrontend::GpgKeyOpera::GetInstance(default_channel);
+  auto keygen_info = std::make_unique<GpgFrontend::GenKeyInfo>();
+  keygen_info->setName("foo");
+  keygen_info->setEmail("[email protected]");
+  keygen_info->setComment("");
+  keygen_info->setAlgo("ed25519");
+  keygen_info->setNonExpired(true);
+  keygen_info->setNonPassPhrase(false);
+
+  GpgFrontend::GpgGenKeyResult result = nullptr;
+  auto err = GpgFrontend::check_gpg_error_2_err_code(
+      key_opera.GenerateKey(keygen_info, result));
+  ASSERT_EQ(err, GPG_ERR_NO_ERROR);
+
+  auto fpr = result->fpr;
+  ASSERT_FALSE(fpr == nullptr);
+
+  auto key =
+      GpgFrontend::GpgKeyGetter::GetInstance(default_channel).GetKey(fpr);
+  ASSERT_TRUE(key.good());
+  key_opera.DeleteKey(fpr);
+}