152 lines
4.3 KiB
Ruby
152 lines
4.3 KiB
Ruby
require File.dirname(__FILE__) + '/../spec_helper'
|
|
|
|
describe MessageHandler do
|
|
before do
|
|
@handler = MessageHandler.new
|
|
@message_body = "I want to pump you up"
|
|
@message_urls = ["http://www.google.com/", "http://yahoo.com/", "http://foo.com/"]
|
|
end
|
|
|
|
describe 'GET messages' do
|
|
describe 'creating a GET query' do
|
|
it 'should be able to add a GET query to the queue with required destinations' do
|
|
EventMachine.run{
|
|
@handler.add_get_request(@message_urls)
|
|
@handler.size.should == @message_urls.size
|
|
EventMachine.stop
|
|
}
|
|
end
|
|
|
|
end
|
|
|
|
describe 'processing a GET query' do
|
|
it 'should remove sucessful http requests from the queue' do
|
|
request = FakeHttpRequest.new(:success)
|
|
request.should_receive(:get).and_return(request)
|
|
EventMachine::HttpRequest.stub!(:new).and_return(request)
|
|
|
|
EventMachine.run {
|
|
@handler.add_get_request("http://www.google.com/")
|
|
@handler.size.should == 1
|
|
@handler.process
|
|
@handler.size.should == 0
|
|
EventMachine.stop
|
|
}
|
|
end
|
|
|
|
|
|
it 'should only retry a bad request three times ' do
|
|
request = FakeHttpRequest.new(:failure)
|
|
request.should_receive(:get).exactly(MessageHandler::NUM_TRIES).times.and_return(request)
|
|
EventMachine::HttpRequest.stub!(:new).and_return(request)
|
|
|
|
EventMachine.run {
|
|
@handler.add_get_request("http://asdfsdajfsdfbasdj.com/")
|
|
@handler.size.should == 1
|
|
@handler.process
|
|
@handler.size.should == 0
|
|
|
|
EventMachine.stop
|
|
}
|
|
end
|
|
end
|
|
end
|
|
|
|
describe 'POST messages' do
|
|
|
|
|
|
it 'should be able to add a post message to the queue' do
|
|
EventMachine.run {
|
|
@handler.size.should ==0
|
|
@handler.add_post_request(@message_urls.first, @message_body)
|
|
@handler.size.should == 1
|
|
|
|
EventMachine.stop
|
|
}
|
|
end
|
|
|
|
it 'should be able to insert many posts into the queue' do
|
|
EventMachine.run {
|
|
@handler.size.should == 0
|
|
@handler.add_post_request(@message_urls, @message_body)
|
|
@handler.size.should == @message_urls.size
|
|
EventMachine.stop
|
|
}
|
|
end
|
|
|
|
it 'should post a single message to a given URL' do
|
|
request = FakeHttpRequest.new(:success)
|
|
request.should_receive(:post).and_return(request)
|
|
EventMachine::HttpRequest.stub!(:new).and_return(request)
|
|
EventMachine.run{
|
|
|
|
@handler.add_post_request(@message_urls.first, @message_body)
|
|
@handler.size.should == 1
|
|
@handler.process
|
|
@handler.size.should == 0
|
|
|
|
EventMachine.stop
|
|
|
|
}
|
|
end
|
|
end
|
|
|
|
describe "Mixed Queries" do
|
|
|
|
it 'should process both POST and GET requests in the same queue' do
|
|
request = FakeHttpRequest.new(:success)
|
|
request.should_receive(:get).exactly(3).times.and_return(request)
|
|
request.should_receive(:post).exactly(3).times.and_return(request)
|
|
EventMachine::HttpRequest.stub!(:new).and_return(request)
|
|
|
|
EventMachine.run{
|
|
@handler.add_post_request(@message_urls,@message_body)
|
|
@handler.size.should == 3
|
|
@handler.add_get_request(@message_urls)
|
|
@handler.size.should == 6
|
|
@handler.process
|
|
timer = EventMachine::Timer.new(1) do
|
|
@handler.size.should == 0
|
|
EventMachine.stop
|
|
end
|
|
}
|
|
end
|
|
|
|
it 'should be able to have seperate POST and GET have different callbacks' do
|
|
request = FakeHttpRequest.new(:success)
|
|
request.should_receive(:get).exactly(1).times.and_return(request)
|
|
request.should_receive(:post).exactly(1).times.and_return(request)
|
|
@handler.should_receive(:send_to_seed).once
|
|
|
|
EventMachine::HttpRequest.stub!(:new).and_return(request)
|
|
|
|
EventMachine.run{
|
|
@handler.add_post_request(@message_urls.first,@message_body)
|
|
@handler.add_get_request(@message_urls.first)
|
|
@handler.process
|
|
|
|
EventMachine.stop
|
|
}
|
|
|
|
end
|
|
end
|
|
end
|
|
|
|
class FakeHttpRequest
|
|
def initialize(callback_wanted)
|
|
@callback = callback_wanted
|
|
end
|
|
def response
|
|
"NOTE YOU ARE IN FAKE HTTP"
|
|
end
|
|
|
|
def post; end
|
|
def get; end
|
|
def callback(&b)
|
|
b.call if @callback == :success
|
|
end
|
|
def errback(&b)
|
|
b.call if @callback == :failure
|
|
end
|
|
end
|
|
|