download_tv 2.2.2 → 2.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: bd060f362bcaa6f8150f6e766ff4f8d882666b8d
-  data.tar.gz: ddc3287c2df8cfb98d5bc5b39b8e559099721bee
+  metadata.gz: f8dda689a52414ffbba9c500ddd35c559726b77e
+  data.tar.gz: 34a781bc82bde130397290dc0038bbb3e842198d
 SHA512:
-  metadata.gz: 68b70848a669d493ec77c5e12d5d82acb4627161cc5ca975d8c0650d5ea17a82bce1c56657ad278bfff903680e0c0b11f668d7ce3c2031e51abfc1452fc44859
-  data.tar.gz: 8cd0a1e1732d280c842c867c0f6ae5227fb8fd7d495481b5e7a147cf3c06956a96100ce31c25be734143cd498c4ebdc89f645d028eedf43095e4375dfcdf950a
+  metadata.gz: ab18d3786f2b7ca4b7a5757355f50a7e1cfcfb871e375dc165841e8212a2518b48d64970a0a069699fecbf6b6b422880bf15704d53b8f87b68b168790d72d8b6
+  data.tar.gz: c5cbe9275f0174e56da29fa4ba3f90054e663b05f1dc2633b36dc05254af3c3cf74dec11f3db5f80996975f27add8e1d1f231525d0556ff796dc37f214b875ec

data/README.md

@@ -37,7 +37,7 @@ Specific options:
 
 By default, it fetches the list of episodes from MyEpisodes.com that have aired since the program was run for the last time and tries to download them. The -o flag can be used in order to re-download the episodes from previous days. The --dry-run option is useful to prevent download_tv from updating the date (for example, when running the application shortly after an episode airs)
 
-In order to download a single episode, use the -d flag: *tv -d Breaking Bad S04E01*
+In order to download a single episode, use the -d flag, quoting the string when it contains spaces: *tv -d "Breaking Bad S04E01"*
 
 The -f flag can be used to read the list of episodes to download from a file. Each line of the file is interpreted as a episode to download: *tv -f /path/to/listofeps*
 

data/bin/tv

@@ -74,13 +74,13 @@ opt_parser.parse!(ARGV)
 begin
   case options[:cmd]
   when 'run'
-    dl = DownloadTV::Downloader.new(options[:offset], config)
-    dl.run(options[:dry])
+    dl = DownloadTV::Downloader.new(config)
+    dl.run(options[:dry], options[:offset].abs)
   when 'dl'
-    dl = DownloadTV::Downloader.new(options[:offset], config)
+    dl = DownloadTV::Downloader.new(config)
     dl.download_single_show(options[:arg])
   when 'file'
-    dl = DownloadTV::Downloader.new(options[:offset], config)
+    dl = DownloadTV::Downloader.new(config)
     dl.download_from_file(options[:arg])
   when 'config'
     DownloadTV::Configuration.new(config, true)
@@ -90,4 +90,7 @@ begin
 rescue Interrupt
   puts 'Interrupt signal detected. Exiting...'
   exit 1
+rescue SocketError, Errno::ECONNRESET, Net::OpenTimeout
+  warn 'Connection error.'
+  exit 1
 end

data/lib/download_tv/configuration.rb

@@ -1,11 +1,13 @@
 module DownloadTV
+  ##
+  # Class used for managing the configuration of the application
   class Configuration
     attr_reader :content, :config_path
-    
-    def initialize(content={}, force_change=false)
-      FileUtils.mkdir_p(File.join(ENV["HOME"], ".config", "download_tv"))
-      @config_path = content[:path] || File.join(ENV["HOME"], ".config", "download_tv", "config")
-      
+
+    def initialize(content = {}, force_change = false)
+      FileUtils.mkdir_p(File.join(ENV['HOME'], '.config', 'download_tv'))
+      @config_path = content[:path] || File.join(ENV['HOME'], '.config', 'download_tv', 'config')
+
       if File.exist? @config_path
         load_config
         @content.merge!(content) unless content.empty?
@@ -17,48 +19,52 @@ module DownloadTV
       end
     end
 
-
     def change_configuration
       if @content[:myepisodes_user]
         print "Enter your MyEpisodes username (#{@content[:myepisodes_user]}) : "
       else
-        print "Enter your MyEpisodes username: "
+        print 'Enter your MyEpisodes username: '
       end
       @content[:myepisodes_user] = STDIN.gets.chomp
 
