chatterbot 0.7.1 → 0.9.0

data/spec/streaming_handler_spec.rb

@@ -0,0 +1,78 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Chatterbot::StreamingHandler" do
+  let(:bot) { test_bot }
+  let(:handler) { StreamingHandler.new(test_bot) }
+
+  def apply_to_handler(&block)
+    handler.apply block
+  end
+
+  describe "endpoint" do
+    it "should default to :user" do
+      expect(handler.endpoint).to eql(:user)
+    end
+
+    it "should pull from opts" do
+      expect(StreamingHandler.new(test_bot, {:endpoint => :bar}).endpoint).to eql(:bar)
+    end
+  end
+  
+  it "should set search query" do
+    apply_to_handler { search 'foo' }
+    expect(handler.search_filter).to eql('foo')
+  end
+  
+  it "should set tweet_handler" do
+    apply_to_handler { replies {'bar'} }
+    expect(handler.tweet_handler.call).to eql('bar')
+  end
+
+  it "should set favorited_handler" do
+    apply_to_handler { favorited {'bar'} }
+    expect(handler.favorite_handler.call).to eql('bar')
+  end
+
+  it "should set dm_handler" do
+    apply_to_handler { direct_message {'bar'} }
+    expect(handler.dm_handler.call).to eql('bar')
+  end
+
+  it "should set follow_handler" do
+    apply_to_handler { followed {'bar'} }
+    expect(handler.follow_handler.call).to eql('bar')
+  end
+
+  it "should set delete_handler" do
+    apply_to_handler { delete {'bar'} }
+    expect(handler.delete_handler.call).to eql('bar')
+  end
+
+  it "should set friends_handler" do
+    apply_to_handler { friends {'bar'} }
+    expect(handler.friends_handler.call).to eql('bar')
+  end
+
+  it "should pull config from bot" do
+    expect(handler.bot).to receive(:config).and_return('config')
+    expect(handler.config).to eql('config')
+  end
+
+  it "limits to one main block type" do
+    apply_to_handler { replies {'bar'} }
+    expect{
+      apply_to_handler { replies {'bar'} }
+    }.to raise_error(RuntimeError)
+  end
+  
+  describe "connection_params" do
+    it "has defaults" do
+      expect(handler.connection_params[:stall_warnings]).to eql("false")
+    end
+
+    it "applies search filter" do
+      apply_to_handler { search 'foo' }
+      expect(handler.connection_params[:track]).to eql("foo")
+    end
+  end
+end

data/spec/streaming_spec.rb

