[ruby/prism] Parse all magic comments

https://github.com/ruby/prism/commit/2b3d59f424

1 files changed, 36 insertions, 0 deletions

diff --git a/test/prism/magic_comment_test.rb b/test/prism/magic_comment_test.rb
new file mode 100644
index 0000000000..c40364ccfa
--- /dev/null
+++ b/test/prism/magic_comment_test.rb
@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require_relative "test_helper"
+
+module Prism
+  class MagicCommentTest < TestCase
+    examples = [
+      "# encoding: ascii",
+      "# coding: ascii",
+      "# eNcOdInG: ascii",
+      "# CoDiNg: ascii",
+      "# \s\t\v encoding \s\t\v : \s\t\v ascii \s\t\v",
+      "# -*- encoding: ascii -*-",
+      "# -*- coding: ascii -*-",
+      "# -*- eNcOdInG: ascii -*-",
+      "# -*- CoDiNg: ascii -*-",
+      "# -*- \s\t\v encoding \s\t\v : \s\t\v ascii \s\t\v -*-",
+      "# -*- foo: bar; encoding: ascii -*-",
+      "# coding \t \r  \v   :     \t \v    \r   ascii-8bit\n"
+    ]
+
+    examples.each do |example|
+      define_method(:"test_magic_comment_#{example}") do
+        assert_magic_comment(example)
+      end
+    end
+
+    private
+
+    def assert_magic_comment(example)
+      expected = Ripper.new(example).tap(&:parse).encoding
+      actual = Prism.parse(example).source.source.encoding
+      assert_equal expected, actual
+    end
+  end
+end