1 files changed, 26 insertions, 0 deletions
diff --git a/nss/gtests/freebl_gtest/dh_unittest.cc b/nss/gtests/freebl_gtest/dh_unittest.cc
new file mode 100644
index 0000000..498ca20
--- /dev/null
+++ b/nss/gtests/freebl_gtest/dh_unittest.cc
@@ -0,0 +1,26 @@
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this file,
+// You can obtain one at http://mozilla.org/MPL/2.0/.
+
+#include "blapi.h"
+#include "gtest/gtest.h"
+
+namespace nss_test {
+
+class DHTest : public ::testing::Test {
+ protected:
+  void TestGenParamSuccess(int size) {
+    DHParams *params;
+    for (int i = 0; i < 10; i++) {
+      EXPECT_EQ(SECSuccess, DH_GenParam(size, &params));
+      PORT_FreeArena(params->arena, PR_TRUE);
+    }
+  }
+};
+
+// Test parameter generation for minimum and some common key sizes
+TEST_F(DHTest, DhGenParamSuccessTest16) { TestGenParamSuccess(16); }
+TEST_F(DHTest, DhGenParamSuccessTest224) { TestGenParamSuccess(224); }
+TEST_F(DHTest, DhGenParamSuccessTest256) { TestGenParamSuccess(256); }
+
+}  // nss_test