testing strings along with binary strings

1 files changed, 47 insertions, 0 deletions

diff --git a/test/yaml/test_string.rb b/test/yaml/test_string.rb
new file mode 100644
index 0000000..7857672
--- /dev/null
+++ b/test/yaml/test_string.rb
@@ -0,0 +1,47 @@
+require 'test/unit'
+require 'psych'
+
+YAML = Psych
+
+module YAML
+  class TestString < Test::Unit::TestCase
+    def test_binary_string_null
+      string = "\x00"
+      yml = YAML.dump string
+      assert_match(/binary/, yml)
+      assert_equal string, YAML.load(yml)
+    end
+
+    def test_binary_string
+      string = binary_string
+      yml = YAML.dump string
+      assert_match(/binary/, yml)
+      assert_equal string, YAML.load(yml)
+    end
+
+    def test_non_binary_string
+      string = binary_string(0.29)
+      yml = YAML.dump string
+      refute_match(/binary/, yml)
+      assert_equal string, YAML.load(yml)
+    end
+
+    def test_string_with_ivars
+      food = "is delicious"
+      ivar = "on rock and roll"
+      food.instance_variable_set(:@we_built_this_city, ivar)
+
+      str = YAML.load YAML.dump food
+      assert_equal ivar, food.instance_variable_get(:@we_built_this_city)
+    end
+
+    def binary_string percentage = 0.31, length = 100
+      string = ''
+      (percentage * length).to_i.times do |i|
+        string << "\b"
+      end
+      string << 'a' * (length - string.length)
+      string
+    end
+  end
+end