1 /*
2 ** mrbtest - Test for Embeddable Ruby
3 **
4 ** This program runs Ruby test programs in test/t directory
5 ** against the current mruby implementation.
6 */
7 
8 
9 #include <stdio.h>
10 #include <stdlib.h>
11 #include <string.h>
12 
13 #include <mruby.h>
14 #include <mruby/proc.h>
15 #include <mruby/data.h>
16 #include <mruby/compile.h>
17 #include <mruby/string.h>
18 #include <mruby/variable.h>
19 #include <mruby/array.h>
20 
21 extern const uint8_t mrbtest_assert_irep[];
22 
23 void mrbgemtest_init(mrb_state* mrb);
24 
25 /* Print a short remark for the user */
26 static void
print_hint(void)27 print_hint(void)
28 {
29   printf("mrbtest - Embeddable Ruby Test\n\n");
30 }
31 
32 static int
eval_test(mrb_state * mrb)33 eval_test(mrb_state *mrb)
34 {
35   /* evaluate the test */
36   mrb_value result = mrb_funcall(mrb, mrb_top_self(mrb), "report", 0);
37   /* did an exception occur? */
38   if (mrb->exc) {
39     mrb_print_error(mrb);
40     mrb->exc = 0;
41     return EXIT_FAILURE;
42   }
43   else {
44     return mrb_bool(result) ? EXIT_SUCCESS : EXIT_FAILURE;
45   }
46 }
47 
48 /* Implementation of print due to the reason that there might be no print */
49 static mrb_value
t_print(mrb_state * mrb,mrb_value self)50 t_print(mrb_state *mrb, mrb_value self)
51 {
52   mrb_value *argv;
53   mrb_int argc;
54 
55   mrb_get_args(mrb, "*!", &argv, &argc);
56   for (mrb_int i = 0; i < argc; ++i) {
57     mrb_value s = mrb_obj_as_string(mrb, argv[i]);
58     fwrite(RSTRING_PTR(s), RSTRING_LEN(s), 1, stdout);
59   }
60   fflush(stdout);
61 
62   return mrb_nil_value();
63 }
64 
65 void
mrb_init_test_driver(mrb_state * mrb,mrb_bool verbose)66 mrb_init_test_driver(mrb_state *mrb, mrb_bool verbose)
67 {
68   struct RClass *krn, *mrbtest;
69 
70   krn = mrb->kernel_module;
71   mrb_define_method(mrb, krn, "t_print", t_print, MRB_ARGS_ANY());
72 
73   mrbtest = mrb_define_module(mrb, "Mrbtest");
74 
75   mrb_define_const(mrb, mrbtest, "FIXNUM_MAX", mrb_fixnum_value(MRB_INT_MAX));
76   mrb_define_const(mrb, mrbtest, "FIXNUM_MIN", mrb_fixnum_value(MRB_INT_MIN));
77   mrb_define_const(mrb, mrbtest, "FIXNUM_BIT", mrb_fixnum_value(MRB_INT_BIT));
78 
79 #ifndef MRB_WITHOUT_FLOAT
80 #ifdef MRB_USE_FLOAT
81   mrb_define_const(mrb, mrbtest, "FLOAT_TOLERANCE", mrb_float_value(mrb, 1e-6));
82 #else
83   mrb_define_const(mrb, mrbtest, "FLOAT_TOLERANCE", mrb_float_value(mrb, 1e-12));
84 #endif
85 #endif
86 
87   if (verbose) {
88     mrb_gv_set(mrb, mrb_intern_lit(mrb, "$mrbtest_verbose"), mrb_true_value());
89   }
90 }
91 
92 void
mrb_t_pass_result(mrb_state * mrb_dst,mrb_state * mrb_src)93 mrb_t_pass_result(mrb_state *mrb_dst, mrb_state *mrb_src)
94 {
95   mrb_value res_src;
96 
97   if (mrb_src->exc) {
98     mrb_print_error(mrb_src);
99     exit(EXIT_FAILURE);
100   }
101 
102 #define TEST_COUNT_PASS(name)                                           \
103   do {                                                                  \
104     res_src = mrb_gv_get(mrb_src, mrb_intern_lit(mrb_src, "$" #name));  \
105     if (mrb_fixnum_p(res_src)) {                                        \
106       mrb_value res_dst = mrb_gv_get(mrb_dst, mrb_intern_lit(mrb_dst, "$" #name)); \
107       mrb_gv_set(mrb_dst, mrb_intern_lit(mrb_dst, "$" #name), mrb_fixnum_value(mrb_fixnum(res_dst) + mrb_fixnum(res_src))); \
108     }                                                                   \
109   } while (FALSE)                                                       \
110 
111   TEST_COUNT_PASS(ok_test);
112   TEST_COUNT_PASS(ko_test);
113   TEST_COUNT_PASS(kill_test);
114   TEST_COUNT_PASS(skip_test);
115 
116 #undef TEST_COUNT_PASS
117 
118   res_src = mrb_gv_get(mrb_src, mrb_intern_lit(mrb_src, "$asserts"));
119 
120   if (mrb_array_p(res_src)) {
121     mrb_int i;
122     mrb_value res_dst = mrb_gv_get(mrb_dst, mrb_intern_lit(mrb_dst, "$asserts"));
123     for (i = 0; i < RARRAY_LEN(res_src); ++i) {
124       mrb_value val_src = RARRAY_PTR(res_src)[i];
125       mrb_ary_push(mrb_dst, res_dst, mrb_str_new(mrb_dst, RSTRING_PTR(val_src), RSTRING_LEN(val_src)));
126     }
127   }
128 }
129 
130 int
main(int argc,char ** argv)131 main(int argc, char **argv)
132 {
133   mrb_state *mrb;
134   int ret;
135   mrb_bool verbose = FALSE;
136 
137   print_hint();
138 
139   /* new interpreter instance */
140   mrb = mrb_open();
141   if (mrb == NULL) {
142     fprintf(stderr, "Invalid mrb_state, exiting test driver");
143     return EXIT_FAILURE;
144   }
145 
146   if (argc == 2 && argv[1][0] == '-' && argv[1][1] == 'v') {
147     printf("verbose mode: enable\n\n");
148     verbose = TRUE;
149   }
150 
151   mrb_init_test_driver(mrb, verbose);
152   mrb_load_irep(mrb, mrbtest_assert_irep);
153   mrbgemtest_init(mrb);
154   ret = eval_test(mrb);
155   mrb_close(mrb);
156 
157   return ret;
158 }
159