1 files changed, 53 insertions, 0 deletions
diff --git a/tests/common/check_stdin.c b/tests/common/check_stdin.c
new file mode 100644
index 0000000..c0d222d
--- /dev/null
+++ b/tests/common/check_stdin.c
@@ -0,0 +1,53 @@
+/*
+ * blogc: A blog compiler.
+ * Copyright (C) 2015-2016 Rafael G. Martins <rafael@rafaelmartins.eng.br>
+ *
+ * This program can be distributed under the terms of the BSD License.
+ * See the file LICENSE.
+ */
+
+#include <stdarg.h>
+#include <stddef.h>
+#include <setjmp.h>
+#include <cmocka.h>
+#include <string.h>
+#include <stdio.h>
+#include "../../src/common/stdin.h"
+
+
+int
+__wrap_fgetc(FILE *stream)
+{
+    assert_int_equal(fileno(stream), fileno(stdin));
+    return mock_type(int);
+}
+
+
+static void
+test_read(void **state)
+{
+    will_return(__wrap_fgetc, EOF);
+    char *t = bc_stdin_read();
+    assert_non_null(t);
+    assert_string_equal(t, "");
+    free(t);
+    will_return(__wrap_fgetc, 'b');
+    will_return(__wrap_fgetc, 'o');
+    will_return(__wrap_fgetc, 'l');
+    will_return(__wrap_fgetc, 'a');
+    will_return(__wrap_fgetc, EOF);
+    t = bc_stdin_read();
+    assert_non_null(t);
+    assert_string_equal(t, "bola");
+    free(t);
+}
+
+
+int
+main(void)
+{
+    const UnitTest tests[] = {
+        unit_test(test_read),
+    };
+    return run_tests(tests);
+}