diaspora/spec/lib/message_handler_spec.rb

#   Copyright (c) 2010, Diaspora Inc.  This file is
#   licensed under the Affero General Public License version 3 or later.  See
#   the COPYRIGHT file.

require 'spec_helper'

describe MessageHandler do

  let(:handler) {MessageHandler.new()}  
  let(:message_body) {"I want to pump you up"} 
  let(:message_urls) {["http://www.google.com/", "http://yahoo.com/", "http://foo.com/"]}
  let(:success) {FakeHttpRequest.new(:success)}
  let(:failure) {FakeHttpRequest.new(:failure)} 

  before do
    handler.clear_queue
  end

  after(:all) do
    handler.clear_queue
  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
        EventMachine::HttpRequest.stub!(:new).and_return(success)

        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 the correct number of times' do
        failure.should_receive(:get).exactly(MessageHandler::NUM_TRIES).times.and_return(failure)
        EventMachine::HttpRequest.stub!(:new).and_return(failure)

        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::HttpRequest.stub!(:new).and_return(success)
      
      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::HttpRequest.stub!(:new).and_return(success)

      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
      success.should_receive(:post).and_return(success)
      EventMachine::HttpRequest.stub!(:new).and_return(success)

      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 "Hub publish" do
    before do
      EventMachine::PubSubHubbub.stub(:new).and_return(:success)
    end
    it 'should correctly queue up a pubsubhubbub publish request' do
      destination = "http://identi.ca/hub/"
      feed_location = "http://google.com/"

      EventMachine.run {
        handler.add_hub_notification(destination, feed_location)
        q = handler.instance_variable_get(:@queue)

        message = ""
        q.pop{|m| message = m}

        message.destination.should == destination
        message.body.should == feed_location

        EventMachine.stop
      }
    end
  end

  describe "Mixed Queries" do

    it 'should process both POST and GET requests in the same queue' do
      success.should_receive(:get).exactly(3).times.and_return(success)
      success.should_receive(:post).exactly(3).times.and_return(success)
      EventMachine::HttpRequest.stub!(:new).and_return(success)

      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
      success.should_receive(:get).exactly(1).times.and_return(success)
      success.should_receive(:post).exactly(1).times.and_return(success)

      EventMachine::HttpRequest.stub!(:new).and_return(success)

      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