-      print "Save cookie? (y)/n: "
-      @content[:cookie] = STDIN.gets.chomp.downcase != "n"
+      print 'Save cookie? (y)/n: '
+      @content[:cookie] = !(STDIN.gets.chomp.casecmp? 'n')
 
       if @content[:ignored]
         puts "Enter a comma-separated list of shows to ignore: (#{@content[:ignored]})"
       else
-        puts "Enter a comma-separated list of shows to ignore: "
+        puts 'Enter a comma-separated list of shows to ignore: '
       end
-      
-      @content[:ignored] = STDIN.gets.chomp.split(",").map(&:strip).map(&:downcase)
+
+      @content[:ignored] = STDIN.gets.chomp.split(',').map(&:strip).map(&:downcase)
       STDOUT.flush
 
       # When modifying existing config, keeps previous values
       # When creating new one, sets defaults
       @content[:auto] ||= true
       @content[:subs] ||= true
-      @content[:grabber] ||= "TorrentAPI"
-      @content[:date] ||= Date.today-1
+      @content[:grabber] ||= 'TorrentAPI'
+      @content[:date] ||= Date.today - 1
       @content[:version] = DownloadTV::VERSION
 
       serialize
     end
 
-
     def serialize
-      File.open(@config_path, "wb") { |f| Marshal.dump(@content, f) }
+      File.write(@config_path, JSON.generate(@content))
     end
 
     def load_config
-      @content = File.open(@config_path, "rb") { |f| Marshal.load(f) }
-      if !@content[:version] || breaking_changes?(@content[:version])
-        change_configuration
-      end
+      source = File.read(@config_path)
+      @content = JSON.parse(source, symbolize_names: true)
+
+      @content[:date] = Date.parse(@content[:date]) if @content[:date]
+
+      change_configuration if !@content[:version] || breaking_changes?(@content[:version])
+    rescue JSON::ParserError
+      @content = {}
+      change_configuration
+      retry
     end
 
     ##
@@ -66,12 +72,11 @@ module DownloadTV
     # Returns false if a patch has been detected
     # Returns nil if it's the same version
     def breaking_changes?(version)
-      DownloadTV::VERSION.split(".").zip(version.split(".")).find_index { |x, y| y > x }&.< 2
+      DownloadTV::VERSION.split('.').zip(version.split('.')).find_index { |x, y| y > x }&.< 2
     end
-    
 
     def print_config
-      @content.each {|k, v| puts "#{k}: #{v}"}
+      @content.each { |k, v| puts "#{k}: #{v}" }
     end
   end
-end
+end

data/lib/download_tv/downloader.rb

@@ -4,8 +4,7 @@ module DownloadTV
   class Downloader
     attr_reader :offset, :config
 
