lib/schematic/generator/sandbox.rb in schematic-0.4.2 vs lib/schematic/generator/sandbox.rb in schematic-0.4.3

@@ -1,14 +1,15 @@
 module Schematic
   module Generator
     class Sandbox
-      attr_accessor :ignored_elements, :added_elements
+      attr_accessor :ignored_elements, :added_elements, :required_elements
 
       def initialize(klass)
         @klass = klass
         @ignored_elements ||= []
         @added_elements ||= {}
+        @required_elements ||= []
       end
 
       def run(&block)
         instance_eval &block
       end
@@ -40,9 +41,13 @@
         if name.is_a? Hash
           added_elements[name.keys.first] = name.values.first
         else
           added_elements[name] = nil
         end
+      end
+
+      def required(*fields)
+        fields.each { |field| required_elements << field }
       end
 
       def method_missing(method, *args, &block)
         @klass.send method, *args, &block
       end