spec/generator_spec.rb in keynote-0.2.3 vs spec/generator_spec.rb in keynote-0.3.0

@@ -13,10 +13,13 @@
     retval = Rails::Generators.invoke 'presenter', args,
       :behavior => :invoke,
       :destination_root => output_path
 
     assert retval, 'Generator must succeed'
-    yield retval.drop(2)
+
+    output_files = Dir["#{output_path}/**/*.rb"]
+
+    yield output_files.map { |path| path.sub("#{output_path}/", '') }.sort
   ensure
     FileUtils.rm_rf(output_path)
   end
 
   def output_path