Check rb_str_set_len() on both binary and UTF-8 strings

Author: eregon

spec/ruby/optional/capi/string_spec.rb

@@ -32,43 +32,45 @@ def to_str
     end
   end
 
-  describe "rb_str_set_len" do
-    before :each do
-      # Make a completely new copy of the string
-      # for every example (#dup doesn't cut it).
-      @str = "abcdefghij".b[0..-1]
-    end
+  [Encoding::BINARY, Encoding::UTF_8].each do |enc|
+    describe "rb_str_set_len on a #{enc.name} String" do
+      before :each do
+        # Make a completely new copy of the string
+        # for every example (#dup doesn't cut it).
+        @str = "abcdefghij".force_encoding(enc)[0..-1]
+      end
 
-    it "reduces the size of the string" do
-      @s.rb_str_set_len(@str, 5).should == "abcde"
-    end
+      it "reduces the size of the string" do
+        @s.rb_str_set_len(@str, 5).should == "abcde"
+      end
 
-    it "inserts a NULL byte at the length" do
-      @s.rb_str_set_len(@str, 5).should == "abcde"
-      @s.rb_str_set_len(@str, 8).should == "abcde\x00gh"
-    end
+      it "inserts a NULL byte at the length" do
+        @s.rb_str_set_len(@str, 5).should == "abcde"
+        @s.rb_str_set_len(@str, 8).should == "abcde\x00gh"
+      end
 
-    it "updates the byte size and character size" do
-      @s.rb_str_set_len(@str, 4)
-      @str.bytesize.should == 4
-      @str.size.should == 4
-      @str.should == "abcd"
-    end
+      it "updates the byte size and character size" do
+        @s.rb_str_set_len(@str, 4)
+        @str.bytesize.should == 4
+        @str.size.should == 4
+        @str.should == "abcd"
+      end
 
-    it "updates the string's attributes visible in C code" do
-      @s.rb_str_set_len_RSTRING_LEN(@str, 4).should == 4
-    end
+      it "updates the string's attributes visible in C code" do
+        @s.rb_str_set_len_RSTRING_LEN(@str, 4).should == 4
+      end
 
-    it "can reveal characters written from C with RSTRING_PTR" do
-      @s.rb_str_set_len(@str, 1)
-      @str.should == "a"
+      it "can reveal characters written from C with RSTRING_PTR" do
+        @s.rb_str_set_len(@str, 1)
+        @str.should == "a"
 
-      @s.RSTRING_PTR_set(@str, 1, 'B'.ord)
-      @s.RSTRING_PTR_set(@str, 2, 'C'.ord)
-      @s.rb_str_set_len(@str, 3)
+        @s.RSTRING_PTR_set(@str, 1, 'B'.ord)
+        @s.RSTRING_PTR_set(@str, 2, 'C'.ord)
+        @s.rb_str_set_len(@str, 3)
 
-      @str.bytesize.should == 3
-      @str.should == "aBC"
+        @str.bytesize.should == 3
+        @str.should == "aBC"
+      end
     end
   end