-    def initialize(offset = 0, config = {})
-      @offset = offset.abs
+    def initialize(config = {})
       @config = Configuration.new(config) # Load configuration
 
       @filters = [
@@ -39,8 +38,8 @@ module DownloadTV
     ##
     # Finds download links for all new episodes aired since the last run of the program
     # It connects to MyEpisodes in order to find which shows to track and which new episodes aired.
-    def run(dont_update_last_run)
-      date = check_date
+    def run(dont_update_last_run, offset = 0)
+      date = check_date(offset)
 
       myepisodes = MyEpisodes.new(@config.content[:myepisodes_user], @config.content[:cookie])
       # Log in using cookie by default
@@ -120,10 +119,10 @@ module DownloadTV
       end
     end
 
-    def check_date
+    def check_date(offset)
       last = @config.content[:date]
-      if last - @offset != Date.today
-        last - @offset
+      if last - offset != Date.today
+        last - offset
       else
         puts 'Everything up to date'
         exit

data/lib/download_tv/grabbers/torrentapi.rb

@@ -10,6 +10,16 @@ module DownloadTV
       @wait = 2.1
     end
 
+    ##
+    # Specific implementation for TorrentAPI (requires token)
+    def online?
+      @agent.read_timeout = 2
+      @agent.get(format(@url, 'test', 'test'))
+      true
+    rescue Mechanize::ResponseCodeError, Net::HTTP::Persistent::Error
+      false
+    end
+
     ##
     # Connects to Torrentapi.org and requests a token.
     # Returns said token.
@@ -54,9 +64,6 @@ module DownloadTV
       links = obj['torrent_results'].collect { |i| i['download'] }
 
       names.zip(links)
-      # Temporary solution for Cloudflare being obnoxious
-    rescue Mechanize::ResponseCodeError
-      raise NoTorrentsError
     end
   end
 end

data/lib/download_tv/grabbers/tpb.rb

@@ -2,7 +2,7 @@ module DownloadTV
   ##
   # ThePirateBay grabber
   class ThePirateBay < LinkGrabber
-    def initialize(tpb_proxy = 'https://pirateproxy.cc')
+    def initialize(tpb_proxy = 'https://thepiratebay.cr')
       proxy = tpb_proxy.gsub(%r{/+$}, '') || 'https://thepiratebay.cr'
 
       super("#{proxy}/search/%s/0/7/0")

data/lib/download_tv/linkgrabber.rb

@@ -7,20 +7,19 @@ module DownloadTV
     def initialize(url)
       @url = url
       @agent = Mechanize.new
+      @agent.user_agent = DownloadTV::USER_AGENT
     end
 
-    def test_connection
-      agent = Mechanize.new
-      agent.read_timeout = 2
-      agent.get(@url)
+    def online?
+      @agent.read_timeout = 2
+      @agent.get(format(@url, 'test'))
+      true
+    rescue Mechanize::ResponseCodeError, Net::HTTP::Persistent::Error
+      false
     end
 
     def get_links(_s)
       raise NotImplementedError
     end
   end
-
-  class NoTorrentsError < StandardError; end
-
-  class NoSubtitlesError < StandardError; end
 end

data/lib/download_tv/myepisodes.rb

@@ -4,6 +4,7 @@ module DownloadTV
   class MyEpisodes
     def initialize(user, save_cookie)
       @agent = Mechanize.new
+      @agent.user_agent = DownloadTV::USER_AGENT
       @user = user
       @save_cookie = save_cookie
       @cookie_path = File.join(ENV['HOME'], '.config', 'download_tv', 'cookie')

data/lib/download_tv/torrent.rb

@@ -2,7 +2,7 @@ module DownloadTV
   ##
   # Class in charge of managing the link grabbers
   class Torrent
-    attr_reader :g_names, :g_instances, :n_grabbers
+    attr_reader :g_names, :g_instances, :tries
 
     def grabbers
       %w[Eztv KAT ThePirateBay TorrentAPI]
@@ -11,11 +11,11 @@ module DownloadTV
     def initialize(default_grabber = nil)
       @g_names = grabbers
       @g_instances = []
-      @n_grabbers = @g_names.size # Initial size
-      @tries = @n_grabbers - 1
+      reset_tries
 
       # Silently ignores bad names
-      @g_names.rotate! @g_names.find_index(default_grabber).to_i + 1
+      found = @g_names.find_index(default_grabber)
+      @g_names.rotate! found + 1 if found
 
       change_grabbers
     end
@@ -24,47 +24,43 @@ module DownloadTV
       if !@g_names.empty?
         # Instantiates the last element from g_names, popping it
         newt = (DownloadTV.const_get @g_names.pop).new
-        newt.test_connection
 
-        @g_instances.unshift newt
-
-      else
-        # Rotates the instantiated grabbers
+        if newt.online?
+          @g_instances.unshift newt
+        else
+          warn "Problem accessing #{newt.class.name}"
+          @tries -= 1 # We won't be using this grabber
+          change_grabbers
+        end
+      else # Rotates the instantiated grabbers
         @g_instances.rotate!
       end
-    rescue Mechanize::ResponseCodeError, Net::HTTP::Persistent::Error
-      warn "Problem accessing #{newt.class.name}"
-      # We won't be using this grabber
-      @n_grabbers -= 1
-      @tries = @n_grabbers - 1
-
-      change_grabbers
-    rescue SocketError, Errno::ECONNRESET, Net::OpenTimeout
-      warn 'Connection error.'
-      exit 1
     end
 
     def get_links(show)
       links = @g_instances.first.get_links(show)
 
-      # Reset the counter
-      @tries = @n_grabbers - 1
+      reset_tries
 
       links
     rescue NoTorrentsError
       puts "No torrents found for #{show} using #{@g_instances.first.class.name}"
 
       # Use next grabber
-      if @tries > 0
+      if @tries.positive?
         @tries -= 1
         change_grabbers
         retry
 
-      else # Reset the counter
-        @tries = @n_grabbers - 1
+      else
+        reset_tries
         # Handle show not found here!!
         return []
       end
     end
+
+    def reset_tries
+      @tries = @g_names.size + @g_instances.size - 1
+    end
   end
 end

data/lib/download_tv/version.rb

@@ -1,3 +1,3 @@
 module DownloadTV
-  VERSION = '2.2.2'.freeze
+  VERSION = '2.3.0'.freeze
 end

data/lib/download_tv.rb

@@ -11,4 +11,13 @@ require 'download_tv/torrent'
 require 'download_tv/myepisodes'
 require 'download_tv/linkgrabber'
 require 'download_tv/subtitles'
+
+module DownloadTV
+  USER_AGENT = "DownloadTV #{DownloadTV::VERSION}".freeze
+
+  class NoTorrentsError < StandardError; end
+
+  class NoSubtitlesError < StandardError; end
+end
+
 Dir[File.join(__dir__, 'download_tv', 'grabbers', '*.rb')].each { |file| require file }

data/test/config_test.rb

@@ -132,14 +132,17 @@ describe DownloadTV::Configuration do
   end
 
   describe 'the serialize method' do
-    it 'stores the configuration in a Marshalled file' do
+    it 'stores the configuration in a JSON file' do
       # Calls serialize
       run_silently do
         STDIN.stub :gets, 'anything' do
           DownloadTV::Configuration.new(path: config_path)
         end
       end
-      content = File.open(config_path, 'rb') { |f| Marshal.load(f) }
+      # content = File.open(config_path, 'rb') { |f| Marshal.load(f) }
+      source = File.read(config_path)
+      content = JSON.parse(source, symbolize_names: true)
+      content[:date] = Date.parse(content[:date])
 
       content[:cookie].must_equal true
       content[:myepisodes_user].must_equal 'anything'
@@ -147,7 +150,7 @@ describe DownloadTV::Configuration do
       content[:auto].must_equal true
       content[:subs].must_equal true
       content[:grabber].must_equal 'TorrentAPI'
-      content[:date].must_equal(Date.today - 1)
+      content[:date].must_equal Date.today - 1
       content[:version].must_equal DownloadTV::VERSION
     end
   end

data/test/downloader_test.rb

@@ -13,19 +13,10 @@ describe DownloadTV::Downloader do
   end
 
   describe 'when creating the object' do
-    it 'should store the first argument as @offset' do
-      DownloadTV::Downloader.new(3, path: config_path).offset.must_equal 3
-      DownloadTV::Downloader.new(-3, path: config_path).offset.must_equal 3
-    end
-
-    it 'should receive an integer for the offset' do
-      -> { DownloadTV::Downloader.new('foo') }.must_raise NoMethodError
-    end
-
     it 'can receive an optional configuration hash' do
-      dl = DownloadTV::Downloader.new(0, auto: true, grabber: 'KAT', path: config_path)
-      dl.config[:auto].must_equal true
-      dl.config[:grabber].must_equal 'KAT'
+      dl = DownloadTV::Downloader.new(auto: true, grabber: 'KAT', path: config_path)
+      dl.config.content[:auto].must_equal true
+      dl.config.content[:grabber].must_equal 'KAT'
     end
   end
 
@@ -34,7 +25,7 @@ describe DownloadTV::Downloader do
       shows = ['Mr. Foo S01E02', 'Bar (UK) S00E22', "Let's S05E03", 'Baz: The Story S05E22']
       result = ['Mr. Foo S01E02', 'Bar S00E22', 'Lets S05E03', 'Baz The Story S05E22']
 
-      dl = DownloadTV::Downloader.new(0, ignored: [], path: config_path)
+      dl = DownloadTV::Downloader.new(ignored: [], path: config_path)
       dl.fix_names(shows).must_equal result
     end
 
@@ -42,67 +33,67 @@ describe DownloadTV::Downloader do
       shows = ['Mr. Foo S01E02', 'Bar (UK) S00E22', 'Ignored S20E22', "Let's S05E03"]
       result = ['Mr. Foo S01E02', 'Bar S00E22', 'Lets S05E03']
 
-      dl = DownloadTV::Downloader.new(0, ignored: ['ignored'], path: config_path)
+      dl = DownloadTV::Downloader.new(ignored: ['ignored'], path: config_path)
       dl.fix_names(shows).must_equal result
     end
   end
 
   describe 'the check_date method' do
     it 'exits the script when up to date' do
-      dl = DownloadTV::Downloader.new(0, date: Date.today, path: config_path)
-      to_run = -> { run_silently { dl.check_date } }
+      dl = DownloadTV::Downloader.new(date: Date.today, path: config_path)
+      to_run = -> { run_silently { dl.check_date(0) } }
       to_run.must_raise SystemExit
     end
 
     it 'uses the offset to adjust the date' do
       # Would exit with offset 0
-      dl = DownloadTV::Downloader.new(1, date: Date.today, path: config_path)
+      dl = DownloadTV::Downloader.new(date: Date.today, path: config_path)
 
-      date = dl.check_date
+      date = dl.check_date(1)
 
       date.must_equal(Date.today - 1)
-      dl.config[:date].must_equal Date.today
+      dl.config.content[:date].must_equal Date.today
     end
   end
 
   describe 'the filter_shows method' do
     it 'removes names with 2160p in them' do
-      dl = DownloadTV::Downloader.new(0, path: config_path)
+      dl = DownloadTV::Downloader.new(path: config_path)
       links = [['Link 1', ''], ['Link 2 2160p', ''], ['Link 3', '']]
       res = [['Link 1', ''], ['Link 3', '']]
       dl.filter_shows(links).must_equal res
     end
 
     it 'removes names with 1080p in them' do
-      dl = DownloadTV::Downloader.new(0, path: config_path)
+      dl = DownloadTV::Downloader.new(path: config_path)
       links = [['Link.1080p', ''], ['Link 2 2160p', ''], ['Link 3', '']]
       res = [['Link 3', '']]
       dl.filter_shows(links).must_equal res
     end
 
     it 'removes names with 720p in them' do
-      dl = DownloadTV::Downloader.new(0, path: config_path)
+      dl = DownloadTV::Downloader.new(path: config_path)
       links = [['Link 1', ''], ['Link 2 720p', ''], ['Link.720p.rip', '']]
       res = [['Link 1', '']]
       dl.filter_shows(links).must_equal res
     end
 
     it 'removes names with WEB in them' do
-      dl = DownloadTV::Downloader.new(0, path: config_path)
+      dl = DownloadTV::Downloader.new(path: config_path)
       links = [['Link 1 WEBRIP', ''], ['Link 2 rip', ''], ['Link.720p.rip', '']]
       res = [['Link 2 rip', '']]
       dl.filter_shows(links).must_equal res
     end
 
     it 'removes names without PROPER or REPACK in them' do
-      dl = DownloadTV::Downloader.new(0, path: config_path)
+      dl = DownloadTV::Downloader.new(path: config_path)
       links = [['Link 1', ''], ['Link 2 2160p', ''], ['Link 3', ''], ['Link 4 PROPER', ''], ['Link REPACK 5', '']]
       res = [['Link 4 PROPER', ''], ['Link REPACK 5', '']]
       dl.filter_shows(links).must_equal res
     end
 
     it "doesn't apply a filter if it would reject every option" do
-      dl = DownloadTV::Downloader.new(0, path: config_path)
+      dl = DownloadTV::Downloader.new(path: config_path)
       links = [['Link 1 720p', ''], ['Link 2 2160p', ''], ['Link 720p 3', '']]
       res = [['Link 1 720p', ''], ['Link 720p 3', '']]
       dl.filter_shows(links).must_equal res
@@ -115,11 +106,12 @@ describe DownloadTV::Downloader do
       show = 'Example Show S01E01'
 
       t.expect(:get_links, [], [show])
-      dl = DownloadTV::Downloader.new(0, auto: true, path: config_path)
+      dl = DownloadTV::Downloader.new(auto: true, path: config_path)
       dl.get_link(t, show).must_equal ''
       t.expect(:get_links, [], [show])
-      dl = DownloadTV::Downloader.new(0, auto: false, path: config_path)
+      dl = DownloadTV::Downloader.new(auto: false, path: config_path)
       dl.get_link(t, show).must_equal ''
+      t.verify
     end
 
     it 'returns the first link when auto is set to true' do
@@ -127,8 +119,9 @@ describe DownloadTV::Downloader do
       show = 'Example Show S01E01'
 
       t.expect(:get_links, [['Name 1', 'Link 1'], ['Name 2', 'Link 2']], [show])
-      dl = DownloadTV::Downloader.new(0, auto: true, path: config_path)
+      dl = DownloadTV::Downloader.new(auto: true, path: config_path)
       dl.get_link(t, show).must_equal 'Link 1'
+      t.verify
     end
   end
 
@@ -137,7 +130,7 @@ describe DownloadTV::Downloader do
       prev = RbConfig::CONFIG['host_os']
       RbConfig::CONFIG['host_os'] = 'linux-gnu'
 
-      dl = DownloadTV::Downloader.new(0, path: config_path)
+      dl = DownloadTV::Downloader.new(path: config_path)
       dl.detect_os.must_equal 'xdg-open'
 
       RbConfig::CONFIG['host_os'] = prev
@@ -147,7 +140,7 @@ describe DownloadTV::Downloader do
       prev = RbConfig::CONFIG['host_os']
       RbConfig::CONFIG['host_os'] = 'darwin15.6.0'
 
-      dl = DownloadTV::Downloader.new(0, path: config_path)
+      dl = DownloadTV::Downloader.new(path: config_path)
       dl.detect_os.must_equal 'open'
 
       RbConfig::CONFIG['host_os'] = prev
@@ -157,7 +150,7 @@ describe DownloadTV::Downloader do
       prev = RbConfig::CONFIG['host_os']
       RbConfig::CONFIG['host_os'] = 'dummy'
 
-      dl = DownloadTV::Downloader.new(0, path: config_path)
+      dl = DownloadTV::Downloader.new(path: config_path)
 
       to_run = -> { run_silently { dl.detect_os.must_equal 'xdg-open' } }
       to_run.must_raise SystemExit

data/test/grabbers_test.rb

@@ -6,14 +6,12 @@ describe DownloadTV::LinkGrabber do
 
   instances.each do |grabber|
     describe grabber do
+      next unless grabber.online?
+
       it 'will have a url attribute on creation' do
         grabber.url.wont_be_nil
       end
 
-      it 'should get a 200 code response' do
-        grabber.test_connection.code.must_equal '200'
-      end
-
       it "will raise NoTorrentsError when torrent can't be found" do
         notfound = -> { grabber.get_links('Totally Fake Show askjdgsaudas') }
         notfound.must_raise DownloadTV::NoTorrentsError

data/test/test_helper.rb

@@ -4,7 +4,7 @@ require 'minitest/autorun'
 
 def create_dummy_config(in_path, config = {})
   config[:version] = DownloadTV::VERSION unless config[:version]
-  File.open(in_path, 'wb') { |f| Marshal.dump(config, f) }
+  File.write(in_path, JSON.generate(config))
 end
 
 def run_silently

data/test/torrent_test.rb

@@ -1,28 +1,44 @@
 require 'test_helper'
 
 describe DownloadTV::Torrent do
-  before do
-    @t = DownloadTV::Torrent.new
-  end
-
   describe 'when creating the object' do
-    it 'will have some grabbers' do
-      @t.g_names.empty?.must_equal false
-      @t.g_instances.empty?.must_equal false
-      @t.n_grabbers.must_be :>, 0
+    before do
+      @t = DownloadTV::Torrent.new
     end
 
     it 'will have the right amount of grabbers' do
-      # Initiakize calls change_grabbers
-      @t.n_grabbers.must_equal @t.g_names.size + 1
+      @t.g_names.size.must_equal @t.grabbers.size - 1
       @t.g_instances.size.must_equal 1
     end
 
     it 'will populate the instances' do
-      @t.n_grabbers.times.each { @t.change_grabbers }
+      @t.grabbers.size.times.each { @t.change_grabbers }
       @t.g_names.empty?.must_equal true
-      @t.g_instances.empty?.must_equal false
-      @t.g_instances.size.must_equal @t.n_grabbers
+      @t.g_instances.size.must_equal @t.grabbers.size
+    end
+
+    it 'will start with all tries available' do
+      @t.tries.must_equal @t.grabbers.size - 1
+    end
+
+    it 'will call get_links on its grabber' do
+      @t.g_instances.first.stub :get_links, %w[test result] do
+        @t.get_links('test show').must_equal %w[test result]
+      end
+    end
+  end
+
+  describe 'when giving it a default grabber' do
+    it 'has a default order' do
+      t = DownloadTV::Torrent.new(nil)
+      t.g_instances.first.class.name.must_equal 'DownloadTV::TorrentAPI'
+    end
+
+    %w[Eztv KAT ThePirateBay TorrentAPI].each do |g|
+      it 'correctly uses the given grabber first' do
+        t = DownloadTV::Torrent.new(g)
+        t.g_instances.first.class.name.must_equal "DownloadTV::#{g}"
+      end
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: download_tv
 version: !ruby/object:Gem::Version
-  version: 2.2.2
+  version: 2.3.0
 platform: ruby
 authors:
 - guille
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-08-06 00:00:00.000000000 Z
+date: 2017-09-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -134,7 +134,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.12
+rubygems_version: 2.6.13
 signing_key: 
 specification_version: 4
 summary: DownloadTV is a tool that allows the user to find magnet links for tv show