sclemmer-robut 0.5.2

data/test/unit/connection_test.rb

@@ -0,0 +1,161 @@
+require 'test_helper'
+require 'robut/plugin/echo'
+
+class SimplePlugin
+  include Robut::Plugin
+  attr_accessor :run
+
+  def initialize(*args)
+    super(*args)
+    @run = false
+  end
+
+  def handle(*args)
+    self.run = true
+  end
+end
+
+class ReplyToUserPlugin
+  include Robut::Plugin
+
+  def initialize(*args)
+    super(*args)
+  end
+
+  def handle(time, nick, message)
+    reply("Reply", nick)
+  end
+end
+
+class ReplyToRoomPlugin
+  include Robut::Plugin
+
+  def initialize(*args)
+    super(*args)
+  end
+
+  def handle(time, nick, message)
+    reply("Reply", :room)
+  end
+end
+
+class ReplyMock
+  attr_accessor :messages
+
+  def initialize
+    @messages = []
+  end
+
+  def send(message)
+    @messages << message
+  end
+
+  def room
+    "my_room@test.com"
+  end
+
+  def add_message_callback(code, reply_to)
+    true
+  end
+
+  def join(path)
+    true
+  end
+end
+
+class RoomMock < Robut::Room
+
+  def initialize(connection)
+    @muc = ReplyMock.new
+    @connection = connection
+  end
+
+end
+
+class PMMock < Robut::PM
+
+  def initialize(connection)
+    @connection = connection
+  end
+end
+
+
+class ConnectionTest < Test::Unit::TestCase
+
+  def setup
+    Robut::Plugin.plugins = [SimplePlugin]
+    @connection = Robut::Connection.new({
+        :jid => 'abc@def.com',
+        :nick => "Test Robut",
+        :mention_name => 'test'
+      })
+  end
+
+  def test_end_to_end_message
+    Robut::Plugin.plugins = [Robut::Plugin::Echo]
+    justin = Jabber::JID.new('justin@example.com')
+    @connection.roster = mock_roster(justin)
+    room = RoomMock.new(@connection)
+    room.handle_message(plugins(room), Time.now, 'Justin Weiss', '@test echo Test Message')
+    message = room.muc.messages.first
+    assert_equal(room.muc.room, message.to.to_s)
+    assert_equal("Test Message", message.body)
+  end
+
+  def test_handle_message_delegates_to_plugin
+    presence = Robut::Presence.new(@connection)
+    plugins = plugins(presence)
+    assert !plugins.first.run, "The plugin was not set up correctly."
+
+    presence.handle_message(plugins, Time.now, 'Justin', 'Test Message')
+    assert plugins.first.run, "The plugin's handle_message method should have been run"
+  end
+
+  def test_handle_message_from_person
+    Robut::Plugin.plugins = [Robut::Plugin::Echo]
+    sender = Jabber::JID.new('justin@example.com')
+    @connection.client = ReplyMock.new
+    pm = PMMock.new(@connection)
+
+    pm.handle_message(plugins(pm, sender), Time.now, 'Justin', '@test echo Test Message')
+    message = pm.connection.client.messages.first
+    assert_equal(sender, message.to)
+    assert_equal("Test Message", message.body)
+  end
+
+  def test_reply_directly_to_user
+    Robut::Plugin.plugins = [ReplyToUserPlugin]
+    @connection.client = ReplyMock.new
+    justin = Jabber::JID.new('justin@example.com')
+    @connection.roster = mock_roster(justin)
+    pm = Robut::PM.new(@connection, justin)
+
+    pm.handle_message(plugins(pm), Time.now, 'justin WEISS', 'Test Message')
+    message = pm.connection.client.messages.first
+    assert_equal(justin, message.to.to_s)
+    assert_equal(:chat, message.type)
+    assert_equal("Reply", message.body)
+  end
+
+  def test_reply_directly_to_room
+    Robut::Plugin.plugins = [ReplyToRoomPlugin]
+    sender = Jabber::JID.new('justin@example.com')
+    room = RoomMock.new(@connection)
+
+    room.handle_message(plugins(room, sender), Time.now, 'Justin', '@test echo Test Message')
+    message = room.muc.messages.first
+    assert_equal(room.muc.room, message.to.to_s)
+    assert_equal("Reply", message.body)
+  end
+
+  private
+
+  def plugins(presence, sender = nil)
+    Robut::Plugin.plugins.map { |p| p.new(presence, sender) }
+  end
+
+  def mock_roster(jid)
+    OpenStruct.new( :items => { jid => OpenStruct.new(:iname => "Justin Weiss") } )
+  end
+
+end