@@ -0,0 +1,172 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+class StreamingHandler
+  attr_accessor :last_object
+end
+
+describe "Chatterbot::Streaming" do
+  let(:bot) { test_bot }
+  let(:user) { fake_user('user', 100) }
+  let(:handler) { StreamingHandler.new(test_bot) }
+  let(:tweet) { fake_tweet(12345) }
+
+  def apply_to_handler(&block)
+    handler.apply block
+  end
+
+  describe "authenticated_user" do
+    it "should get user from client" do
+      expect(bot.client).to receive(:user).and_return('user')
+      expect(bot.authenticated_user).to eql('user')
+    end
+  end
+
+  describe "do_streaming" do
+
+  end
+
+  describe "handle_streaming_object" do
+    before(:each) {
+      allow(bot.client).to receive(:user).and_return(user)
+    }
+
+    describe "Twitter::Tweet" do
+      it "works if no handler" do
+        bot.handle_streaming_object(tweet, handler)
+      end
+
+      context "with handler" do
+        before(:each) do
+          apply_to_handler { replies { |t| @last_object = t } }
+        end
+
+        it "ignores tweets from authenticated user" do
+          expect(tweet).to receive(:user).and_return(user)
+          bot.handle_streaming_object(tweet, handler)
+          expect(handler.last_object).to be_nil
+        end
+        
+        it "passes to handler" do
+          bot.handle_streaming_object(tweet, handler)
+          expect(handler.last_object).to eql(tweet)
+        end
+
+        it "ignores tweets from blacklist" do
+          bot.blacklist = ['chatterbot']
+          bot.handle_streaming_object(tweet, handler)
+          expect(handler.last_object).to be_nil
+        end
+
+        it "ignores tweets if skip_me is true" do
+          bot.exclude = ['tweet']
+          bot.handle_streaming_object(tweet, handler)
+          expect(handler.last_object).to be_nil         
+        end
+      end
+    end
+
+    describe "Twitter::Streaming::DeletedTweet" do
+      it "works if no handler" do
+        obj = Twitter::Streaming::DeletedTweet.new(:id => 1)
+        bot.handle_streaming_object(obj, handler)
+      end
+
+      it "passes to handler" do
+        apply_to_handler { delete { |t| @last_object = t } }
+        obj = Twitter::Streaming::DeletedTweet.new(:id => 1)
+        bot.handle_streaming_object(obj, handler)
+        expect(handler.last_object).to eql(obj)
+      end
+    end
+
+    describe "Twitter::DirectMessage" do
+      it "works if no handler" do
+        obj = Twitter::DirectMessage.new(:id => 1)
+        bot.handle_streaming_object(obj, handler)
+      end
+
+      it "passes to handler" do
+        apply_to_handler { direct_message { |t| @last_object = t } }
+        obj = Twitter::DirectMessage.new(:id => 1)
+        bot.handle_streaming_object(obj, handler)
+        expect(handler.last_object).to eql(obj)
+      end
+    end
+
+    describe "Twitter::Streaming::Event" do
+      it "ignores events generated by authenticated user" do
+        event = Twitter::Streaming::Event.new(
+                                               :event => :follow,
+                                               :source => {:id => user.id, :name => 'name', :screen_name => 'name'},
+                                               :target => {:id => user.id, :name => 'name', :screen_name => 'name'})
+
+        apply_to_handler { followed { |t| @last_object = t } }
+        bot.handle_streaming_object(event, handler)
+        expect(handler.last_object).to be_nil
+      end
+
+      describe "follow" do
+        before(:each) do
+          @event = Twitter::Streaming::Event.new(
+                                                :event => :follow,
+                                                :source => {:id => 12345, :name => 'name', :screen_name => 'name'},
+                                                :target => {:id => user.id, :name => 'name', :screen_name => 'name'})
+
+        end
+
+        it "works if no handler" do
+          bot.handle_streaming_object(@event, handler)
+          expect(handler.last_object).to be_nil
+        end
+
+        it "passes to handler" do
+          apply_to_handler { followed { |t| @last_object = t } }
+          bot.handle_streaming_object(@event, handler)
+          expect(handler.last_object.class).to be(Twitter::User)
+          expect(handler.last_object.id).to be(12345)
+        end
+      end
+
+      describe "favorite" do
+        before(:each) do
+          @event = Twitter::Streaming::Event.new(
+                                                 :event => :favorite,
+                                                 :source => {:id => 12345, :name => 'name', :screen_name => 'name'},
+                                                 :target => {:id => user.id, :name => 'name', :screen_name => 'name'},
+                                                 :target_object => {
+                                                   :from_user => "chatterbot",
+                                                   :id => 56789,
+                                                   :text => "I am a tweet!",
+                                                   :user => { :id => 1, :screen_name => "chatterbot" }
+                                                 })
+        end
+
+        it "works if no handler" do
+          bot.handle_streaming_object(@event, handler)
+          expect(handler.last_object).to be_nil
+         end
+
+        it "passes to handler" do
+          apply_to_handler { favorited { |_, t| @last_object = t } }
+          bot.handle_streaming_object(@event, handler)
+          expect(handler.last_object.class).to be(Twitter::Tweet)
+          expect(handler.last_object.text).to eq("I am a tweet!")
+        end
+      end
+    end
+
+    describe "Twitter::Streaming::FriendList" do
+      it "works if no handler" do
+        obj = Twitter::Streaming::FriendList.new
+        bot.handle_streaming_object(obj, handler)
+      end
+
+      it "passes to handler" do
+        apply_to_handler { friends { |t| @last_object = t } }
+        obj = Twitter::Streaming::FriendList.new
+        bot.handle_streaming_object(obj, handler)
+        expect(handler.last_object).to eql(obj)
+      end
+    end
+  end
+end

data/spec/tweet_spec.rb

@@ -7,32 +7,32 @@ describe "Chatterbot::Tweet" do
     end
 
     it "calls require_login when tweeting" do
