1 files changed, 80 insertions, 0 deletions
diff --git a/test/psych/test_array.rb b/test/psych/test_array.rb
new file mode 100644
index 0000000000..0dc82439d4
--- /dev/null
+++ b/test/psych/test_array.rb
@@ -0,0 +1,80 @@
+# frozen_string_literal: true
+require_relative 'helper'
+
+module Psych
+  class TestArray < TestCase
+    class X < Array
+    end
+
+    class Y < Array
+      attr_accessor :val
+    end
+
+    def setup
+      super
+      @list = [{ :a => 'b' }, 'foo']
+    end
+
+    def test_enumerator
+      x = [1, 2, 3, 4]
+      y = Psych.load Psych.dump x.to_enum
+      assert_equal x, y
+    end
+
+    def test_another_subclass_with_attributes
+      y = Y.new.tap {|o| o.val = 1}
+      y << "foo" << "bar"
+      y = Psych.unsafe_load Psych.dump y
+
+      assert_equal %w{foo bar}, y
+      assert_equal Y, y.class
+      assert_equal 1, y.val
+    end
+
+    def test_subclass
+      yaml = Psych.dump X.new
+      assert_match X.name, yaml
+
+      list = X.new
+      list << 1
+      assert_equal X, list.class
+      assert_equal 1, list.first
+    end
+
+    def test_subclass_with_attributes
+      y = Psych.unsafe_load Psych.dump Y.new.tap {|o| o.val = 1}
+      assert_equal Y, y.class
+      assert_equal 1, y.val
+    end
+
+    def test_backwards_with_syck
+      x = Psych.unsafe_load "--- !seq:#{X.name} []\n\n"
+      assert_equal X, x.class
+    end
+
+    def test_self_referential
+      @list << @list
+      assert_cycle(@list)
+    end
+
+    def test_recursive_array
+      @list << @list
+
+      loaded = Psych.load(Psych.dump(@list), aliases: true)
+
+      assert_same loaded, loaded.last
+    end
+
+    def test_recursive_array_uses_alias
+      @list << @list
+
+      assert_raise(AliasesNotEnabled) do
+        Psych.load(Psych.dump(@list), aliases: false)
+      end
+    end
+
+    def test_cycle
+      assert_cycle(@list)
+    end
+  end
+end