km-psych 0.1.0

data/test/psych/test_document.rb

@@ -0,0 +1,46 @@
+require_relative 'helper'
+
+module Psych
+  class TestDocument < TestCase
+    def setup
+      super
+      @stream = Psych.parse_stream(<<-eoyml)
+%YAML 1.1
+%TAG ! tag:tenderlovemaking.com,2009:
+--- !fun
+      eoyml
+      @doc = @stream.children.first
+    end
+
+    def test_parse_tag
+      assert_equal([['!', 'tag:tenderlovemaking.com,2009:']],
+        @doc.tag_directives)
+    end
+
+    def test_emit_tag
+      assert_match('%TAG ! tag:tenderlovemaking.com,2009:', @stream.to_yaml)
+    end
+
+    def test_emit_multitag
+      @doc.tag_directives << ['!!', 'foo.com,2009:']
+      yaml = @stream.to_yaml
+      assert_match('%TAG ! tag:tenderlovemaking.com,2009:', yaml)
+      assert_match('%TAG !! foo.com,2009:', yaml)
+    end
+
+    def test_emit_bad_tag
+      assert_raises(RuntimeError) do
+        @doc.tag_directives = [['!']]
+        @stream.to_yaml
+      end
+    end
+
+    def test_parse_version
+      assert_equal([1,1], @doc.version)
+    end
+
+    def test_emit_version
+      assert_match('%YAML 1.1', @stream.to_yaml)
+    end
+  end
+end

data/test/psych/test_emitter.rb

@@ -0,0 +1,88 @@
+# -*- coding: utf-8 -*-
+
+require_relative 'helper'
+
+module Psych
+  class TestEmitter < TestCase
+    def setup
+      super
+      @out = StringIO.new('')
+      @emitter = Psych::Emitter.new @out
+    end
+
+    def test_set_canonical
+      @emitter.canonical = true
+      assert_equal true, @emitter.canonical
+
+      @emitter.canonical = false
+      assert_equal false, @emitter.canonical
+    end
+
+    def test_indentation_set
+      assert_equal 2, @emitter.indentation
+      @emitter.indentation = 5
+      assert_equal 5, @emitter.indentation
+    end
+
+    def test_emit_utf_8
+      @emitter.start_stream Psych::Nodes::Stream::UTF8
+      @emitter.start_document [], [], false
+      @emitter.scalar '日本語', nil, nil, false, true, 1
+      @emitter.end_document true
+      @emitter.end_stream
+      assert_match('日本語', @out.string)
+    end
+
+    def test_start_stream_arg_error
+      assert_raises(TypeError) do
+        @emitter.start_stream 'asdfasdf'
+      end
+    end
+
+    def test_start_doc_arg_error
+      @emitter.start_stream Psych::Nodes::Stream::UTF8
+
+      [
+        [nil, [], false],
+        [[nil, nil], [], false],
+        [[], 'foo', false],
+        [[], ['foo'], false],
+        [[], [nil,nil], false],
+      ].each do |args|
+        assert_raises(TypeError) do
+          @emitter.start_document(*args)
+        end
+      end
+    end
+
+    def test_scalar_arg_error
+      @emitter.start_stream Psych::Nodes::Stream::UTF8
+      @emitter.start_document [], [], false
+
+      [
+        [:foo, nil, nil, false, true, 1],
+        ['foo', Object.new, nil, false, true, 1],
+        ['foo', nil, Object.new, false, true, 1],
+        ['foo', nil, nil, false, true, :foo],
+        [nil, nil, nil, false, true, 1],
+      ].each do |args|
+        assert_raises(TypeError) do
+          @emitter.scalar(*args)
+        end
+      end
+    end
+
+    def test_start_sequence_arg_error
+      @emitter.start_stream Psych::Nodes::Stream::UTF8
+      @emitter.start_document [], [], false
+
+      assert_raises(TypeError) do
+        @emitter.start_sequence(nil, Object.new, true, 1)
+      end
+
+      assert_raises(TypeError) do
+        @emitter.start_sequence(nil, nil, true, :foo)
+      end
+    end
+  end
+end

data/test/psych/test_encoding.rb