-      @bot.should_receive(:require_login).and_return(false)
+      expect(@bot).to receive(:require_login).and_return(false)
       @bot.tweet "tweet test!"
     end
 
     it "calls client.update with the right values" do
       bot = test_bot
 
-      bot.should_receive(:require_login).and_return(true)
-      bot.stub(:client).and_return(double(Twitter::Client))
+      expect(bot).to receive(:require_login).and_return(true)
+      allow(bot).to receive(:client).and_return(double(Twitter::Client))
 
-      bot.stub(:debug_mode?).and_return(false)
+      allow(bot).to receive(:debug_mode?).and_return(false)
 
       test_str = "test!"
-      bot.client.should_receive(:update).with(test_str, {})
+      expect(bot.client).to receive(:update).with(test_str, {})
       bot.tweet test_str
     end
 
     it "doesn't tweet when debug_mode? is set" do
       bot = test_bot
       
-      bot.should_receive(:require_login).and_return(true)
-      bot.stub(:client).and_return(double(Twitter::Client))
+      expect(bot).to receive(:require_login).and_return(true)
+      allow(bot).to receive(:client).and_return(double(Twitter::Client))
 
-      bot.stub(:debug_mode?).and_return(true)
+      allow(bot).to receive(:debug_mode?).and_return(true)
 
-      bot.client.should_not_receive(:update)
+      expect(bot.client).not_to receive(:update)
       bot.tweet "no tweet!"
     end
   end
@@ -40,35 +40,35 @@ describe "Chatterbot::Tweet" do
   describe "#reply" do
     it "calls require_login when replying" do
       bot = test_bot
-      bot.should_receive(:require_login).and_return(false)
+      expect(bot).to receive(:require_login).and_return(false)
       bot.reply "reply test!", {"id" => 100}
     end
 
     it "calls client.update with the right values" do
       bot = test_bot
-      bot.should_receive(:require_login).and_return(true)
-      bot.stub(:client).and_return(double(Twitter::Client))
+      expect(bot).to receive(:require_login).and_return(true)
+      allow(bot).to receive(:client).and_return(double(Twitter::Client))
 
-      bot.stub(:debug_mode?).and_return(false)
+      allow(bot).to receive(:debug_mode?).and_return(false)
 
       test_str = "test!"
 
       s = {
         :id => 100
       }
-      bot.client.should_receive(:update).with(test_str, {:in_reply_to_status_id => 100})
+      expect(bot.client).to receive(:update).with(test_str, {:in_reply_to_status_id => 100})
       bot.reply test_str, s
     end
 
 
     it "doesn't reply when debug_mode? is set" do
       bot = test_bot
-      bot.should_receive(:require_login).and_return(true)
-      bot.stub(:client).and_return(double(Twitter::Client))
+      expect(bot).to receive(:require_login).and_return(true)
+      allow(bot).to receive(:client).and_return(double(Twitter::Client))
 
-      bot.stub(:debug_mode?).and_return(true)
+      allow(bot).to receive(:debug_mode?).and_return(true)
 
-      bot.client.should_not_receive(:update)
+      expect(bot.client).not_to receive(:update)
       bot.reply "no reply test!", {:id => 100}
     end
   end

data/spec/utils_spec.rb

@@ -0,0 +1,29 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Chatterbot::Utils" do
+  before(:each) do
+    @bot = test_bot
+  end
+  
+  describe "id_from_tweet" do
+    it "works with numbers" do
+      expect(@bot.id_from_tweet(1234)).to eq(1234)
+    end
+    
+    it "works with tweets" do
+      t = Twitter::Tweet.new(:id => 7890, :text => "did you know that i hate bots?")
+      expect(@bot.id_from_tweet(t)).to eq(7890)
+    end
+  end
+  
+  describe "id_from_user" do
+    it "works with numbers" do
+      expect(@bot.id_from_user(1234)).to eq(1234)
+    end
+    
+    it "works with users" do
+      t = Twitter::User.new(:id => 2345, :name => "name")
+      expect(@bot.id_from_user(t)).to eq(2345)
+    end
+  end
+end

data/specs.watchr

@@ -20,7 +20,7 @@ end
 
 def run(files_to_run)
   puts("Running: #{files_to_run}")
