lib/longleaf/models/service_definition.rb in longleaf-0.1.0 vs lib/longleaf/models/service_definition.rb in longleaf-0.2.0.pre.1

@@ -1,20 +1,21 @@
 require_relative 'service_fields'
 
-# Definition of a preservation service
 module Longleaf
+  # Definition of a configured preservation service
   class ServiceDefinition
     attr_reader :name
-    attr_reader :work_script
+    attr_reader :work_script, :work_class
     attr_reader :frequency, :delay
     attr_reader :properties
     
-    def initialize(name:, work_script:, frequency: nil, delay: nil, properties: Hash.new)
+    def initialize(name:, work_script:, work_class: nil, frequency: nil, delay: nil, properties: Hash.new)
       raise ArgumentError.new("Parameters name and work_script are required") unless name && work_script
       
       @properties = properties
       @name = name
       @work_script = work_script
+      @work_class = work_class
       @frequency = frequency
       @delay = delay
     end
   end
 end
\ No newline at end of file