Add specs for rb_interned_str and rb_interned_str_cstr

Author: herwinw

optional/capi/ext/string_spec.c

@@ -581,6 +581,14 @@ static VALUE string_spec_rb_str_to_interned_str(VALUE self, VALUE str) {
   return rb_str_to_interned_str(str);
 }
 
+static VALUE string_spec_rb_interned_str(VALUE self, VALUE str, VALUE len) {
+  return rb_interned_str(RSTRING_PTR(str), FIX2LONG(len));
+}
+
+static VALUE string_spec_rb_interned_str_cstr(VALUE self, VALUE str) {
+  return rb_interned_str_cstr(RSTRING_PTR(str));
+}
+
 void Init_string_spec(void) {
   VALUE cls = rb_define_class("CApiStringSpecs", rb_cObject);
   rb_define_method(cls, "rb_cstr2inum", string_spec_rb_cstr2inum, 2);
@@ -681,6 +689,8 @@ void Init_string_spec(void) {
   rb_define_method(cls, "rb_enc_interned_str_cstr", string_spec_rb_enc_interned_str_cstr, 2);
   rb_define_method(cls, "rb_enc_interned_str", string_spec_rb_enc_interned_str, 3);
   rb_define_method(cls, "rb_str_to_interned_str", string_spec_rb_str_to_interned_str, 1);
+  rb_define_method(cls, "rb_interned_str", string_spec_rb_interned_str, 2);
+  rb_define_method(cls, "rb_interned_str_cstr", string_spec_rb_interned_str_cstr, 1);
 }
 
 #ifdef __cplusplus

optional/capi/string_spec.rb

@@ -1373,4 +1373,106 @@ def inspect
       @s.rb_str_to_interned_str("hello").should.equal?(-"hello")
     end
   end
+
+  describe "rb_interned_str" do
+    it "returns a frozen string" do
+      str = "hello"
+      result = @s.rb_interned_str(str, str.bytesize)
+      result.should.is_a?(String)
+      result.should.frozen?
+      result.encoding.should == Encoding::US_ASCII
+    end
+
+    it "returns the same frozen string" do
+      str = "hello"
+      result1 = @s.rb_interned_str(str, str.bytesize)
+      result2 = @s.rb_interned_str(str, str.bytesize)
+      result1.should.equal?(result2)
+    end
+
+    it "supports strings with embedded null bytes" do
+      str = "foo\x00bar\x00baz".b
+      result = @s.rb_interned_str(str, str.bytesize)
+      result.should == str
+    end
+
+    it "support binary strings that are invalid in ASCII encoding" do
+      str = "foo\x81bar\x82baz".b
+      result = @s.rb_interned_str(str, str.bytesize)
+      result.encoding.should == Encoding::US_ASCII
+      result.should == str.dup.force_encoding(Encoding::US_ASCII)
+      result.should_not.valid_encoding?
+    end
+
+    it "returns the same frozen strings for different encodings" do
+      str1 = "hello".dup.force_encoding(Encoding::US_ASCII)
+      str2 = "hello".dup.force_encoding(Encoding::UTF_8)
+      result1 = @s.rb_interned_str(str1, str1.bytesize)
+      result2 = @s.rb_interned_str(str2, str2.bytesize)
+      result1.should.equal?(result2)
+    end
+
+    it 'returns the same string when using non-ascii characters' do
+      str = 'こんにちは'
+      result1 = @s.rb_interned_str(str, str.bytesize)
+      result2 = @s.rb_interned_str(str, str.bytesize)
+      result1.should.equal?(result2)
+    end
+
+    it "returns the same string as String#-@" do
+      str = "hello".dup.force_encoding(Encoding::US_ASCII)
+      @s.rb_interned_str(str, str.bytesize).should.equal?(-str)
+    end
+  end
+
+  describe "rb_interned_str_cstr" do
+    it "returns a frozen string" do
+      str = "hello"
+      result = @s.rb_interned_str_cstr(str)
+      result.should.is_a?(String)
+      result.should.frozen?
+      result.encoding.should == Encoding::US_ASCII
+    end
+
+    it "returns the same frozen string" do
+      str = "hello"
+      result1 = @s.rb_interned_str_cstr(str)
+      result2 = @s.rb_interned_str_cstr(str)
+      result1.should.equal?(result2)
+    end
+
+    it "does not support strings with embedded null bytes" do
+      str = "foo\x00bar\x00baz".b
+      result = @s.rb_interned_str_cstr(str)
+      result.should == "foo"
+    end
+
+    it "support binary strings that are invalid in ASCII encoding" do
+      str = "foo\x81bar\x82baz".b
+      result = @s.rb_interned_str_cstr(str)
+      result.encoding.should == Encoding::US_ASCII
+      result.should == str.dup.force_encoding(Encoding::US_ASCII)
+      result.should_not.valid_encoding?
+    end
+
+    it "returns the same frozen strings for different encodings" do
+      str1 = "hello".dup.force_encoding(Encoding::US_ASCII)
+      str2 = "hello".dup.force_encoding(Encoding::UTF_8)
+      result1 = @s.rb_interned_str_cstr(str1)
+      result2 = @s.rb_interned_str_cstr(str2)
+      result1.should.equal?(result2)
+    end
+
+    it 'returns the same string when using non-ascii characters' do
+      str = 'こんにちは'
+      result1 = @s.rb_interned_str_cstr(str)
+      result2 = @s.rb_interned_str_cstr(str)
+      result1.should.equal?(result2)
+    end
+
+    it "returns the same string as String#-@" do
+      str = "hello".dup.force_encoding(Encoding::US_ASCII)
+      @s.rb_interned_str_cstr(str).should.equal?(-str)
+    end
+  end
 end