data/test/unit/plugin/alias_test.rb

@@ -0,0 +1,76 @@
+require 'test_helper'
+require 'robut/plugin/alias'
+require 'robut/plugin/echo'
+
+class Robut::Plugin::AliasTest < Test::Unit::TestCase
+
+  def setup
+    @connection = Robut::ConnectionMock.new
+    @presence = Robut::PresenceMock.new(@connection)
+    @plugin = Robut::Plugin::Alias.new(@presence)
+    @plugin.aliases = {}
+  end
+  
+  def test_aliases_this_to_that
+    @plugin.handle(Time.now, "@john", "@robut alias w weather?")
+    assert_equal 'weather?', @plugin.aliases['w']
+  end
+  
+  def test_thinks_this_is_that
+    @plugin.handle(Time.now, "@john", "@robut alias this @robut echo that")
+    @plugin.handle(Time.now, "@john", "this")
+    message = @plugin.reply_to.messages.first
+    assert_equal "@john", message[1]
+    assert_equal "@robut echo that", message[2]
+  end
+  
+  def test_doesnt_alias_when_it_shouldnt
+    @plugin.handle(Time.now, "@john", "@robut somthing alias w weather?")
+    assert @plugin.aliases.empty?
+  end  
+  
+  def test_can_alias_with_quotes
+    @plugin.handle(Time.now, "@john", '@robut alias "long string" "some other long string"')
+    assert_equal 'some other long string', @plugin.aliases['long string']
+  end
+  
+  def test_can_apply_aliases_with_quotes
+    @plugin.handle(Time.now, "@john", '@robut alias "long string" "some other long string"')
+    @plugin.handle(Time.now, "@john", "long string")
+    message = @plugin.reply_to.messages.first
+    assert_equal "@john", message[1]
+    assert_equal "some other long string", message[2]    
+  end
+  
+  def test_aliases_can_be_removed
+    @plugin.handle(Time.now, "@john", "@robut alias this that")
+    @plugin.handle(Time.now, "@john", "@robut remove alias this")
+    assert @plugin.aliases.empty?
+  end
+  
+  def test_remove_aliases_doesnt_choke_on_missing_key
+    @plugin.handle(Time.now, "@john", "@robut alias this that")
+    @plugin.handle(Time.now, "@john", "@robut remove alias")
+    assert_equal({'this' => 'that'}, @plugin.aliases)
+  end
+  
+  def test_remove_alias_handles_quotes
+    @plugin.handle(Time.now, "@john", '@robut alias "long string" "that"')
+    @plugin.handle(Time.now, "@john", '@robut remove alias "long string"')
+    assert @plugin.aliases.empty?
+  end
+  
+  def test_can_list_all_aliases
+    @plugin.handle(Time.now, "@john", "@robut alias this that")
+    @plugin.handle(Time.now, "@john", "@robut alias something something else")
+    @plugin.handle(Time.now, "@john", "@robut aliases")
+    assert_equal ["something => something else\nthis => that"], @plugin.reply_to.replies
+  end
+  
+  def test_can_clear_aliases
+    @plugin.handle(Time.now, "@john", "@robut alias this that")
+    @plugin.handle(Time.now, "@john", "@robut clear aliases")
+    assert_equal({}, @plugin.aliases)
+  end
+  
+end

data/test/unit/plugin/echo_test.rb

@@ -0,0 +1,27 @@
+require 'test_helper'
+require 'robut/plugin/echo'
+
+class Robut::Plugin::EchoTest < Test::Unit::TestCase
+
+  def setup
+    @connection = Robut::ConnectionMock.new
+    @presence = Robut::PresenceMock.new(@connection)
+    @plugin = Robut::Plugin::Echo.new(@presence)
+  end
+
+  def test_replies_with_this
+    @plugin.handle(Time.now, "@john", "@robut echo this")
+    assert_equal ["this"], @plugin.reply_to.replies
+  end
+
+  def test_replies_with_nicks
+    @plugin.handle(Time.now, "@john", "@robut echo @justin look over here!")
+    assert_equal ["@justin look over here!"], @plugin.reply_to.replies
+  end
+  
+  def test_doesnt_reply_with_empty
+    @plugin.handle(Time.now, "@john", "@robut echo")
+    assert_equal [], @plugin.reply_to.replies
+  end
+  
+end