-  system("clear;rspec -cfs #{files_to_run}")
+  system("clear;rspec -cfd #{files_to_run}")
   no_int_for_you
 end
 
@@ -45,16 +45,4 @@ def no_int_for_you
   @sent_an_int = nil
 end
 
-Signal.trap 'INT' do
-  if @sent_an_int then
-    puts " A second INT? Ok, I get the message. Shutting down now."
-    exit
-  else
-    puts " Did you just send me an INT? Ugh. I'll quit for real if you do it again."
-    @sent_an_int = true
-    Kernel.sleep 1.5
-    run_all_specs
-  end
-end
-
 run_all_specs

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: chatterbot
 version: !ruby/object:Gem::Version
-  version: 0.7.1
+  version: 0.9.0
 platform: ruby
 authors:
 - Colin Mitchell
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-12-18 00:00:00.000000000 Z
+date: 2014-07-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redcarpet
@@ -30,28 +30,28 @@ dependencies:
     requirements:
     - - '>='
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.4.7
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '>='
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.4.7
 - !ruby/object:Gem::Dependency
   name: twitter
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 4.8.1
+        version: 5.8.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 4.8.1
+        version: 5.8.0
 - !ruby/object:Gem::Dependency
   name: launchy
   requirement: !ruby/object:Gem::Requirement
@@ -112,30 +112,30 @@ dependencies:
   name: rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ~>
       - !ruby/object:Gem::Version
-        version: 2.14.1
+        version: 3.0.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ~>
       - !ruby/object:Gem::Version
-        version: 2.14.1
+        version: 3.0.0
 - !ruby/object:Gem::Dependency
   name: rspec-mocks
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ~>
       - !ruby/object:Gem::Version
-        version: 2.14.1
+        version: 3.0.2
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ~>
       - !ruby/object:Gem::Version
-        version: 2.14.1
+        version: 3.0.2
 - !ruby/object:Gem::Dependency
   name: rdoc
   requirement: !ruby/object:Gem::Requirement
@@ -165,7 +165,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: watchr
+  name: observr
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - '>='
@@ -205,6 +205,7 @@ files:
 - examples/echoes_bot.rb
 - examples/loop_bot.rb
 - examples/search_bot.rb
+- examples/streaming_bot.rb
 - examples/tweet_logger.rb
 - lib/chatterbot.rb
 - lib/chatterbot/blacklist.rb
@@ -213,15 +214,20 @@ files:
 - lib/chatterbot/config.rb
 - lib/chatterbot/db.rb
 - lib/chatterbot/dsl.rb
+- lib/chatterbot/favorite.rb
 - lib/chatterbot/followers.rb
 - lib/chatterbot/helpers.rb
 - lib/chatterbot/logging.rb
+- lib/chatterbot/profile.rb
 - lib/chatterbot/reply.rb
 - lib/chatterbot/retweet.rb
 - lib/chatterbot/search.rb
 - lib/chatterbot/skeleton.rb
+- lib/chatterbot/streaming.rb
+- lib/chatterbot/streaming_handler.rb
 - lib/chatterbot/tweet.rb
 - lib/chatterbot/ui.rb
+- lib/chatterbot/utils.rb
 - lib/chatterbot/version.rb
 - spec/blacklist_spec.rb
 - spec/bot_spec.rb
@@ -229,15 +235,20 @@ files:
 - spec/config_spec.rb
 - spec/db_spec.rb
 - spec/dsl_spec.rb
+- spec/favorite_spec.rb
 - spec/followers_spec.rb
 - spec/helpers_spec.rb
 - spec/logging_spec.rb
+- spec/profile_spec.rb
 - spec/reply_spec.rb
 - spec/retweet_spec.rb
 - spec/search_spec.rb
 - spec/skeleton_spec.rb
 - spec/spec_helper.rb
+- spec/streaming_handler_spec.rb
+- spec/streaming_spec.rb
 - spec/tweet_spec.rb
+- spec/utils_spec.rb
 - specs.watchr
 - templates/skeleton.txt
 homepage: http://github.com/muffinista/chatterbot
@@ -260,7 +271,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: chatterbot
-rubygems_version: 2.1.11
+rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
 summary: A ruby framework for writing Twitter bots