test_word_count.rb in documentcloud-cloud-crowd-0.2.1

- old
+ new

@@ -3,20 +3,17 @@
 class WordCountTest < Test::Unit::TestCase
   
   context "the word_count action" do
     
     setup do
+      WorkUnit.expects(:distribute_to_nodes).returns(true)
       @asset_store = AssetStore.new
       @browser = Rack::Test::Session.new(Rack::MockSession.new(CloudCrowd::Server))
       @browser.put('/worker', :name => 'test_worker', :thread_status => 'sleeping')
       post_job_to_count_words_in_this_file
       @job_id = JSON.parse(@browser.last_response.body)['id']
     end
-    
-    teardown do
-      CloudCrowd::Job.destroy_all
-    end
 
     should "be able to create a word_count job" do
       assert @browser.last_response.ok? 
       info = JSON.parse(@browser.last_response.body)
       assert info['status'] == 'processing'
@@ -24,10 +21,10 @@
     end
     
     should "be able to perform the processing stage of a word_count" do
       action = CloudCrowd.actions['word_count'].new(1, "file://#{File.expand_path(__FILE__)}", {}, @asset_store)
       count = action.process
-      assert count == 104
+      assert count == 101
     end
     
   end
   
   def post_job_to_count_words_in_this_file