-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathadd_coverage_result_set.patch
52 lines (49 loc) · 1.53 KB
/
add_coverage_result_set.patch
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
commit c6e1784e9ec10f34e07c20c687004df9d3a84732
Author: Ryan Davis <[email protected]>
Date: Wed Oct 5 17:09:27 2016 -0700
Added Coverage.result=.
diff --git a/ext/coverage/coverage.c b/ext/coverage/coverage.c
index 9a9011a..6b7457c 100644
--- a/ext/coverage/coverage.c
+++ b/ext/coverage/coverage.c
@@ -75,6 +75,34 @@ rb_coverage_result(VALUE klass)
return ncoverages;
}
+static int
+coverage_merge_result_i(VALUE path, VALUE coverage, VALUE coverages)
+{
+ VALUE ary = rb_hash_lookup2(coverages, path, Qnil);
+
+ if (RTEST(ary)) {
+ rb_ary_replace(ary, coverage);
+ }
+
+ return ST_CONTINUE;
+}
+
+static VALUE
+rb_coverage_set_result(VALUE klass, VALUE ncoverages)
+{
+ VALUE coverages = rb_get_coverages();
+
+ Check_Type(ncoverages, T_HASH);
+
+ if (!RTEST(coverages)) {
+ rb_raise(rb_eRuntimeError, "coverage measurement is not enabled");
+ }
+
+ rb_hash_foreach(ncoverages, coverage_merge_result_i, coverages);
+
+ return ncoverages;
+}
+
/* Coverage provides coverage measurement feature for Ruby.
* This feature is experimental, so these APIs may be changed in future.
*
@@ -114,5 +142,6 @@ Init_coverage(void)
VALUE rb_mCoverage = rb_define_module("Coverage");
rb_define_module_function(rb_mCoverage, "start", rb_coverage_start, 0);
rb_define_module_function(rb_mCoverage, "result", rb_coverage_result, 0);
+ rb_define_module_function(rb_mCoverage, "result=", rb_coverage_set_result, 1);
rb_define_module_function(rb_mCoverage, "peek_result", rb_coverage_peek_result, 0);
}