[rubygems/rubygems] add safe_load_marshal for gem::version and gem::spec

https://github.com/rubygems/rubygems/commit/2ea2ead1b3

1 files changed, 12 insertions, 0 deletions

diff --git a/spec/bundler/bundler/bundler_spec.rb b/spec/bundler/bundler/bundler_spec.rb
index 5b07f8159b..862a2c3e07 100644
--- a/spec/bundler/bundler/bundler_spec.rb
+++ b/spec/bundler/bundler/bundler_spec.rb
@@ -27,6 +27,18 @@ RSpec.describe Bundler do
       data = Marshal.dump(simple_structure)
       expect(Bundler.safe_load_marshal(data)).to eq(simple_structure)
     end
+
+    it "loads Gem::Version" do
+      gem_version = Gem::Version.new("3.7.2")
+      data = Marshal.dump(gem_version)
+      expect(Bundler.safe_load_marshal(data)).to eq(gem_version)
+    end
+
+    it "loads Gem::Specification" do
+      gem_spec = Gem::Specification.new("name", "3.7.2")
+      data = Marshal.dump(gem_spec)
+      expect(Bundler.safe_load_marshal(data)).to eq(gem_spec)
+    end
   end
 
   describe "#load_gemspec_uncached" do