Add specs for rb_struct_s_members

ruby · Apr 1, 2017 · af5412d · af5412d
1 parent ecc868d
commit af5412d
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/optional/capi/ext/rubyspec.h b/optional/capi/ext/rubyspec.h
@@ -578,6 +578,7 @@
 #define HAVE_RB_STRUCT_DEFINE              1
 #define HAVE_RB_STRUCT_NEW                 1
 #define HAVE_RB_STRUCT_GETMEMBER           1
+#define HAVE_RB_STRUCT_S_MEMBERS           1
 #ifdef RUBY_VERSION_IS_2_4
 #define HAVE_RB_STRUCT_SIZE                1
 #endif

diff --git a/optional/capi/ext/struct_spec.c b/optional/capi/ext/struct_spec.c
@@ -19,6 +19,13 @@ static VALUE struct_spec_rb_struct_getmember(VALUE self, VALUE st, VALUE key) {
 }
 #endif
 
+#ifdef HAVE_RB_STRUCT_S_MEMBERS
+static VALUE struct_spec_rb_struct_s_members(VALUE self, VALUE klass)
+{
+  return rb_ary_dup(rb_struct_s_members(klass));
+}
+#endif
+
 #ifdef HAVE_RB_STRUCT_ASET
 static VALUE struct_spec_rb_struct_aset(VALUE self, VALUE st, VALUE key, VALUE value) {
   return rb_struct_aset(st, key, value);
@@ -69,6 +76,10 @@ void Init_struct_spec(void) {
   rb_define_method(cls, "rb_struct_getmember", struct_spec_rb_struct_getmember, 2);
 #endif
 
+#ifdef HAVE_RB_STRUCT_S_MEMBERS
+  rb_define_method(cls, "rb_struct_s_members", struct_spec_rb_struct_s_members, 1);
+#endif
+
 #ifdef HAVE_RB_STRUCT_ASET
   rb_define_method(cls, "rb_struct_aset", struct_spec_rb_struct_aset, 3);
 #endif

diff --git a/optional/capi/struct_spec.rb b/optional/capi/struct_spec.rb
@@ -97,6 +97,12 @@
     end
   end
 
+  describe "rb_struct_s_members" do
+    it "returns the struct members as an array of symbols" do
+      @s.rb_struct_s_members(@klass).should == [:a, :b, :c]
+    end
+  end
+
   describe "rb_struct_aset" do
     it "sets the value of a struct member with a symbol key" do
       @s.rb_struct_aset(@struct, :a, 1)