Ruby EventMachine测试

时间:2013-05-21 14:39:29

标签: ruby rspec eventmachine

关于Ruby的第一个问题。 我正在尝试在Reactor循环中测试EventMachine交互 - 我想它可以被归类为“功能”测试。

假设我有两个类 - 服务器和客户端。我想测试双方 - 我需要确定他们的互动。

服务器:

require 'singleton'

class EchoServer < EM::Connection
  include EM::Protocols::LineProtocol

  def post_init
    puts "-- someone connected to the echo server!"
  end

  def receive_data data
    send_data ">>>you sent: #{data}"
    close_connection if data =~ /quit/i
  end

  def unbind
    puts "-- someone disconnected from the echo server!"
  end
end

客户端:

class EchoClient < EM::Connection
  include EM::Protocols::LineProtocol

  def post_init
    send_data "Hello"
  end

  def receive_data(data)
    @message = data
    p data
  end

  def unbind
    puts "-- someone disconnected from the echo server!"
  end
end

所以,我尝试了不同的方法并且没有提出任何建议。

根本问题是 - 我可以使用should_recive吗?

以某种方式使用RSpec测试我的代码

EventMachine参数应该是一个类或一个模块,所以我不能在里面发送实例化/模拟代码。正确?

这样的东西?

describe 'simple rspec test' do
  it 'should pass the test' do
    EventMachine.run {
      EventMachine::start_server "127.0.0.1", 8081, EchoServer
      puts 'running echo server on 8081'

      EchoServer.should_receive(:receive_data)

      EventMachine.connect '127.0.0.1', 8081, EchoClient

      EventMachine.add_timer 1 do
        puts 'Second passed. Stop loop.'
        EventMachine.stop_event_loop
      end
    }
  end
end

如果没有,你会怎么用EM :: SpecHelper做到这一点?我有这个代码使用它,无法弄清楚我做错了什么。

describe 'when server is run and client sends data' do
  include EM::SpecHelper

  default_timeout 2

  def start_server
    EM.start_server('0.0.0.0', 12345) { |ws|
      yield ws if block_given?
    }
  end

  def start_client
    client = EM.connect('0.0.0.0', 12345, FakeWebSocketClient)
    yield client if block_given?
    return client
  end

  describe "examples from the spec" do
    it "should accept a single-frame text message" do
      em {
        start_server

        start_client { |client|
          client.onopen {
            client.send_data("\x04\x05Hello")
          }
        }
      }
    end
  end
end

尝试了很多这些测试的变化,我只是想不出来。我确定我在这里遗漏了一些东西......

感谢您的帮助。

1 个答案:

答案 0 :(得分:4)

我能想到的最简单的解决方案是改变这个:

EchoServer.should_receive(:receive_data)

对此:

EchoServer.any_instance.should_receive(:receive_data)

由于EM期望一个类启动一个服务器,上面的any_instance技巧会期望该类的任何实例都接收该方法。

EMSpecHelper示例(虽然是官方/标准)非常令人费解,我宁愿坚持使用第一个rspec并使用any_instance,只是为了简单起见。