data/test/unit/plugin/help_test.rb

@@ -0,0 +1,46 @@
+require 'test_helper'
+require 'robut/plugin/echo'
+require 'robut/plugin/help'
+
+class Robut::Plugin::PluginWithoutHelp
+  include Robut::Plugin
+
+  def usage
+    super
+  end
+end
+
+class Robut::Plugin::HelpTest < Test::Unit::TestCase
+
+  def setup
+    @connection = Robut::ConnectionMock.new
+    @presence = Robut::PresenceMock.new(@connection)
+    Robut::Plugin.plugins << Robut::Plugin::Echo
+    Robut::Plugin.plugins << Robut::Plugin::Help
+    @plugin = Robut::Plugin::Help.new(@presence)
+  end
+
+  def teardown
+    Robut::Plugin.plugins = []
+  end
+
+  def test_help
+    @plugin.handle(Time.now, "@justin", "@robut help")
+    assert_equal [
+      "Supported commands:",
+      "@robut echo <message> - replies to the channel with <message>",
+      "@robut help - displays this message",
+    ], @plugin.reply_to.replies
+  end
+
+  def test_empty_help
+    Robut::Plugin.plugins << Robut::Plugin::PluginWithoutHelp
+    @plugin.handle(Time.now, "@justin", "@robut help")
+    assert_equal [
+      "Supported commands:",
+      "@robut echo <message> - replies to the channel with <message>",
+      "@robut help - displays this message",
+    ], @plugin.reply_to.replies
+  end
+end
+  

data/test/unit/plugin/later_test.rb

@@ -0,0 +1,40 @@
+require 'test_helper'
+require 'robut/plugin/later'
+
+class Robut::Plugin::LaterTest < Test::Unit::TestCase
+
+  def setup
+    @connection = Robut::ConnectionMock.new
+    @presence = Robut::PresenceMock.new(@connection)
+    @plugin = Robut::Plugin::Later.new(@presence)
+    @plugin.instance_eval do
+      def threader; yield; end # no threads
+      def sleep(*prms); end # just skip the whole sleeping part
+    end
+  end
+
+  def test_replies_with_minutes
+    @plugin.handle(Time.now, "@john", "@robut in 0 minutes msg me")
+    assert_equal ["Ok, see you in 0 minutes"], @plugin.reply_to.replies
+    message = @plugin.reply_to.messages.first
+    assert message
+    assert_equal message[1], "@john"
+    assert_equal message[2], "@robut msg me"
+  end
+
+  def test_replies_with_sec
+    @plugin.handle(Time.now, "@john", "@robut in 1 sec msg me")
+    assert_equal ["Ok, see you in 1 sec"], @plugin.reply_to.replies
+  end
+  
+  def test_replies_with_hr
+    @plugin.handle(Time.now, "@john", "@robut in 1 hr msg me")
+    assert_equal ["Ok, see you in 1 hr"], @plugin.reply_to.replies
+  end
+
+  def test_replies_with_hrs
+    @plugin.handle(Time.now, "@john", "@robut in 2 hrs msg me")
+    assert_equal ["Ok, see you in 2 hrs"], @plugin.reply_to.replies
+  end
+  
+end

data/test/unit/plugin/lunch_test.rb

