fake_smith 0.5.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: cb87d276577cb4a8711926015ca52a49707b200f
-  data.tar.gz: 3a265d7f71bc8d99a7e6ebd48980267a97c3072b
+  metadata.gz: 2de2b0adf126522d621754afe65470839e4f114b
+  data.tar.gz: 3e5d68a2fb74f4065ad5fd098e40f6fd556b1e5c
 SHA512:
-  metadata.gz: df4d8b79f99c6aa551530546a6a3bb0587bb6107d06f3fd7425a9630dcc7f84e575c4aba93362f01b9439d684aa21551e2cda74a2f94d0a5f882ffa0c60d7bff
-  data.tar.gz: 0f5edb9b34929e2ead1298d6e2d66aaa507c635ac06b7512ad0a0652a30ed9943c3bb1bede28f5bd4adf7ce5d04ee72784d260346bd1c530e0b36f84313355ce
+  metadata.gz: ad868e516f31e7574b4e73d595314cf36759a2a9070302f5775041ef2f2ca38de7f9f9d9d3fb95c3cbb82c0d8767d36af80f2f216e2098ae93c87e2b42d07273
+  data.tar.gz: e7239578521cb08218545c23d1b21dafd2cdb09bcbf9d0f2d7a0e79fb3f28754575eae3454aca30c4e6743408374a8efef857f52dcc7f9742a13bd20cd2e0d8e

data/lib/fake_smith/version.rb

@@ -1,3 +1,3 @@
 class FakeSmith
-  VERSION = "0.5.0"
+  VERSION = "0.6.0"
 end

data/lib/fake_smith.rb

@@ -22,6 +22,15 @@ class FakeSmith
 
   class MessageAckedTwiceError < StandardError; end
 
+  def self.set_reply_handler(queue_name, &blk)
+    @reply_handlers ||= {}
+    @reply_handlers[queue_name] = blk
+  end
+
+  def self.reply_handlers
+    @reply_handlers
+  end
+
   def self.send_message(queue_name, payload, receiver)
     raise "no subscribers on queue: #{queue_name}" unless subscriptions[queue_name]
     receiver = ReceiverDecorator.new(receiver)
@@ -150,9 +159,19 @@ module Smith
         blk.call(self) if block_given?
       end
 
+      def on_reply(_opts, &blk)
+        @on_reply = blk
+      end
+
+      def on_timeout(&blk)
+      end
+
       def publish(message, &blk)
         FakeSmith.add_message(@queue_name, message)
         blk.call if block_given?
+        if FakeSmith.reply_handlers[@queue_name] && @on_reply
+          @on_reply.call(FakeSmith.reply_handlers[@queue_name].call(message))
+        end
       end
 
       def message_count(&blk)

data/spec/fake_smith_spec.rb

@@ -5,6 +5,7 @@ class MyAgent < Smith::Agent
     receiver("ack_twice_queue", :auto_ack => false).subscribe(&method(:ack_twice))
     receiver("ack_once_queue", :auto_ack => false).subscribe(&method(:ack_once))
     receiver("auto_ack_and_ack_queue", :auto_ack => true).subscribe(&method(:auto_ack_and_ack))
+    receiver("send_and_expect_reply", :auto_ack => true).subscribe(&method(:send_and_expect_reply))
   end
 
   def ack_twice(payload, receiver)
@@ -19,6 +20,18 @@ class MyAgent < Smith::Agent
   def auto_ack_and_ack(payload, receiver)
     receiver.ack
   end
+
+  def send_and_expect_reply(payload, receiver)
+    sender("replying_queue") do |q|
+      q.on_reply(:reply_queue_name => "somereplyqueuename") { |response| @replying_worked = true if response == {:correct => "response"} }
+      q.on_timeout { raise "replying epic fail" }
+      q.publish({:correct => "message"})
+    end
+  end
+
+  def replying_worked?
+    @replying_worked
+  end
 end
 
 describe FakeSmith do
@@ -45,4 +58,13 @@ describe FakeSmith do
       FakeSmith.send_message("ack_once_queue", message, receiver)
     end
   end
+
+  describe 'replying' do
+    it 'allows setting reply handlers' do
+      FakeSmith.set_reply_handler("replying_queue") do |send_message|
+        {:correct => "response"} if send_message == {:correct => "message"}
+      end
+      FakeSmith.send_message("send_and_expect_reply", message, receiver)
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: fake_smith
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.6.0
 platform: ruby
 authors:
 - Dylan Griffith
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-06-17 00:00:00.000000000 Z
+date: 2015-06-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler