Contents

module DeliveryBoy

  # A fake implementation that is useful for testing.
  class Fake
    FakeMessage = Struct.new(:value, :topic, :key, :offset, :partition, :partition_key)

    def initialize
      @messages = Hash.new {|h, k| h[k] = [] }
    end

    def deliver(value, topic:, key: nil, partition: nil, partition_key: nil)
      offset = @messages[topic].count
      message = FakeMessage.new(value, topic, key, offset, partition, partition_key)

      @messages[topic] << message

      nil
    end

    alias deliver_async! deliver

    def shutdown
      clear
    end

    # Clear all messages stored in memory.
    def clear
      @messages.clear
    end

    # Return all messages written to the specified topic.
    def messages_for(topic)
      @messages[topic]
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
delivery_boy-0.2.6	lib/delivery_boy/fake.rb
delivery_boy-0.2.5	lib/delivery_boy/fake.rb