Skip to content

Commit

Permalink
[rubygems/rubygems] Add a test for safe yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
segiddins authored and matzbot committed Feb 21, 2024
1 parent 5dcc7a0 commit 466ed0e
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions test/rubygems/test_gem_safe_yaml.rb
@@ -0,0 +1,23 @@
# frozen_string_literal: true

require_relative "helper"

Gem.load_yaml

class TestGemSafeYAML < Gem::TestCase
def test_aliases_enabled_by_default
assert_predicate Gem::SafeYAML, :aliases_enabled?
assert_equal({ "a" => "a", "b" => "a" }, Gem::SafeYAML.safe_load("a: &a a\nb: *a\n"))
end

def test_aliases_disabled
aliases_enabled = Gem::SafeYAML.aliases_enabled?
Gem::SafeYAML.aliases_enabled = false
refute_predicate Gem::SafeYAML, :aliases_enabled?
assert_raise Psych::AliasesNotEnabled do
Gem::SafeYAML.safe_load("a: &a\nb: *a\n")
end
ensure
Gem::SafeYAML.aliases_enabled = aliases_enabled
end
end

0 comments on commit 466ed0e

Please sign in to comment.