lib/rspec-system/node_set/base.rb in rspec-system-1.3.0 vs lib/rspec-system/node_set/base.rb in rspec-system-1.4.0

@@ -4,21 +4,23 @@
   #
   # @abstract Subclass and override methods to create a new NodeSet variant.
   class NodeSet::Base
     attr_reader :config
     attr_reader :setname
+    attr_reader :custom_prefabs_path
     attr_reader :nodes
     attr_reader :destroy
 
     # Create new NodeSet, populating necessary data structures.
-    def initialize(setname, config, options)
+    def initialize(setname, config, custom_prefabs_path, options)
       @setname = setname
       @config = config
+      @custom_prefabs_path = custom_prefabs_path
       @destroy = options[:destroy]
 
       @nodes = {}
       config['nodes'].each do |k,v|
-        @nodes[k] = RSpecSystem::Node.node_from_yaml(self, k, v)
+        @nodes[k] = RSpecSystem::Node.node_from_yaml(self, k, v, custom_prefabs_path)
       end
     end
 
     # Setup the NodeSet by starting all nodes.
     def setup