Minor adjustment: use session_id instead of session

lib/utm_conversion/storage/base.rb

@@ -4,15 +4,15 @@ module UTMConversion
   module Storage
     # Base class for storage adapters
     class Base
-      def store(session, utm_params = nil)
+      def store(session_id, utm_params = nil)
         raise NotImplementedError
       end
 
       def retrieve(session_id)
         raise NotImplementedError
       end
 
-      def record_conversion(session, event_data)
+      def record_conversion(session_id, event_data)
         raise NotImplementedError
       end
     end

lib/utm_conversion/storage/in_memory.rb

@@ -11,12 +11,12 @@ def initialize
         @data = {}
       end
 
-      def store(session, utm_params = nil)
+      def store(session_id, utm_params = nil)
         utm_data = utm_params || utm_data(session)
-        return if utm_data.blank?
+        return if utm_data.nil? || utm_data == {}
 
-        @data[session.id] = {
-          utm_data: utm_params || utm_data(session),
+        @data[session_id] = {
+          utm_data: utm_data,
           conversions: []
         }
       end
@@ -25,16 +25,8 @@ def retrieve(session_id)
         @data[session_id]&.fetch(:utm_data, nil)
       end
 
-      def record_conversion(session, event_data)
-        return if utm_data(session).blank?
-
-        @data[session.id][:conversions] << event_data if @data[session_id]
-      end
-
-      private
-
-      def utm_data(session)
-        @utm_data ||= UTMConversion::Session::UTMData.load(session)
+      def record_conversion(session_id, event_data)
+        @data[session_id][:conversions] << event_data if @data[session_id]
       end
     end
   end