lib/cloudtasker/backend/memory_task.rb in cloudtasker-0.7.0 vs lib/cloudtasker/backend/memory_task.rb in cloudtasker-0.8.0

@@ -5,11 +5,11 @@
 module Cloudtasker
   module Backend
     # Manage local tasks pushed to memory.
     # Used for testing.
     class MemoryTask
-      attr_reader :id, :http_request, :schedule_time
+      attr_reader :id, :http_request, :schedule_time, :queue
 
       #
       # Return the task queue. A worker class name
       #
       # @return [Array<Hash>] <description>
@@ -114,14 +114,15 @@
       #
       # @param [String] id The ID of the task.
       # @param [Hash] http_request The HTTP request content.
       # @param [Integer] schedule_time When to run the task (Unix timestamp)
       #
-      def initialize(id:, http_request:, schedule_time: nil)
+      def initialize(id:, http_request:, schedule_time: nil, queue: nil)
         @id = id
         @http_request = http_request
         @schedule_time = Time.at(schedule_time || 0)
+        @queue = queue
       end
 
       #
       # Return task payload.
       #
@@ -147,10 +148,11 @@
       #
       def to_h
         {
           id: id,
           http_request: http_request,
-          schedule_time: schedule_time.to_i
+          schedule_time: schedule_time.to_i,
+          queue: queue
         }
       end
 
       #
       # Return the worker attached to this task.