Correctly fix eventmachine#228 - race condition in test.

caichengjun01 · Aug 1, 2011 · eeb7f12 · eeb7f12
1 parent e510f57
commit eeb7f12
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/tests/test_threaded_resource.rb b/tests/test_threaded_resource.rb
@@ -14,12 +14,15 @@ def teardown
   end
 
   def test_dispatch_completion
-    completion = resource.dispatch do |o|
-      o[:foo] = :bar
-      :foo
-    end
-    completion.callback do |result|
-      assert_equal :foo, result
+    EM.run do
+      completion = resource.dispatch do |o|
+        o[:foo] = :bar
+        :foo
+      end
+      completion.callback do |result|
+        assert_equal :foo, result
+        EM.stop
+      end
     end
     assert_equal :bar, object[:foo]
   end