spec/models/agents/twitter_stream_agent_spec.rb

@@ -221,20 +221,20 @@ describe Agents::TwitterStreamAgent do

                       context "callback handling" do
              

                         it "logs error messages" do
              

                           stub_without(:on_error).on_error.yields('woups')
              

                -          mock(STDERR).puts(" --> Twitter error: woups <--")
              

                +          mock(STDERR).puts(anything) { |text| expect(text).to match(/woups/) }
              

                           @worker.send(:stream!, ['agent'], @agent)
              

                         end
              

                 
              

                         it "stop when no data was received"do
              

                           stub_without(:on_no_data).on_no_data.yields
              

                           mock(@worker).restart!
              

                -          mock(STDERR).puts(" --> Got no data for awhile; trying to reconnect.")
              

                +          mock(STDERR).puts(anything)
              

                           @worker.send(:stream!, ['agent'], @agent)
              

                         end
              

                 
              

                         it "sleeps for 60 seconds on_max_reconnects" do
              

                           stub_without(:on_max_reconnects).on_max_reconnects.yields
              

                -          mock(STDERR).puts(" --> Oops, tried too many times! <--")
              

                +          mock(STDERR).puts(anything)
              

                           mock(@worker).sleep(60)
              

                           mock(@worker).restart!
              

                           @worker.send(:stream!, ['agent'], @agent)
              
@@ -251,22 +251,21 @@ describe Agents::TwitterStreamAgent do

                 
              

                     context "#handle_status" do
              

                       it "skips retweets" do
              

                -        mock.instance_of(IO).puts('Skipping retweet: retweet')
              

                -        @worker.send(:handle_status, {'text' => 'retweet', 'retweeted_status' => {one: true}})
              

                +        @worker.send(:handle_status, {'text' => 'retweet', 'retweeted_status' => {one: true}, 'id_str' => '123' })
              

                +        expect(@worker.instance_variable_get(:'@recent_tweets')).not_to include('123')
              

                       end
              

                 
              

                       it "deduplicates tweets" do
              

                -        mock.instance_of(IO).puts("dup")
              

                -        @worker.send(:handle_status, {'text' => 'dup', 'id_str' => 1})
              

                -        mock.instance_of(IO).puts("Skipping duplicate tweet: dup")
              

                -        @worker.send(:handle_status, {'text' => 'dup', 'id_str' => 1})
              

                +        @worker.send(:handle_status, {'text' => 'dup', 'id_str' => '1'})
              

                +        @worker.send(:handle_status, {'text' => 'dup', 'id_str' => '1'})
              

                +        expect(@worker.instance_variable_get(:'@recent_tweets').select { |str| str == '1' }.length).to eq 1
              

                       end
              

                 
              

                       it "calls the agent to process the tweet" do
              

                -        stub.instance_of(IO).puts
              

                         mock(@mock_agent).name { 'mock' }
              

                         mock(@mock_agent).process_tweet('agent', {'text' => 'agent'})
              

                -        @worker.send(:handle_status, {'text' => 'agent'})
              

                +        @worker.send(:handle_status, {'text' => 'agent', 'id_str' => '123'})
              

                +        expect(@worker.instance_variable_get(:'@recent_tweets')).to include('123')
              

                       end
              

                     end
              

                   end