@@ -0,0 +1,36 @@
+require 'test_helper'
+require 'robut/plugin/lunch'
+
+class Robut::Plugin::LunchTest < Test::Unit::TestCase
+
+  def setup
+    @connection = Robut::ConnectionMock.new
+    @presence = Robut::PresenceMock.new(@connection)
+    @plugin = Robut::Plugin::Lunch.new(@presence)
+    @plugin.places = ["Pho"]
+  end
+
+  def test_handle_returns_pho_for_lunch
+    @plugin.handle(Time.now, "John", "lunch?")
+    assert_equal ["Pho!"], @plugin.reply_to.replies
+  end
+  
+  def test_handle_returns_all_places_for_lunch_places
+    @plugin.new_place("Teriyaki")
+    @plugin.handle(Time.now, "John", "@robut lunch places")
+    assert_equal ["Pho, Teriyaki"], @plugin.reply_to.replies
+  end
+  
+  def test_handle_new_lunch_place
+    @plugin.handle(Time.now, "John", "@robut new lunch place Green Leaf")
+    assert_equal ["Ok, I'll add \"Green Leaf\" to the the list of lunch places"], @plugin.reply_to.replies
+    assert @plugin.places.include?("Green Leaf")
+  end
+  
+  def test_handle_remove_lunch_place
+    @plugin.handle(Time.now, "John", "@robut remove lunch place Pho")
+    assert_equal ["I removed \"Pho\" from the list of lunch places"], @plugin.reply_to.replies
+    assert @plugin.places.empty?
+  end
+
+end

data/test/unit/plugin/pick_test.rb

@@ -0,0 +1,35 @@
+require 'test_helper'
+require 'robut/plugin/pick'
+require 'mocha'
+
+class Robut::Plugin::PickTest < Test::Unit::TestCase
+
+  def setup
+    @connection = Robut::ConnectionMock.new
+    @presence = Robut::PresenceMock.new(@connection)
+    @plugin = Robut::Plugin::Pick.new(@presence)
+  end
+
+  def test_replies_with_correct_response
+    @plugin.stubs(:random).returns(0)
+    @plugin.handle(Time.now, "@john", "@robut pick a, b, c")
+    assert_equal ["And the winner is... a"], @plugin.reply_to.replies
+  end
+  
+  def test_does_nothing_when_no_options_are_given
+    @plugin.handle(Time.now, "@john", "@robut pick")
+    assert @plugin.reply_to.replies.empty?
+  end
+
+  def test_replies_only_option_if_given_one_options
+    @plugin.handle(Time.now, "@john", "@robut pick a")
+    assert_equal ["And the winner is... a"], @plugin.reply_to.replies
+  end
+  
+  def test_handles_spaces
+    @plugin.stubs(:random).returns(1)
+    @plugin.handle(Time.now, "@john", "@robut pick this is the first option, this is the second option, this is the third option")
+    assert_equal ["And the winner is... this is the second option"], @plugin.reply_to.replies
+  end
+  
+end

data/test/unit/plugin/ping_test.rb

@@ -0,0 +1,22 @@
+require 'test_helper'
+require 'robut/plugin/ping'
+
+class Robut::Plugin::PingTest < Test::Unit::TestCase
+
+  def setup
+    @connection = Robut::ConnectionMock.new
+    @presence = Robut::PresenceMock.new(@connection)
+    @plugin = Robut::Plugin::Ping.new(@presence)
+  end
+
+  def test_replies_wih_pong
+    @plugin.handle(Time.now, "@john", "@robut ping")
+    assert_equal ["pong"], @plugin.reply_to.replies
+  end
+
+  def test_replies_wih_sec
+    @plugin.handle(Time.now, "@john", "@robut ping pong")
+    assert_equal [], @plugin.reply_to.replies
+  end
+  
+end

data/test/unit/plugin/quips_test.rb