@@ -0,0 +1,179 @@
+# -*- coding: utf-8 -*-
+
+require_relative 'helper'
+
+module Psych
+  class TestEncoding < TestCase
+    class EncodingCatcher < Handler
+      attr_reader :strings
+      def initialize
+        @strings = []
+      end
+
+      (Handler.instance_methods(true) -
+       Object.instance_methods).each do |m|
+        class_eval %{
+          def #{m} *args
+            @strings += args.flatten.find_all { |a|
+              String === a
+            }
+          end
+        }
+      end
+    end
+
+    def setup
+      super
+      @buffer  = StringIO.new
+      @handler = EncodingCatcher.new
+      @parser  = Psych::Parser.new @handler
+      @utf8    = Encoding.find('UTF-8')
+      @emitter = Psych::Emitter.new @buffer
+    end
+
+    def test_emit_alias
+      @emitter.start_stream Psych::Parser::UTF8
+      @emitter.start_document [], [], true
+      e = assert_raises(RuntimeError) do
+        @emitter.alias 'ドラえもん'.encode('EUC-JP')
+      end
+      assert_match(/alias value/, e.message)
+    end
+
+    def test_start_mapping
+      foo = 'foo'
+      bar = 'バー'
+
+      @emitter.start_stream Psych::Parser::UTF8
+      @emitter.start_document [], [], true
+      @emitter.start_mapping(
+        foo.encode('Shift_JIS'),
+        bar.encode('UTF-16LE'),
+        false, Nodes::Sequence::ANY)
+      @emitter.end_mapping
+      @emitter.end_document false
+      @emitter.end_stream
+
+      @parser.parse @buffer.string
+      assert_encodings @utf8, @handler.strings
+      assert_equal [foo, bar], @handler.strings
+    end
+
+    def test_start_sequence
+      foo = 'foo'
+      bar = 'バー'
+
+      @emitter.start_stream Psych::Parser::UTF8
+      @emitter.start_document [], [], true
+      @emitter.start_sequence(
+        foo.encode('Shift_JIS'),
+        bar.encode('UTF-16LE'),
+        false, Nodes::Sequence::ANY)
+      @emitter.end_sequence
+      @emitter.end_document false
+      @emitter.end_stream
+
+      @parser.parse @buffer.string
+      assert_encodings @utf8, @handler.strings
+      assert_equal [foo, bar], @handler.strings
+    end
+
+    def test_doc_tag_encoding
+      key = '鍵'
+      @emitter.start_stream Psych::Parser::UTF8
+      @emitter.start_document(
+        [1, 1],
+        [['!'.encode('EUC-JP'), key.encode('EUC-JP')]],
+        true
+      )
+      @emitter.scalar 'foo', nil, nil, true, false, Nodes::Scalar::ANY
+      @emitter.end_document false
+      @emitter.end_stream
+
+      @parser.parse @buffer.string
+      assert_encodings @utf8, @handler.strings
+      assert_equal key, @handler.strings[1]
+    end
+
+    def test_emitter_encoding
+      str  = "壁に耳あり、障子に目あり"
+      thing = Psych.load Psych.dump str.encode('EUC-JP')
+      assert_equal str, thing
+    end
+
+    def test_default_internal
+      before = Encoding.default_internal
+
+      Encoding.default_internal = 'EUC-JP'
+
+      str  = "壁に耳あり、障子に目あり"
+      yaml = "--- #{str}"
+      assert_equal @utf8, str.encoding
+
+      @parser.parse str
+      assert_encodings Encoding.find('EUC-JP'), @handler.strings
+      assert_equal str, @handler.strings.first.encode('UTF-8')
+    ensure
+      Encoding.default_internal = before
+    end
+
+    def test_scalar
+      @parser.parse("--- a")
+      assert_encodings @utf8, @handler.strings
+    end
+
+    def test_alias
+      @parser.parse(<<-eoyml)
+%YAML 1.1
+---
+!!seq [
+  !!str "Without properties",
+  &A !!str "Anchored",
+  !!str "Tagged",
+  *A,
+  !!str "",
+]
+      eoyml
+      assert_encodings @utf8, @handler.strings
+    end
+
+    def test_list_anchor
+      list = %w{ a b }
+      list << list
+      @parser.parse(Psych.dump(list))
+      assert_encodings @utf8, @handler.strings
+    end
+
+    def test_map_anchor
+      h = {}
+      h['a'] = h
+      @parser.parse(Psych.dump(h))
+      assert_encodings @utf8, @handler.strings
+    end
+
+    def test_map_tag
+      @parser.parse(<<-eoyml)
+%YAML 1.1
+---
+!!map { a : b }
+      eoyml
+      assert_encodings @utf8, @handler.strings
+    end
+
+    def test_doc_tag
+      @parser.parse(<<-eoyml)
+%YAML 1.1
+%TAG ! tag:tenderlovemaking.com,2009:
+--- !fun
+      eoyml
+      assert_encodings @utf8, @handler.strings
+    end
+
+    private
+    def assert_encodings encoding, strings
+      strings.each do |str|
+        assert_equal encoding, str.encoding, str
+      end
+    end
+  end
+end

data/test/psych/test_engine_manager.rb

