km-psych 0.1.0

data/test/psych/test_scalar.rb

@@ -0,0 +1,11 @@
+# -*- coding: utf-8 -*-
+
+require_relative 'helper'
+
+module Psych
+  class TestScalar < TestCase
+    def test_utf_8
+      assert_equal "日本語", Psych.load("--- 日本語")
+    end
+  end
+end

data/test/psych/test_scalar_scanner.rb

@@ -0,0 +1,70 @@
+require_relative 'helper'
+
+module Psych
+  class TestScalarScanner < TestCase
+    def test_scan_time
+      [ '2001-12-15T02:59:43.1Z',
+        '2001-12-14t21:59:43.10-05:00',
+        '2001-12-14 21:59:43.10 -5',
+        '2010-01-06 00:00:00 -08:00',
+        '2001-12-15 2:59:43.10',
+      ].each do |time|
+        ss = Psych::ScalarScanner.new
+        assert_instance_of Time, ss.tokenize(time)
+      end
+    end
+
+    attr_reader :ss
+
+    def setup
+      super
+      @ss = Psych::ScalarScanner.new
+    end
+
+    def test_scan_date
+      date = '1980-12-16'
+      token = @ss.tokenize date
+      assert_equal 1980, token.year
+      assert_equal 12, token.month
+      assert_equal 16, token.day
+    end
+
+    def test_scan_inf
+      assert_equal(1 / 0.0, ss.tokenize('.inf'))
+    end
+
+    def test_scan_minus_inf
+      assert_equal(-1 / 0.0, ss.tokenize('-.inf'))
+    end
+
+    def test_scan_nan
+      assert ss.tokenize('.nan').nan?
+    end
+
+    def test_scan_null
+      assert_equal nil, ss.tokenize('null')
+      assert_equal nil, ss.tokenize('~')
+      assert_equal nil, ss.tokenize('')
+    end
+
+    def test_scan_symbol
+      assert_equal :foo, ss.tokenize(':foo')
+    end
+
+    def test_scan_sexagesimal_float
+      assert_equal 685230.15, ss.tokenize('190:20:30.15')
+    end
+
+    def test_scan_sexagesimal_int
+      assert_equal 685230, ss.tokenize('190:20:30')
+    end
+
+    def test_scan_float
+      assert_equal 1.2, ss.tokenize('1.2')
+    end
+
+    def test_scan_true
+      assert_equal true, ss.tokenize('true')
+    end
+  end
+end

data/test/psych/test_serialize_subclasses.rb

@@ -0,0 +1,38 @@
+require_relative 'helper'
+
+module Psych
+  class TestSerializeSubclasses < TestCase
+    class SomeObject
+      def initialize one, two
+        @one = one
+        @two = two
+      end
+
+      def == other
+        @one == other.instance_eval { @one } &&
+          @two == other.instance_eval { @two }
+      end
+    end
+
+    def test_some_object
+      so = SomeObject.new('foo', [1,2,3])
+      assert_equal so, Psych.load(Psych.dump(so))
+    end
+
+    class StructSubclass < Struct.new(:foo)
+      def initialize foo, bar
+        super(foo)
+        @bar = bar
+      end
+
+      def == other
+        super(other) && @bar == other.instance_eval{ @bar }
+      end
+    end
+
+    def test_struct_subclass
+      so = StructSubclass.new('foo', [1,2,3])
+      assert_equal so, Psych.load(Psych.dump(so))
+    end
+  end
+end

data/test/psych/test_set.rb

@@ -0,0 +1,49 @@
+require_relative 'helper'
+
+module Psych
+  class TestSet < TestCase
+    def setup
+      super
+      @set = Psych::Set.new
+      @set['foo'] = 'bar'
+      @set['bar'] = 'baz'
+    end
+
+    def test_dump
+      assert_match(/!set/, Psych.dump(@set))
+    end
+
+    def test_roundtrip
+      assert_cycle(@set)
+    end
+
+    ###
+    # FIXME: Syck should also support !!set as shorthand
+    def test_load_from_yaml
+      loaded = Psych.load(<<-eoyml)
+--- !set
+foo: bar
+bar: baz
+      eoyml
+      assert_equal(@set, loaded)
+    end
+
+    def test_loaded_class
+      assert_instance_of(Psych::Set, Psych.load(Psych.dump(@set)))
+    end
+
+    def test_set_shorthand
+      loaded = Psych.load(<<-eoyml)
+--- !!set
+foo: bar
+bar: baz
+      eoyml
+      assert_instance_of(Psych::Set, loaded)
+    end
+
+    def test_set_self_reference
+      @set['self'] = @set
+      assert_cycle(@set)
+    end
+  end
+end