@@ -0,0 +1,58 @@
+require 'test_helper'
+require 'robut/plugin/quips'
+
+class Robut::Plugin::QuipsTest < Test::Unit::TestCase
+
+  def setup
+    @connection = Robut::ConnectionMock.new
+    @presence = Robut::PresenceMock.new(@connection)
+    @plugin = Robut::Plugin::Quips.new(@presence)
+    @plugin.store["quips"] = ["the fancier i try and make it, the more complicated it gets!", "It works on my machine ..."]
+  end
+
+  def test_add_quip_adds_a_quip_to_the_database
+    @plugin.handle(Time.now, "John", "@robut add quip So, are we self managing now?")
+    assert_equal ["I added the quip to the quip database"], @plugin.reply_to.replies
+    assert_equal "So, are we self managing now?", @plugin.quips.last
+  end
+
+  def test_add_quip_without_a_quip_is_ignored
+    @plugin.handle(Time.now, "John", "@robut add quip ")
+    assert_equal [], @plugin.reply_to.replies
+    assert_equal 2, @plugin.quips.length
+  end
+
+  def test_add_quip_doesnt_add_duplicates
+    @plugin.handle(Time.now, "John", "@robut add quip the fancier i try and make it, the more complicated it gets!")
+    assert_equal ["I didn't add the quip, since it was already added"], @plugin.reply_to.replies
+    assert_equal 2, @plugin.quips.length
+  end
+
+  def test_remove_quip
+    @plugin.handle(Time.now, "John", "@robut remove quip the fancier i try and make it, the more complicated it gets!")
+    assert_equal ["I removed the quip from the quip database"], @plugin.reply_to.replies
+    assert_equal 1, @plugin.quips.length
+  end
+
+  def test_remove_nonexisting_quip_returns_error_message
+    @plugin.handle(Time.now, "John", "@robut remove quip boooooooo")
+    assert_equal ["I couldn't remove the quip, since it wasn't in the quip database"], @plugin.reply_to.replies
+    assert_equal 2, @plugin.quips.length
+  end
+
+  def test_quip_me_returns_a_random_quip
+    seed = srand(1)
+    begin
+      @plugin.handle(Time.now, "John", "@robut quip")
+      assert_equal ["It works on my machine ..."], @plugin.reply_to.replies
+    ensure
+      srand(seed)
+    end
+  end
+
+  def test_quip_me_returns_an_error_if_no_quips
+    @plugin.store["quips"] = []
+    @plugin.handle(Time.now, "John", "@robut quip")
+    assert_equal ["I don't know any quips"], @plugin.reply_to.replies
+  end
+end

data/test/unit/plugin/say_test.rb

@@ -0,0 +1,34 @@
+require 'test_helper'
+require 'robut/plugin/say'
+
+class Robut::Plugin::SayTest < Test::Unit::TestCase
+
+  def setup
+    @connection = Robut::ConnectionMock.new
+    @presence = Robut::PresenceMock.new(@connection)
+    @plugin = Robut::Plugin::Say.new(@presence)
+    @plugin.instance_eval do
+      # Stub out system
+      def system(c); system_calls << c; end
+      def system_calls; @system_calls ||= []; end;
+    end
+  end
+
+  def test_says_stuff
+    @plugin.handle(Time.now, "@john", "@robut say stuff")
+    assert_equal [], @plugin.reply_to.replies # shouldn't reply to the chat room
+    assert_equal ["say stuff"], @plugin.system_calls
+  end
+  
+  def test_doesnt_say_stuff
+    @plugin.handle(Time.now, "@john", "@robut ok don't say stuff")
+    assert_equal [], @plugin.reply_to.replies
+    assert_equal [], @plugin.system_calls
+  end
+  
+  def test_strips_bad_chars
+    @plugin.handle(Time.now, "@john", "@robut say it's time for $%@ more_test 123 p\"")
+    assert_equal ["say its time for more test 123 p"], @plugin.system_calls
+  end
+  
+end

data/test/unit/plugin/weather_test.rb

