Return Gem::Version from Bundle#gem_version_from_gemfile_lock

Shopify · Sep 30, 2024 · a900bc4 · a900bc4
1 parent 16bc1a5
commit a900bc4
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/lib/spoom/context/bundle.rb b/lib/spoom/context/bundle.rb
@@ -58,9 +58,9 @@ def gemfile_lock_specs
       # Get `gem` version from the `Gemfile.lock` content
       #
       # Returns `nil` if `gem` cannot be found in the Gemfile.
-      sig { params(gem: String).returns(T.nilable(String)) }
+      sig { params(gem: String).returns(T.nilable(Gem::Version)) }
       def gem_version_from_gemfile_lock(gem)
-        gemfile_lock_specs[gem]&.version&.to_s
+        gemfile_lock_specs[gem]&.version
       end
     end
   end

diff --git a/lib/spoom/coverage.rb b/lib/spoom/coverage.rb
@@ -70,8 +70,8 @@ def snapshot(context, rbi: true, sorbet_bin: nil)
           end
         end
 
-        snapshot.version_static = context.gem_version_from_gemfile_lock("sorbet-static")
-        snapshot.version_runtime = context.gem_version_from_gemfile_lock("sorbet-runtime")
+        snapshot.version_static = context.gem_version_from_gemfile_lock("sorbet-static").to_s
+        snapshot.version_runtime = context.gem_version_from_gemfile_lock("sorbet-runtime").to_s
 
         files = context.srb_files(with_config: new_config)
         snapshot.rbi_files = files.count { |file| file.end_with?(".rbi") }

diff --git a/test/spoom/context/bundle_test.rb b/test/spoom/context/bundle_test.rb
@@ -143,15 +143,15 @@ def test_version_from_gemfile_lock_return_sorbet_version
         STR
         assert_equal(
           "0.5.5001",
-          context.gem_version_from_gemfile_lock("sorbet"),
+          context.gem_version_from_gemfile_lock("sorbet").to_s,
         )
         assert_equal(
           "0.5.5002",
-          context.gem_version_from_gemfile_lock("sorbet-runtime"),
+          context.gem_version_from_gemfile_lock("sorbet-runtime").to_s,
         )
         assert_equal(
           "0.5.5003",
-          context.gem_version_from_gemfile_lock("sorbet-static"),
+          context.gem_version_from_gemfile_lock("sorbet-static").to_s,
         )
       end
     end