mirror of
https://github.com/codeninjasllc/discourse.git
synced 2024-12-04 21:01:27 -05:00
180 lines
4.4 KiB
Ruby
180 lines
4.4 KiB
Ruby
require 'spec_helper'
|
|
require 'message_bus'
|
|
require 'rack/test'
|
|
|
|
describe MessageBus::Rack::Middleware do
|
|
include Rack::Test::Methods
|
|
|
|
class FakeAsyncMiddleware
|
|
|
|
def self.in_async?
|
|
@@in_async if defined? @@in_async
|
|
end
|
|
|
|
def initialize(app,config={})
|
|
@app = app
|
|
end
|
|
|
|
def call(env)
|
|
result = nil
|
|
EM.run {
|
|
env['async.callback'] = lambda { |r|
|
|
# more judo with deferrable body, at this point we just have headers
|
|
r[2].callback do
|
|
# even more judo cause rack test does not call each like the spec says
|
|
body = ""
|
|
r[2].each do |m|
|
|
body << m
|
|
end
|
|
r[2] = [body]
|
|
result = r
|
|
end
|
|
}
|
|
catch(:async) {
|
|
result = @app.call(env)
|
|
}
|
|
|
|
EM::Timer.new(1) { EM.stop }
|
|
|
|
defer = lambda {
|
|
if !result
|
|
@@in_async = true
|
|
EM.next_tick do
|
|
defer.call
|
|
end
|
|
else
|
|
EM.next_tick { EM.stop }
|
|
end
|
|
}
|
|
defer.call
|
|
}
|
|
|
|
@@in_async = false
|
|
result || [500, {}, ['timeout']]
|
|
end
|
|
end
|
|
|
|
def app
|
|
@app ||= Rack::Builder.new {
|
|
use FakeAsyncMiddleware
|
|
use MessageBus::Rack::Middleware
|
|
run lambda {|env| [500, {'Content-Type' => 'text/html'}, 'should not be called' ]}
|
|
}.to_app
|
|
end
|
|
|
|
describe "long polling" do
|
|
before do
|
|
MessageBus.sockets_enabled = false
|
|
MessageBus.long_polling_enabled = true
|
|
end
|
|
|
|
it "should respond right away if dlp=t" do
|
|
post "/message-bus/ABC?dlp=t", '/foo1' => 0
|
|
FakeAsyncMiddleware.in_async?.should == false
|
|
last_response.should be_ok
|
|
end
|
|
|
|
it "should respond right away to long polls that are polling on -1 with the last_id" do
|
|
post "/message-bus/ABC", '/foo' => -1
|
|
last_response.should be_ok
|
|
parsed = JSON.parse(last_response.body)
|
|
parsed.length.should == 1
|
|
parsed[0]["channel"].should == "/__status"
|
|
parsed[0]["data"]["/foo"].should == MessageBus.last_id("/foo")
|
|
end
|
|
|
|
it "should respond to long polls when data is available" do
|
|
|
|
Thread.new do
|
|
wait_for(2000) { FakeAsyncMiddleware.in_async? }
|
|
MessageBus.publish "/foo", "bar"
|
|
end
|
|
|
|
post "/message-bus/ABC", '/foo' => nil
|
|
|
|
last_response.should be_ok
|
|
parsed = JSON.parse(last_response.body)
|
|
parsed.length.should == 1
|
|
parsed[0]["data"].should == "bar"
|
|
end
|
|
|
|
it "should timeout within its alloted slot" do
|
|
begin
|
|
MessageBus.long_polling_interval = 10
|
|
s = Time.now.to_f * 1000
|
|
post "/message-bus/ABC", '/foo' => nil
|
|
(Time.now.to_f * 1000 - s).should < 30
|
|
ensure
|
|
MessageBus.long_polling_interval = 5000
|
|
end
|
|
|
|
|
|
end
|
|
end
|
|
|
|
|
|
describe "polling" do
|
|
before do
|
|
MessageBus.sockets_enabled = false
|
|
MessageBus.long_polling_enabled = false
|
|
end
|
|
|
|
it "should respond with a 200 to a subscribe" do
|
|
client_id = "ABCD"
|
|
|
|
# client always keeps a list of channels with last message id they got on each
|
|
post "/message-bus/#{client_id}", {
|
|
'/foo' => nil,
|
|
'/bar' => nil
|
|
}
|
|
last_response.should be_ok
|
|
end
|
|
|
|
it "should correctly understand that -1 means stuff from now onwards" do
|
|
|
|
MessageBus.publish('foo', 'bar')
|
|
|
|
post "/message-bus/ABCD", {
|
|
'/foo' => -1
|
|
}
|
|
last_response.should be_ok
|
|
parsed = JSON.parse(last_response.body)
|
|
parsed.length.should == 1
|
|
parsed[0]["channel"].should == "/__status"
|
|
parsed[0]["data"]["/foo"].should == MessageBus.last_id("/foo")
|
|
|
|
end
|
|
|
|
it "should respond with the data if messages exist in the backlog" do
|
|
id = MessageBus.last_id('/foo')
|
|
|
|
MessageBus.publish("/foo", "barbs")
|
|
MessageBus.publish("/foo", "borbs")
|
|
|
|
client_id = "ABCD"
|
|
post "/message-bus/#{client_id}", {
|
|
'/foo' => id,
|
|
'/bar' => nil
|
|
}
|
|
|
|
parsed = JSON.parse(last_response.body)
|
|
parsed.length.should == 2
|
|
parsed[0]["data"].should == "barbs"
|
|
parsed[1]["data"].should == "borbs"
|
|
end
|
|
|
|
it "should not get consumed messages" do
|
|
MessageBus.publish("/foo", "barbs")
|
|
id = MessageBus.last_id('/foo')
|
|
|
|
client_id = "ABCD"
|
|
post "/message-bus/#{client_id}", {
|
|
'/foo' => id
|
|
}
|
|
|
|
parsed = JSON.parse(last_response.body)
|
|
parsed.length.should == 0
|
|
end
|
|
end
|
|
|
|
end
|