data/test/psych/test_stream.rb

@@ -0,0 +1,49 @@
+require_relative 'helper'
+
+module Psych
+  class TestStream < TestCase
+    def test_explicit_documents
+      io     = StringIO.new
+      stream = Psych::Stream.new(io)
+      stream.start
+      stream.push({ 'foo' => 'bar' })
+
+      assert !stream.finished?, 'stream not finished'
+      stream.finish
+      assert stream.finished?, 'stream finished'
+
+      assert_match(/^---/, io.string)
+      assert_match(/\.\.\.$/, io.string)
+    end
+
+    def test_start_takes_block
+      io     = StringIO.new
+      stream = Psych::Stream.new(io)
+      stream.start do |emitter|
+        emitter.push({ 'foo' => 'bar' })
+      end
+
+      assert stream.finished?, 'stream finished'
+      assert_match(/^---/, io.string)
+      assert_match(/\.\.\.$/, io.string)
+    end
+
+    def test_no_backreferences
+      io     = StringIO.new
+      stream = Psych::Stream.new(io)
+      stream.start do |emitter|
+        x = { 'foo' => 'bar' }
+        emitter.push x
+        emitter.push x
+      end
+
+      assert stream.finished?, 'stream finished'
+      assert_match(/^---/, io.string)
+      assert_match(/\.\.\.$/, io.string)
+      assert_equal 2, io.string.scan('---').length
+      assert_equal 2, io.string.scan('...').length
+      assert_equal 2, io.string.scan('foo').length
+      assert_equal 2, io.string.scan('bar').length
+    end
+  end
+end

data/test/psych/test_string.rb

@@ -0,0 +1,49 @@
+require_relative 'helper'
+
+module Psych
+  class TestString < TestCase
+    def test_binary_string_null
+      string = "\x00"
+      yml = Psych.dump string
+      assert_match(/binary/, yml)
+      assert_equal string, Psych.load(yml)
+    end
+
+    def test_binary_string
+      string = binary_string
+      yml = Psych.dump string
+      assert_match(/binary/, yml)
+      assert_equal string, Psych.load(yml)
+    end
+
+    def test_non_binary_string
+      string = binary_string(0.29)
+      yml = Psych.dump string
+      refute_match(/binary/, yml)
+      assert_equal string, Psych.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 = Psych.load Psych.dump food
+      assert_equal ivar, food.instance_variable_get(:@we_built_this_city)
+    end
+
+    def test_binary
+      string = [0, 123,22, 44, 9, 32, 34, 39].pack('C*')
+      assert_cycle string
+    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

data/test/psych/test_struct.rb

@@ -0,0 +1,51 @@
+require_relative 'helper'
+
+class PsychStructWithIvar < Struct.new(:foo)
+  attr_reader :bar
+  def initialize *args
+    super
+    @bar = 'hello'
+  end
+end
+
+module Psych
+  class TestStruct < TestCase
+    class StructSubclass < Struct.new(:foo)
+      def initialize foo, bar
+        super(foo)
+        @bar = bar
+      end
+    end
+
+    def test_self_referential_struct
+      ss = StructSubclass.new(nil, 'foo')
+      ss.foo = ss
+
+      loaded = Psych.load(Psych.dump(ss))
+      assert_instance_of(StructSubclass, loaded.foo)
+
+      # FIXME: This seems to cause an infinite loop.  wtf.  Must report a bug
+      # in ruby.
+      # assert_equal(ss, loaded)
+    end
+
+    def test_roundtrip
+      thing = PsychStructWithIvar.new('bar')
+      struct = Psych.load(Psych.dump(thing))
+
+      assert_equal 'hello', struct.bar
+      assert_equal 'bar', struct.foo
+    end
+
+    def test_load
+      obj = Psych.load(<<-eoyml)
+--- !ruby/struct:PsychStructWithIvar 
+:foo: bar
+:@bar: hello
+      eoyml
+
+      assert_equal 'hello', obj.bar
+      assert_equal 'bar', obj.foo
+    end
+  end
+end

data/test/psych/test_symbol.rb

