test/test_rabbitmq.rb in waffle-0.1.3 vs test/test_rabbitmq.rb in waffle-0.2.1

@@ -1,23 +1,20 @@
 require 'test/unit'
-require 'waffle/configuration'
-require 'waffle/strategies/rabbitmq'
+require 'mocha'
+require 'bunny'
+require 'waffle/transports/rabbitmq'
 
 class RabbitmqTest < Test::Unit::TestCase
 
-  def test_bad_initialization
-    assert_raise ArgumentError do
-      Waffle::Strategies::Rabbitmq.new
-    end
-
-    assert_raise ArgumentError do
-      Waffle::Strategies::Rabbitmq.new 'crap'
-    end
+  def test_publish
+    Bunny.setup nil
+    Bunny::Exchange.any_instance.stubs(:publish).returns(nil)
+    assert_equal nil, Waffle::Transports::Rabbitmq.new.publish('events', 'message')
   end
 
-  def test_good_initialization
-    assert_nothing_raised do
-      Waffle::Strategies::Rabbitmq.new Waffle::Configuration.new
-    end
+  def test_subscribe
+    Bunny.setup nil
+    Bunny::Queue.any_instance.stubs(:subscribe).returns(nil)
+    assert_equal nil, Waffle::Transports::Rabbitmq.new.subscribe{ |m| puts m }
   end
 
 end