t 0.2.1 → 0.3.0

data/lib/t/cli/list/remove.rb

@@ -1,5 +1,7 @@
+require 'retryable'
 require 't/core_ext/enumerable'
 require 't/core_ext/string'
+require 't/collectable'
 require 't/rcfile'
 require 'thor'
 require 'twitter'
@@ -8,6 +10,8 @@ module T
   class CLI
     class List
       class Remove < Thor
+        include T::Collectable
+
         DEFAULT_HOST = 'api.twitter.com'
         DEFAULT_PROTOCOL = 'https'
 
@@ -18,12 +22,105 @@ module T
           @rcfile = RCFile.instance
         end
 
+        desc "friends LIST_NAME", "Remove all friends from a list."
+        def friends(list_name)
+          list_member_ids = collect_with_cursor do |cursor|
+            client.list_members(list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+          end
+          friend_ids = collect_with_cursor do |cursor|
+            client.friend_ids(:cursor => cursor)
+          end
+          list_member_ids_to_remove = (friend_ids - list_member_ids)
+          number = list_member_ids_to_remove.length
+          if number.zero?
+            return say "None of @#{@rcfile.default_profile[0]}'s friends are members of the list \"#{list_name}\"."
+          else
+            return unless yes? "Are you sure you want to remove #{number} #{number == 1 ? 'friend' : 'friends'} from the list \"#{list_name}\"?"
+          end
+          list_member_ids_to_remove.threaded_map do |list_member_id|
+            retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+              client.list_remove_member(list_name, list_member_id)
+            end
+          end
+          say "@#{@rcfile.default_profile[0]} removed #{number} #{number == 1 ? 'friend' : 'friends'} from the list \"#{list_name}\"."
+          say
+          say "Run `#{$0} list add all friends #{list_name}` to undo."
+        end
+
+        desc "followers LIST_NAME", "Remove all followers from a list."
+        def followers(list_name)
+          list_member_ids = collect_with_cursor do |cursor|
+            client.list_members(list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+          end
+          follower_ids = collect_with_cursor do |cursor|
+            client.follower_ids(:cursor => cursor)
+          end
+          list_member_ids_to_remove = (follower_ids - list_member_ids)
+          number = list_member_ids_to_remove.length
+          if number.zero?
+            return say "None of @#{@rcfile.default_profile[0]}'s followers are members of the list \"#{list_name}\"."
+          else
+            return unless yes? "Are you sure you want to remove #{number} #{number == 1 ? 'follower' : 'followers'} from the list \"#{list_name}\"?"
+          end
+          list_member_ids_to_remove.threaded_map do |list_member_id|
+            retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+              client.list_remove_member(list_name, list_member_id)
+            end
+          end
+          say "@#{@rcfile.default_profile[0]} removed #{number} #{number == 1 ? 'follower' : 'followers'} from the list \"#{list_name}\"."
+          say
+          say "Run `#{$0} list add all followers #{list_name}` to undo."
+        end
+
+        desc "listed FROM_LIST_NAME TO_LIST_NAME", "Remove all list members from a list."
+        def listed(from_list_name, to_list_name)
+          to_list_members = collect_with_cursor do |cursor|
+            client.list_members(to_list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+          end
+          from_list_members = collect_with_cursor do |cursor|
+            client.list_members(from_list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+          end
+          list_member_ids_to_remove = (from_list_members.collect(&:id) - to_list_members.collect(&:id))
+          number = list_member_ids_to_remove.length
+          if number.zero?
+            return say "None of the members of the list \"#{from_list_name}\" are members of the list \"#{to_list_name}\"."
+          else
+            return unless yes? "Are you sure you want to remove #{number} #{number == 1 ? 'member' : 'members'} from the list \"#{to_list_name}\"?"
+          end
+          list_member_ids_to_remove.threaded_map do |list_member_id|
+            retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+              client.list_remove_member(to_list_name, list_member_id)
+            end
+          end
+          say "@#{@rcfile.default_profile[0]} removed #{number} #{number == 1 ? 'member' : 'members'} from the list \"#{to_list_name}\"."
+          say
+          say "Run `#{$0} list add all listed #{from_list_name} #{to_list_name}` to undo."
+        end
+
+        desc "members LIST_NAME", "Remove all members from a list."
+        def members(list_name)
+          list_members = collect_with_cursor do |cursor|
+            client.list_members(list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+          end
+          number = list_members.length
+          return say "The list \"#{list_name}\" doesn't have any members." if number.zero?
+          return unless yes? "Are you sure you want to remove #{number} #{number == 1 ? 'member' : 'members'} from the list \"#{list_name}\"?"
+          list_members.collect(&:id).threaded_map do |list_member_id|
+            retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+              client.list_remove_member(list_name, list_member_id)
+            end
+          end
+          say "@#{@rcfile.default_profile[0]} removed #{number} #{number == 1 ? 'member' : 'members'} from the list \"#{list_name}\"."
+        end
+        map %w(all) => :members
+
         desc "users LIST_NAME SCREEN_NAME [SCREEN_NAME...]", "Remove users from a list."
         def users(list_name, screen_name, *screen_names)
           screen_names.unshift(screen_name)
           screen_names.threaded_map do |screen_name|
-            screen_name = screen_name.strip_at
-            client.list_remove_member(list_name, screen_name)
+            retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+              client.list_remove_member(list_name, screen_name)
+            end
           end
           number = screen_names.length
           say "@#{@rcfile.default_profile[0]} removed #{number} #{number == 1 ? 'user' : 'users'} from the list \"#{list_name}\"."
@@ -31,10 +128,6 @@ module T
           say "Run `#{$0} list add users #{list_name} #{screen_names.join(' ')}` to undo."
         end
 
-        desc "all SUBCOMMAND ...ARGS", "Remove all users to a list."
-        require 't/cli/list/remove/all'
-        subcommand 'all', CLI::List::Remove::All
-
       private
 
         def base_url

data/lib/t/cli/search.rb

@@ -0,0 +1,128 @@
+require 'action_view'
+require 'retryable'
+require 't/core_ext/enumerable'
+require 't/rcfile'
+require 'thor'
+require 'twitter'
+
+module T
+  class CLI
+    class Search < Thor
+      include ActionView::Helpers::DateHelper
+
+      DEFAULT_HOST = 'api.twitter.com'
+      DEFAULT_PROTOCOL = 'https'
+      DEFAULT_NUM_RESULTS = 20
+      MAX_PAGES = 16
+      MAX_NUM_RESULTS = 200
+      MAX_SCREEN_NAME_SIZE = 20
+
+      check_unknown_options!
+
+      def initialize(*)
+        super
+        @rcfile = RCFile.instance
+      end
+
+      desc "all QUERY", "Returns the #{DEFAULT_NUM_RESULTS} most recent Tweets that match a specified query."
+      method_option :number, :aliases => "-n", :type => :numeric, :default => DEFAULT_NUM_RESULTS
+      method_option :reverse, :aliases => "-r", :type => :boolean, :default => false
+      def all(query)
+        defaults = {:include_entities => false}
+        defaults.merge!(:rpp => options['number']) if options['number']
+        timeline = client.search(query, defaults)
+        timeline.reverse! if options['reverse']
+        run_pager
+        timeline.each do |status|
+          say "#{status.from_user.rjust(MAX_SCREEN_NAME_SIZE)}: #{status.text} (#{time_ago_in_words(status.created_at)} ago)"
+        end
+      end
+
+      desc "timeline QUERY", "Returns Tweets in your timeline that match a specified query."
+      def timeline(query)
+        timeline = 1.upto(MAX_PAGES).threaded_map do |page|
+          retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+            client.home_timeline(:page => page, :count => MAX_NUM_RESULTS).map do |status|
+              status if /#{query}/i.match(status.text)
+            end
+          end
+        end
+        run_pager
+        timeline.flatten.compact.each do |status|
+          say "#{status.user.screen_name.rjust(MAX_SCREEN_NAME_SIZE)}: #{status.text} (#{time_ago_in_words(status.created_at)} ago)"
+        end
+      end
+      map %w(tl) => :timeline
+
+      desc "user SCREEN_NAME QUERY", "Returns Tweets in a user's timeline that match a specified query."
+      def user(screen_name, query)
+        screen_name = screen_name.strip_at
+        timeline = 1.upto(MAX_PAGES).threaded_map do |page|
+          retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+            client.user_timeline(screen_name, :page => page, :count => MAX_NUM_RESULTS).map do |status|
+              status if /#{query}/i.match(status.text)
+            end
+          end
+        end
+        run_pager
+        timeline.flatten.compact.each do |status|
+          say "#{status.user.screen_name.rjust(MAX_SCREEN_NAME_SIZE)}: #{status.text} (#{time_ago_in_words(status.created_at)} ago)"
+        end
+      end
+
+    private
+
+      def base_url
+        "#{protocol}://#{host}"
+      end
+
+      def client
+        return @client if @client
+        @rcfile.path = parent_options['profile'] if parent_options['profile']
+        @client = Twitter::Client.new(
+          :endpoint => base_url,
+          :consumer_key => @rcfile.default_consumer_key,
+          :consumer_secret => @rcfile.default_consumer_secret,
+          :oauth_token => @rcfile.default_token,
+          :oauth_token_secret  => @rcfile.default_secret
+        )
+      end
+
+      def host
+        parent_options['host'] || DEFAULT_HOST
+      end
+
+      def protocol
+        parent_options['no_ssl'] ? 'http' : DEFAULT_PROTOCOL
+      end
+
+      def run_pager
+        return if RUBY_PLATFORM =~ /win32/
+        return if ENV["T_ENV"] == "test"
+        return unless STDOUT.tty?
+
+        read, write = IO.pipe
+
+        unless Kernel.fork # Child process
+          STDOUT.reopen(write)
+          STDERR.reopen(write) if STDERR.tty?
+          read.close
+          write.close
+          return
+        end
+
+        # Parent process, become pager
+        STDIN.reopen(read)
+        read.close
+        write.close
+
+        ENV['LESS'] = 'FSRX' # Don't page if the input is short enough
+
+        Kernel.select [STDIN] # Wait until we have input before we start the pager
+        pager = ENV['PAGER'] || 'less'
+        exec pager rescue exec "/bin/sh", "-c", pager
+      end
+
+    end
+  end
+end

data/lib/t/cli/set.rb

@@ -26,7 +26,7 @@ module T
       def default(screen_name, consumer_key=nil)
         screen_name = screen_name.strip_at
         @rcfile.path = parent_options['profile'] if parent_options['profile']
-        consumer_key = rcfile[screen_name].keys.last if consumer_key.nil?
+        consumer_key = @rcfile[screen_name].keys.last if consumer_key.nil?
         @rcfile.default_profile = {'username' => screen_name, 'consumer_key' => consumer_key}
         say "Default account has been updated."
       end

data/lib/t/cli/unfollow.rb

@@ -1,5 +1,7 @@
+require 'retryable'
 require 't/core_ext/enumerable'
 require 't/core_ext/string'
+require 't/collectable'
 require 't/rcfile'
 require 'thor'
 require 'twitter'
@@ -7,6 +9,8 @@ require 'twitter'
 module T
   class CLI
     class Unfollow < Thor
+      include T::Collectable
+
       DEFAULT_HOST = 'api.twitter.com'
       DEFAULT_PROTOCOL = 'https'
 
@@ -17,12 +21,96 @@ module T
         @rcfile = RCFile.instance
       end
 
+      desc "listed LIST_NAME", "Unfollow all members of a list."
+      def listed(list_name)
+        list_member_collection = collect_with_cursor do |cursor|
+          client.list_members(list_name, :cursor => cursor, :include_entities => false, :skip_status => true)
+        end
+        number = list_member_collection.length
+        return say "@#{@rcfile.default_profile[0]} is already not following any list members." if number.zero?
+        return unless yes? "Are you sure you want to unfollow #{number} #{number == 1 ? 'user' : 'users'}?"
+        list_member_collection.threaded_map do |list_member|
+          retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+            client.unfollow(list_member.id, :include_entities => false)
+          end
+        end
+        say "@#{@rcfile.default_profile[0]} is no longer following #{number} #{number == 1 ? 'user' : 'users'}."
+        say
+        say "Run `#{$0} follow all listed #{list_name}` to follow again."
+      end
+
+      desc "followers", "Unfollow all followers."
+      def followers
+        follower_ids = collect_with_cursor do |cursor|
+          client.follower_ids(:cursor => cursor)
+        end
+        friend_ids = collect_with_cursor do |cursor|
+          friends = client.friend_ids(:cursor => cursor)
+        end
+        follow_ids = (follower_ids - friend_ids)
+        number = follow_ids.length
+        return say "@#{@rcfile.default_profile[0]} is already not following any followers." if number.zero?
+        return unless yes? "Are you sure you want to unfollow #{number} #{number == 1 ? 'user' : 'users'}?"
+        screen_names = follow_ids.threaded_map do |follow_id|
+          retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+            client.unfollow(follow_id, :include_entities => false)
+          end
+        end
+        say "@#{@rcfile.default_profile[0]} is no longer following #{number} #{number == 1 ? 'user' : 'users'}."
+        say
+        say "Run `#{$0} follow all followers` to stop."
+      end
+
+      desc "friends", "Unfollow all friends."
+      def friends
+        friend_ids = collect_with_cursor do |cursor|
+          client.friend_ids(:cursor => cursor)
+        end
+        number = friend_ids.length
+        return say "@#{@rcfile.default_profile[0]} is already not following anyone." if number.zero?
+        return unless yes? "Are you sure you want to unfollow #{number} #{number == 1 ? 'user' : 'users'}?"
+        screen_names = friend_ids.threaded_map do |friend_id|
+          retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+            user = client.unfollow(friend_id, :include_entities => false)
+            user.screen_name
+          end
+        end
+        say "@#{@rcfile.default_profile[0]} is no longer following #{number} #{number == 1 ? 'user' : 'users'}."
+        say
+        say "Run `#{$0} follow users #{screen_names.join(' ')}` to follow again."
+      end
+      map %w(everyone everybody) => :friends
+
+      desc "nonfollowers", "Unfollow all non-followers."
+      def nonfollowers
+        friend_ids = collect_with_cursor do |cursor|
+          client.friend_ids(:cursor => cursor)
+        end
+        follower_ids = collect_with_cursor do |cursor|
+          client.follower_ids(:cursor => cursor)
+        end
+        unfollow_ids = (friend_ids - follower_ids)
+        number = unfollow_ids.length
+        return say "@#{@rcfile.default_profile[0]} is already not following any non-followers." if number.zero?
+        return unless yes? "Are you sure you want to unfollow #{number} #{number == 1 ? 'user' : 'users'}?"
+        screen_names = unfollow_ids.threaded_map do |unfollow_id|
+          retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+            user = client.unfollow(unfollow_id, :include_entities => false)
+            user.screen_name
+          end
+        end
+        say "@#{@rcfile.default_profile[0]} is no longer following #{number} #{number == 1 ? 'user' : 'users'}."
+        say
+        say "Run `#{$0} follow users #{screen_names.join(' ')}` to follow again."
+      end
+
       desc "users SCREEN_NAME [SCREEN_NAME...]", "Allows you to stop following users."
       def users(screen_name, *screen_names)
         screen_names.unshift(screen_name)
         screen_names.threaded_map do |screen_name|
-          screen_name = screen_name.strip_at
-          client.unfollow(screen_name, :include_entities => false)
+          retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+            client.unfollow(screen_name, :include_entities => false)
+          end
         end
         number = screen_names.length
         say "@#{@rcfile.default_profile[0]} is no longer following #{number} #{number == 1 ? 'user' : 'users'}."
@@ -30,10 +118,6 @@ module T
         say "Run `#{$0} follow users #{screen_names.join(' ')}` to follow again."
       end
 
-      desc "all SUBCOMMAND ...ARGS", "Follow all users."
-      require 't/cli/unfollow/all'
-      subcommand 'all', CLI::Unfollow::All
-
     private
 
       def base_url

data/lib/t/collectable.rb

@@ -0,0 +1,12 @@
+module T
+  module Collectable
+
+    def collect_with_cursor(collection=[], cursor=-1, &block)
+      return collection if cursor == 0
+      object = yield cursor
+      collection += object.collection
+      collect_with_cursor(collection, object.next_cursor)
+    end
+
+  end
+end

data/lib/t/version.rb

@@ -8,12 +8,12 @@ module T
 
     # @return [Integer]
     def self.minor
-      2
+      3
     end
 
     # @return [Integer]
     def self.patch
-      1
+      0
     end
 
     # @return [String, NilClass]

data/spec/cli/follow_spec.rb

@@ -16,6 +16,181 @@ describe T::CLI::Follow do
     $stdout = @old_stdout
   end
 
+  describe "#followers" do
+    before do
+      @t.options = @t.options.merge(:profile => fixture_path + "/.trc")
+    end
+    context "no followers" do
+      before do
+        stub_get("/1/followers/ids.json").
+          with(:query => {:cursor => "-1"}).
+          to_return(:body => fixture("friends_ids.json"), :headers => {:content_type => "application/json; charset=utf-8"})
+        stub_get("/1/friends/ids.json").
+          with(:query => {:cursor => "-1"}).
+          to_return(:body => fixture("friends_ids.json"), :headers => {:content_type => "application/json; charset=utf-8"})
+      end
+      it "should request the correct resource" do
+        @t.follow("followers")
+        a_get("/1/followers/ids.json").
+          with(:query => {:cursor => "-1"}).
+          should have_been_made
+        a_get("/1/friends/ids.json").
+          with(:query => {:cursor => "-1"}).
+          should have_been_made
+      end
+      it "should have the correct output" do
+        @t.follow("followers")
+        $stdout.string.chomp.should == "@testcli is already following all followers."
+      end
+    end
+    context "one follower" do
+      before do
+        stub_get("/1/followers/ids.json").
+          with(:query => {:cursor => "-1"}).
+          to_return(:body => fixture("friends_ids.json"), :headers => {:content_type => "application/json; charset=utf-8"})
+        stub_get("/1/friends/ids.json").
+          with(:query => {:cursor => "-1"}).
+          to_return(:body => fixture("followers_ids.json"), :headers => {:content_type => "application/json; charset=utf-8"})
+      end
+      it "should request the correct resource" do
+        stub_post("/1/friendships/create.json").
+          with(:body => {:user_id => "7505382", :include_entities => "false"}).
+          to_return(:body => fixture("sferik.json"), :headers => {:content_type => "application/json; charset=utf-8"})
+        $stdout.should_receive(:print).with("Are you sure you want to follow 1 user? ")
+        $stdin.should_receive(:gets).and_return("yes")
+        @t.follow("followers")
+        a_get("/1/followers/ids.json").
+          with(:query => {:cursor => "-1"}).
+          should have_been_made
+        a_get("/1/friends/ids.json").
+          with(:query => {:cursor => "-1"}).
+          should have_been_made
+        a_post("/1/friendships/create.json").
+          with(:body => {:user_id => "7505382", :include_entities => "false"}).
+          should have_been_made
+      end
+      context "yes" do
+        it "should have the correct output" do
+          stub_post("/1/friendships/create.json").
+            with(:body => {:user_id => "7505382", :include_entities => "false"}).
+            to_return(:body => fixture("sferik.json"), :headers => {:content_type => "application/json; charset=utf-8"})
+          $stdout.should_receive(:print).with("Are you sure you want to follow 1 user? ")
+          $stdin.should_receive(:gets).and_return("yes")
+          @t.follow("followers")
+          $stdout.string.should =~ /^@testcli is now following 1 more user\.$/
+        end
+      end
+      context "no" do
+        it "should have the correct output" do
+          $stdout.should_receive(:print).with("Are you sure you want to follow 1 user? ")
+          $stdin.should_receive(:gets).and_return("no")
+          @t.follow("followers")
+          $stdout.string.chomp.should == ""
+        end
+      end
+      context "Twitter is down" do
+        it "should retry 3 times and then raise an error" do
+          stub_post("/1/friendships/create.json").
+            with(:body => {:user_id => "7505382", :include_entities => "false"}).
+            to_return(:status => 502)
+          $stdout.should_receive(:print).with("Are you sure you want to follow 1 user? ")
+          $stdin.should_receive(:gets).and_return("yes")
+          lambda do
+            @t.follow("followers")
+          end.should raise_error("Twitter is down or being upgraded.")
+          a_post("/1/friendships/create.json").
+            with(:body => {:user_id => "7505382", :include_entities => "false"}).
+            should have_been_made.times(3)
+        end
+      end
+    end
+  end
+
+  describe "#listed" do
+    before do
+      @t.options = @t.options.merge(:profile => fixture_path + "/.trc")
+      stub_get("/1/account/verify_credentials.json").
+        to_return(:body => fixture("sferik.json"), :headers => {:content_type => "application/json; charset=utf-8"})
+    end
+    context "no users" do
+      before do
+        stub_get("/1/lists/members.json").
+          with(:query => {:cursor => "-1", :include_entities => "false", :owner_screen_name => "sferik", :skip_status => "true", :slug => "presidents"}).
+          to_return(:body => fixture("empty_cursor.json"), :headers => {:content_type => "application/json; charset=utf-8"})
+      end
+      it "should request the correct resource" do
+        @t.follow("listed", "presidents")
+        a_get("/1/account/verify_credentials.json").
+          should have_been_made
+        a_get("/1/lists/members.json").
+          with(:query => {:cursor => "-1", :include_entities => "false", :owner_screen_name => "sferik", :skip_status => "true", :slug => "presidents"}).
+          should have_been_made
+      end
+      it "should have the correct output" do
+        @t.follow("listed", "presidents")
+        $stdout.string.chomp.should == "@testcli is already following all list members."
+      end
+    end
+    context "one user" do
+      before do
+        @t.options = @t.options.merge(:profile => fixture_path + "/.trc")
+        stub_get("/1/lists/members.json").
+          with(:query => {:cursor => "-1", :include_entities => "false", :owner_screen_name => "sferik", :skip_status => "true", :slug => "presidents"}).
+          to_return(:body => fixture("users_list.json"), :headers => {:content_type => "application/json; charset=utf-8"})
+      end
+      it "should request the correct resource" do
+        stub_post("/1/friendships/create.json").
+          with(:body => {:user_id => "7505382", :include_entities => "false"}).
+          to_return(:body => fixture("sferik.json"), :headers => {:content_type => "application/json; charset=utf-8"})
+        $stdout.should_receive(:print).with("Are you sure you want to follow 1 user? ")
+        $stdin.should_receive(:gets).and_return("yes")
+        @t.follow("listed", "presidents")
+        a_get("/1/account/verify_credentials.json").
+          should have_been_made
+        a_get("/1/lists/members.json").
+          with(:query => {:cursor => "-1", :include_entities => "false", :owner_screen_name => "sferik", :skip_status => "true", :slug => "presidents"}).
+          should have_been_made
+        a_post("/1/friendships/create.json").
+          with(:body => {:user_id => "7505382", :include_entities => "false"}).
+          should have_been_made
+      end
+      context "yes" do
+        it "should have the correct output" do
+          stub_post("/1/friendships/create.json").
+            with(:body => {:user_id => "7505382", :include_entities => "false"}).
+            to_return(:body => fixture("sferik.json"), :headers => {:content_type => "application/json; charset=utf-8"})
+          $stdout.should_receive(:print).with("Are you sure you want to follow 1 user? ")
+          $stdin.should_receive(:gets).and_return("yes")
+          @t.follow("listed", "presidents")
+          $stdout.string.should =~ /^@testcli is now following 1 more user\.$/
+        end
+      end
+      context "no" do
+        it "should have the correct output" do
+          $stdout.should_receive(:print).with("Are you sure you want to follow 1 user? ")
+          $stdin.should_receive(:gets).and_return("no")
+          @t.follow("listed", "presidents")
+          $stdout.string.chomp.should == ""
+        end
+      end
+      context "Twitter is down" do
+        it "should retry 3 times and then raise an error" do
+          stub_post("/1/friendships/create.json").
+            with(:body => {:user_id => "7505382", :include_entities => "false"}).
+            to_return(:status => 502)
+          $stdout.should_receive(:print).with("Are you sure you want to follow 1 user? ")
+          $stdin.should_receive(:gets).and_return("yes")
+          lambda do
+            @t.follow("listed", "presidents")
+          end.should raise_error("Twitter is down or being upgraded.")
+          a_post("/1/friendships/create.json").
+            with(:body => {:user_id => "7505382", :include_entities => "false"}).
+            should have_been_made.times(3)
+        end
+      end
+    end
+  end
+
   describe "#users" do
     before do
       @t.options = @t.options.merge(:profile => fixture_path + "/.trc")
@@ -28,21 +203,35 @@ describe T::CLI::Follow do
       end
     end
     context "one user" do
-      before do
+      it "should request the correct resource" do
         stub_post("/1/friendships/create.json").
           with(:body => {:screen_name => "sferik", :include_entities => "false"}).
           to_return(:body => fixture("sferik.json"), :headers => {:content_type => "application/json; charset=utf-8"})
-      end
-      it "should request the correct resource" do
         @t.follow("users", "sferik")
         a_post("/1/friendships/create.json").
           with(:body => {:screen_name => "sferik", :include_entities => "false"}).
           should have_been_made
       end
       it "should have the correct output" do
+        stub_post("/1/friendships/create.json").
+          with(:body => {:screen_name => "sferik", :include_entities => "false"}).
+          to_return(:body => fixture("sferik.json"), :headers => {:content_type => "application/json; charset=utf-8"})
         @t.follow("users", "sferik")
         $stdout.string.should =~ /^@testcli is now following 1 more user\.$/
       end
+      context "Twitter is down" do
+        it "should retry 3 times and then raise an error" do
+          stub_post("/1/friendships/create.json").
+            with(:body => {:screen_name => "sferik", :include_entities => "false"}).
+            to_return(:status => 502)
+          lambda do
+            @t.follow("users", "sferik")
+          end.should raise_error("Twitter is down or being upgraded.")
+          a_post("/1/friendships/create.json").
+            with(:body => {:screen_name => "sferik", :include_entities => "false"}).
+            should have_been_made.times(3)
+        end
+      end
     end
   end