@@ -0,0 +1,17 @@
+require_relative 'helper'
+
+module Psych
+  class TestSymbol < TestCase
+    def test_cycle
+      assert_cycle :a
+    end
+
+    def test_stringy
+      assert_cycle :"1"
+    end
+
+    def test_load_quoted
+      assert_equal :"1", Psych.load("--- :'1'\n")
+    end
+  end
+end

data/test/psych/test_to_yaml_properties.rb

@@ -0,0 +1,63 @@
+require_relative 'helper'
+
+module Psych
+  class TestToYamlProperties < MiniTest::Unit::TestCase
+    class Foo
+      attr_accessor :a, :b, :c
+      def initialize
+        @a = 1
+        @b = 2
+        @c = 3
+      end
+
+      def to_yaml_properties
+        [:@a, :@b]
+      end
+    end
+
+    def test_object_dump_yaml_properties
+      foo = Psych.load(Psych.dump(Foo.new))
+      assert_equal 1, foo.a
+      assert_equal 2, foo.b
+      assert_nil foo.c
+    end
+
+    class Bar < Struct.new(:foo, :bar)
+      attr_reader :baz
+      def initialize *args
+        super
+        @baz = 'hello'
+      end
+
+      def to_yaml_properties
+        []
+      end
+    end
+
+    def test_struct_dump_yaml_properties
+      bar = Psych.load(Psych.dump(Bar.new('a', 'b')))
+      assert_equal 'a', bar.foo
+      assert_equal 'b', bar.bar
+      assert_nil bar.baz
+    end
+
+    def test_string_dump
+      string = "okonomiyaki"
+      class << string
+        def to_yaml_properties
+          [:@tastes]
+        end
+      end
+
+      string.instance_variable_set(:@tastes, 'delicious')
+      v = Psych.load Psych.dump string
+      assert_equal 'delicious', v.instance_variable_get(:@tastes)
+    end
+
+    def test_string_load_syck
+      str = Psych.load("--- !str \nstr: okonomiyaki\n:@tastes: delicious\n")
+      assert_equal 'okonomiyaki', str
+      assert_equal 'delicious', str.instance_variable_get(:@tastes)
+    end
+  end
+end

data/test/psych/test_tree_builder.rb

@@ -0,0 +1,79 @@
+require_relative 'helper'
+
+module Psych
+  class TestTreeBuilder < TestCase
+    def setup
+      super
+      @parser = Psych::Parser.new TreeBuilder.new
+      @parser.parse(<<-eoyml)
+%YAML 1.1
+---
+- foo
+- {
+  bar : &A !!str baz,
+  boo : *A
+}
+- *A
+      eoyml
+      @tree = @parser.handler.root
+    end
+
+    def test_stream
+      assert_instance_of Nodes::Stream, @tree
+    end
+
+    def test_documents
+      assert_equal 1, @tree.children.length
+      assert_instance_of Nodes::Document, @tree.children.first
+      doc = @tree.children.first
+
+      assert_equal [1,1], doc.version
+      assert_equal [], doc.tag_directives
+      assert_equal false, doc.implicit
+    end
+
+    def test_sequence
+      doc = @tree.children.first
+      assert_equal 1, doc.children.length
+
+      seq = doc.children.first
+      assert_instance_of Nodes::Sequence, seq
+      assert_nil seq.anchor
+      assert_nil seq.tag
+      assert_equal true, seq.implicit
+      assert_equal Nodes::Sequence::BLOCK, seq.style
+    end
+
+    def test_scalar
+      doc = @tree.children.first
+      seq = doc.children.first
+
+      assert_equal 3, seq.children.length
+      scalar = seq.children.first
+      assert_instance_of Nodes::Scalar, scalar
+      assert_equal 'foo', scalar.value
+      assert_nil scalar.anchor
+      assert_nil scalar.tag
+      assert_equal true, scalar.plain
+      assert_equal false, scalar.quoted
+      assert_equal Nodes::Scalar::PLAIN, scalar.style
+    end
+
+    def test_mapping
+      doc = @tree.children.first
+      seq = doc.children.first
+      map = seq.children[1]
+
+      assert_instance_of Nodes::Mapping, map
+    end
+
+    def test_alias
+      doc = @tree.children.first
+      seq = doc.children.first
+      assert_equal 3, seq.children.length
+      al  = seq.children[2]
+      assert_instance_of Nodes::Alias, al
+      assert_equal 'A', al.anchor
+    end
+  end
+end