@@ -0,0 +1,57 @@
+require_relative 'helper'
+require 'yaml'
+
+module Psych
+ class TestEngineManager < TestCase
+   def teardown
+     YAML::ENGINE.yamler = 'syck'
+   end
+
+   def test_bad_engine
+     assert_raises(ArgumentError) do
+       YAML::ENGINE.yamler = 'foooo'
+     end
+   end
+
+   def test_set_psych
+     YAML::ENGINE.yamler = 'psych'
+     assert_equal Psych, YAML
+     assert_equal 'psych', YAML::ENGINE.yamler
+   end
+
+   def test_set_syck
+     YAML::ENGINE.yamler = 'syck'
+     assert_equal Syck, YAML
+     assert_equal 'syck', YAML::ENGINE.yamler
+   end
+
+   A = Struct.new(:name)
+
+   def test_dump_types
+     YAML::ENGINE.yamler = 'psych'
+
+     assert_to_yaml ::Object.new
+     assert_to_yaml Time.now
+     assert_to_yaml Date.today
+     assert_to_yaml('a' => 'b')
+     assert_to_yaml A.new('foo')
+     assert_to_yaml %w{a b}
+     assert_to_yaml Exception.new('foo')
+     assert_to_yaml "hello!"
+     assert_to_yaml :fooo
+     assert_to_yaml(1..10)
+     assert_to_yaml(/hello!~/)
+     assert_to_yaml 1
+     assert_to_yaml 1.2
+     assert_to_yaml Rational(1, 2)
+     assert_to_yaml Complex(1, 2)
+     assert_to_yaml true
+     assert_to_yaml false
+     assert_to_yaml nil
+   end
+
+   def assert_to_yaml obj
+     assert obj.to_yaml, "#{obj.class} to_yaml works"
+   end
+ end
+end

data/test/psych/test_exception.rb

@@ -0,0 +1,39 @@
+require_relative 'helper'
+
+module Psych
+  class TestException < TestCase
+    class Wups < Exception
+      attr_reader :foo, :bar
+      def initialize *args
+        super
+        @foo = 1
+        @bar = 2
+      end
+    end
+
+    def setup
+      super
+      @wups = Wups.new
+    end
+
+    def test_convert
+      w = Psych.load(Psych.dump(@wups))
+      assert_equal @wups, w
+      assert_equal 1, w.foo
+      assert_equal 2, w.bar
+    end
+
+    def test_to_yaml_properties
+      class << @wups
+        def to_yaml_properties
+          [:@foo]
+        end
+      end
+
+      w = Psych.load(Psych.dump(@wups))
+      assert_equal @wups, w
+      assert_equal 1, w.foo
+      assert_nil w.bar
+    end
+  end
+end

data/test/psych/test_hash.rb

@@ -0,0 +1,30 @@
+require_relative 'helper'
+
+module Psych
+  class TestHash < TestCase
+    def setup
+      super
+      @hash = { :a => 'b' }
+    end
+
+    def test_self_referential
+      @hash['self'] = @hash
+      assert_cycle(@hash)
+    end
+
+    def test_cycles
+      assert_cycle(@hash)
+    end
+
+    def test_ref_append
+      hash = Psych.load(<<-eoyml)
+---
+foo: &foo
+  hello: world
+bar:
+  <<: *foo
+eoyml
+      assert_equal({"foo"=>{"hello"=>"world"}, "bar"=>{"hello"=>"world"}}, hash)
+    end
+  end
+end

data/test/psych/test_json_tree.rb

@@ -0,0 +1,43 @@
+require_relative 'helper'
+
+module Psych
+  class TestJSONTree < TestCase
+    def test_string
+      assert_match(/(['"])foo\1/, Psych.to_json("foo"))
+    end
+
+    def test_symbol
+      assert_match(/(['"])foo\1/, Psych.to_json(:foo))
+    end
+
+    def test_nil
+      assert_match(/^null/, Psych.to_json(nil))
+    end
+
+    def test_int
+      assert_match(/^10/, Psych.to_json(10))
+    end
+
+    def test_float
+      assert_match(/^1.2/, Psych.to_json(1.2))
+    end
+
+    def test_hash
+      hash = { 'one' => 'two' }
+      json = Psych.to_json(hash)
+      assert_match(/}$/, json)
+      assert_match(/^\{/, json)
+      assert_match(/['"]one['"]/, json)
+      assert_match(/['"]two['"]/, json)
+    end
+
+    def test_list_to_json
+      list = %w{ one two }
+      json = Psych.to_json(list)
+      assert_match(/]$/, json)
+      assert_match(/^\[/, json)
+      assert_match(/['"]one['"]/, json)
+      assert_match(/['"]two['"]/, json)
+    end
+  end
+end