@@ -0,0 +1,101 @@
+require 'test_helper'
+require 'webmock/test_unit'
+require 'time-warp'
+require 'robut/plugin/weather'
+
+class Robut::Plugin::WeatherTest < Test::Unit::TestCase
+
+  def setup
+    @connection = Robut::ConnectionMock.new
+    @presence = Robut::PresenceMock.new(@connection)
+    @plugin = Robut::Plugin::Weather.new(@presence)
+  end
+
+  def teardown
+    Robut::Plugin::Weather.default_location = nil
+  end
+
+  def test_handle_no_weather
+    @plugin.handle(Time.now, "John", "lunch?")
+    assert_equal( [], @plugin.reply_to.replies )
+
+    @plugin.handle(Time.now, "John", "?")
+    assert_equal( [], @plugin.reply_to.replies )
+  end
+
+  def test_handle_no_location_no_default
+    @plugin.handle(Time.now, "John", "weather?")
+    assert_equal( ["I don't have a default location!"], @plugin.reply_to.replies )
+  end
+
+  def test_handle_no_location_default_set
+    Robut::Plugin::Weather.default_location = "Seattle"
+    stub_request(:any, "http://www.google.com/ig/api?weather=Seattle").to_return(:body => File.open(File.expand_path("../../../fixtures/seattle.xml", __FILE__), "r").read)
+
+    @plugin.handle(Time.now, "John", "weather?")
+    assert_equal( ["Weather for Seattle, WA: Mostly Cloudy, 58F"], @plugin.reply_to.replies )
+  end
+
+  def test_handle_location
+    stub_request(:any, "http://www.google.com/ig/api?weather=tacoma").to_return(:body => File.open(File.expand_path("../../../fixtures/tacoma.xml", __FILE__), "r").read)
+
+    @plugin.handle(Time.now, "John", "tacoma weather?")
+    assert_equal( ["Weather for Tacoma, WA: Cloudy, 60F"], @plugin.reply_to.replies )
+  end
+
+  def test_no_question_mark
+    @plugin.handle(Time.now, "John", "seattle weather")
+    assert_equal( [], @plugin.reply_to.replies )
+  end
+
+  def test_handle_day
+    stub_request(:any, "http://www.google.com/ig/api?weather=Seattle").to_return(:body => File.open(File.expand_path("../../../fixtures/seattle.xml", __FILE__), "r").read)
+
+    pretend_now_is(2011,"may",23,17) do
+      @plugin.handle(Time.now, "John", "Seattle weather tuesday?")
+      assert_equal( ["Forecast for Seattle, WA on Tue: Partly Cloudy, High: 67F, Low: 51F"], @plugin.reply_to.replies )
+    end
+  end
+
+  def test_handle_tomorrow
+    stub_request(:any, "http://www.google.com/ig/api?weather=Seattle").to_return(:body => File.open(File.expand_path("../../../fixtures/seattle.xml", __FILE__), "r").read)
+
+    pretend_now_is(2011,"may",23,17) do
+      @plugin.handle(Time.now, "John", "Seattle weather tomorrow?")
+      assert_equal( ["Forecast for Seattle, WA on Tue: Partly Cloudy, High: 67F, Low: 51F"], @plugin.reply_to.replies )
+    end
+  end
+
+  def test_handle_today
+    stub_request(:any, "http://www.google.com/ig/api?weather=Seattle").to_return(:body => File.open(File.expand_path("../../../fixtures/seattle.xml", __FILE__), "r").read)
+
+    pretend_now_is(2011,"may",23,17) do
+      @plugin.handle(Time.now, "John", "Seattle weather today?")
+      assert_equal( ["Forecast for Seattle, WA on Mon: Partly Cloudy, High: 59F, Low: 48F"], @plugin.reply_to.replies )
+    end
+  end
+
+  def test_handle_multi_word_location
+    stub_request(:any, "http://www.google.com/ig/api?weather=Las%20Vegas").to_return(:body => File.open(File.expand_path("../../../fixtures/las_vegas.xml", __FILE__), "r").read)
+    @plugin.handle(Time.now, "John", "Las Vegas weather?")
+    assert_equal( ["Weather for Las Vegas, NV: Mostly Cloudy, 83F"], @plugin.reply_to.replies )
+  end
+
+  def test_handle_location_with_comma
+    stub_request(:any, "http://www.google.com/ig/api?weather=Las%20Vegas,%20NV").to_return(:body => File.open(File.expand_path("../../../fixtures/las_vegas.xml", __FILE__), "r").read)
+    @plugin.handle(Time.now, "John", "Las Vegas, NV weather?")
+    assert_equal( ["Weather for Las Vegas, NV: Mostly Cloudy, 83F"], @plugin.reply_to.replies )
+  end
+
+  def test_handle_bad_location
+    stub_request(:any, "http://www.google.com/ig/api?weather=asdf").to_return(:body => File.open(File.expand_path("../../../fixtures/bad_location.xml", __FILE__), "r").read)
+    @plugin.handle(Time.now, "John", "asdf weather?")
+    assert_equal( ["I don't recognize the location: \"asdf\""], @plugin.reply_to.replies )
+  end
+
+  def test_handle_bad_date
+    @plugin.handle(Time.now, "John", "Seattle weather asdf?")
+    assert_equal( ["I don't recognize the date: \"asdf\""], @plugin.reply_to.replies )
+  end
+
+end