2 files changed, 47 insertions, 0 deletions

diff --git a/test/src/io/test_input.c b/test/src/io/test_input.c
index 30bbcb1..de4ac0f 100644
--- a/test/src/io/test_input.c
+++ b/test/src/io/test_input.c
@@ -55,6 +55,12 @@ Test(input, test_prime_nonprime) {
 	cr_assert(gequal(p, gen_m1),);
 }
 
+Test(input, test_prime_newline) {
+	fprintf(write_in, "\n");
+	GEN p = input_prime(NULL, 10);
+	cr_assert(gequal(p, gen_m1),);
+}
+
 Test(input, test_int) {
 	fprintf(write_in, "256\n");
 	GEN i = input_int(NULL, 10);
@@ -79,3 +85,10 @@ Test(input, test_str) {
 	GEN expected = strtoGENstr("something");
 	cr_assert(gequal(s, expected),);
 }
+
+Test(input, test_str_newline) {
+	fprintf(write_in, "\n");
+	GEN s = input_string(NULL);
+	GEN expected = strtoGENstr("");
+	cr_assert(gequal(s, expected),);
+}
diff --git a/test/src/util/test_random.c b/test/src/util/test_random.c
new file mode 100644
index 0000000..5e7f854
--- /dev/null
+++ b/test/src/util/test_random.c
@@ -0,0 +1,34 @@
+/*
+ * ecgen, tool for generating Elliptic curve domain parameters
+ * Copyright (C) 2017 J08nY
+ */
+
+#include "test/utils.h"
+#include "util/random.h"
+#include <criterion/criterion.h>
+
+void random_setup() {
+	default_setup();
+	random_init();
+}
+
+TestSuite(random, .init = random_setup, .fini = default_teardown);
+
+Test(random, test_random_prime) {
+	for (size_t i = 0; i < 100; ++i) {
+		GEN p = random_prime(10);
+		cr_assert(isprime(p),);
+		cr_assert_lt(gcmp(p, int2n(10)), 0,);
+		cr_assert_gt(gcmp(p, int2n(9)), 0,);
+	}
+}
+
+Test(random, test_random_int) {
+	for (size_t i = 0; i < 100; ++i) {
+		GEN j = random_int(10);
+		cr_assert_lt(gcmp(j, int2n(10)), 0,);
+		cr_assert_gt(gcmp(j, int2n(9)), 0,);
+	}
+}
+
+