lib/venom/generators/workshop.rb in venom-0.0.3 vs lib/venom/generators/workshop.rb in venom-0.0.4

@@ -33,13 +33,16 @@
       def create_workshop_structure
         empty_directory("#{name}/specs")
         empty_directory("#{name}/specs/support")
         if use_library
           empty_directory("#{name}/lib/test_steps")
+          empty_directory("#{name}/lib/test_data")
+          empty_directory("#{name}/lib/test_helpers")
         else
           empty_directory("#{name}/specs/support/test_steps")
+          empty_directory("#{name}/specs/support/test_data")
+          empty_directory("#{name}/specs/support/test_helpers")
         end
-        empty_directory("#{name}/specs/support/test_data")
       end
 
       def create_definitions_directory
         if use_library
           empty_directory("#{name}/lib/definitions")