Contents

require "yaml"

module Lignite
  class Connection
    # Errors that may happen when using a {Replay} connection.
    class ReplayError < RuntimeError
    end

    # Replays a recorded communication.
    # It checks that #send matches the stored sends, replays the #receive.
    class Replay < Connection
      include Bytes

      def initialize(filename)
        @filename = filename

        # [
        #  {"SEND" => "foo"},
        #  {"SEND" => "foo2"},
        #  {"RECV" => "adfafd"},
        #  {"SEND" => "foo2"},
        #  {"RECV" => "adfafd"}
        # ]
        @stream = YAML.load_file(filename)
      end

      # @param payload [ByteString]
      def send(payload)
        recorded = @stream.shift
        raise ReplayError, "Nothing left in the recording (#{@filename})" if recorded.nil?
        hex = recorded["SEND"]
        raise ReplayError, "Called SEND but the recording says RECV" if hex.nil?
        data = hex_to_bin(hex)
        return if data == payload

        details = "actual: #{bin_to_hex(payload)}, recorded: #{hex}"
        raise ReplayError, "Called SEND but the recorded data does not match: #{details}"
      end

      # @return [ByteString] a complete message
      def receive
        recorded = @stream.shift
        raise ReplayError, "Nothing left in the recording" if recorded.nil?
        hex = recorded["RECV"]
        raise ReplayError, "Called RECV but the recording says SEND" if hex.nil?
        hex_to_bin(hex)
      end

      def close
        super
        raise ReplayError, "Called close but the recording has leftover data" unless @stream.empty?
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
lignite-0.6.0	lib/lignite/connection/replay.rb
lignite-0.5.0	lib/lignite/connection/replay.rb
lignite-0.4.0	lib/lignite/connection/replay.rb