Question

我正在将ruby-mqtt gem包装到一个实现subscribe和publish方法的类中。 subscribe方法连接到服务器并在单独的线程中侦听，因为此调用是同步的。

module PubSub
class MQTT
    attr_accessor :host, :port, :username, :password

    def initialize(params = {})
        params.each do |attr, value|
            self.public_send("#{attr}=", value)
        end if params
        super()
    end

    def connection_options
        {
            remote_host: self.host,
            remote_port: self.port,
            username: self.username,
            password: self.password,
        }
    end

    def subscribe(name, &block)
        channel = name
        connect_opts = connection_options
        code_block = block
        ::Thread.new do
            ::MQTT::Client.connect(connect_opts) do |c|
                c.get(channel) do |topic, message|
                    puts "channel: #{topic} data: #{message.inspect}"
                    code_block.call topic, message
                end
            end
        end
    end

    def publish(channel = nil, data)
        ::MQTT::Client.connect(connection_options) do |c|
            c.publish(channel, data)
        end
    end     

end
end

我有一个测试，我用rspec来测试该类，但它没有通过。

mqtt = ::PubSub::MQTT.new({host: "localhost",port: 1883})
block = lambda { |channel, data|  puts "channel: #{channel} data: #{data.inspect}"}
block.should_receive(:call).with("channel", {"some" => "data"})
thr = mqtt.subscribe("channel", &block)
mqtt.publish("channel", {"some" => "data"})

当我运行以下ruby-mqtt-example时，我现在遇到了问题。

uri = URI.parse ENV['CLOUDMQTT_URL'] || 'mqtt://localhost:1883'
conn_opts = {
  remote_host: uri.host,
  remote_port: uri.port,
  username: uri.user,
  password: uri.password,
}

# Subscribe example
Thread.new do
  puts conn_opts
  MQTT::Client.connect(conn_opts) do |c|
    # The block will be called when you messages arrive to the topic
    c.get('test') do |topic, message|
      puts "#{topic}: #{message}"
    end
  end
end

# Publish example
puts conn_opts
MQTT::Client.connect(conn_opts) do |c|
  # publish a message to the topic 'test'
  loop do
    c.publish('test', 'Hello World')
    sleep 1
  end
end

所以我的问题是，当我简单地创建一个类并分离出发布和订阅逻辑时，我做错了什么？我的猜测是它与函数调用中的线程有关，但我似乎无法弄明白。非常感谢任何帮助。

更新

我相信我知道为什么测试没有通过，这是因为当我将lambda传递给subscribe期待它接收一个电话时它实际上不会在它退出时接到电话方法或直到publish被调用。所以我想将问题改为：我如何测试在一个线程中调用一个块？如果有人回答“你没有”，那么问题是：如何测试该块是在无限循环中调用的，就像在get gem中调用ruby-mqtt的例子一样。

Answer 1

RSpec期望机制可以在线程中正常工作，如以下示例所示，该示例通过：

def foo(&block)
  block.call(42)
end

describe "" do
  it "" do
    l = lambda {}
    expect(l).to receive(:call).with(42)
    Thread.new { foo(&l) }.join
  end
end

join等待线程完成后再继续。

如何测试在一个线程中调用一个块？

1 个答案: