spec/invoker_spec.rb in awestruct-0.5.4.beta1 vs spec/invoker_spec.rb in awestruct-0.5.4.rc

@@ -62,9 +62,10 @@
     invoker.should_not_receive( :invoke_server )
     invoker.invoke!
   end
 
   it "should return false on failure" do
+    require 'awestruct/cli/generate'
     generator = mock
     Awestruct::CLI::Generate.should_receive( :new ).and_return( generator )
     generator.should_receive( :run ).and_return( false )
     Awestruct::CLI::Invoker.new( '--generate' ).invoke!.should be_false
   end