# encoding: UTF-8
require File.dirname(__FILE__) + '/test_helper.rb'
require 'media_wiki'

class TestPluginWikipedia < Test::Unit::TestCase
  @@files = {
    'flpo' => 'Flashpoint.S04E04.German.Dubbed.WEB-DL.XViD.avi',
    'dani' => 'Dr.Dani.Santino.S01E04.German.Dubbed.WEB-DL.XViD.avi',
    'two'  => 'Two.and.a.half.Men.S09E07.German.Dubbed.WS.WEB-DL.XviD-GDR.avi',
    'simp' => 'Die.Simpsons.S09E07.German.Dubbed.WS.WEB-DL.XviD-GDR.avi',
    'sea'  => 'tcpa-seapatrol_s05e11.avi',
  }

  @@directories = {
    'chuck'  => 'Chuck.S01E01.German.Dubbed.BLURAYRiP.WEB-DL',
    'chuck2' => 'Chuck.S02E10.German.Dubbed.BLURAYRiP.WEB-DL',
  }

  def setup
    TestHelper.create_test_files(@@files.values)
    TestHelper.create_test_dirs(@@directories.values)
    TestHelper.cwd
  end

  def teardown
    TestHelper.clean
  end

  def test_episode_list_page_data_extraction

    VCR.use_cassette("wiki_#{method_name}") do
      wiki = MediaWiki::Gateway.new('http://de.wikipedia.org/w/api.php')

      data = wiki.get("Liste der Episoden von Bones – Die Knochenjägerin")
      seasons = Plugin::Wikipedia.parse_episodelist_page_data(data)

      assert_equal("Die Frau im Teich", seasons[1][1])
      assert_equal("Ein Verräter im Feuer", seasons[2][11])
      assert_equal("Vom Vergehen und Werden", seasons[6][23])

      data = wiki.get("Liste der Criminal-Minds-Episoden")
      seasons = Plugin::Wikipedia.parse_episodelist_page_data(data)

      assert_equal("Der Abgrund", seasons[1][1])
      assert_equal("Rivalen", seasons[2][9])
      assert_equal("Valhalla", seasons[6][17])

      data = wiki.get("Liste der Dexter-Episoden")
      seasons = Plugin::Wikipedia.parse_episodelist_page_data(data)

      assert_equal("Rot wie Blut", seasons[1][10])
      assert_equal("Hitzewelle", seasons[2][4])
      assert_equal("Familienväter", seasons[4][6])

      data = wiki.get("Liste der Simpsons-Episoden")
      seasons = Plugin::Wikipedia.parse_episodelist_page_data(data, true)

      assert_equal("Es weihnachtet schwer", seasons[1][1])
      assert_equal("Bösartige Spiele", seasons[4][5])
      assert_equal("Homer an der Uni", seasons[5][3])
      assert_equal("Hochzeit auf Indisch", seasons[9][7])
      assert_equal("G.I. Homer", seasons[18][5])
      assert_equal("Die Farbe Gelb", seasons[21][13])

      data = wiki.get("Liste der Misfits-Episoden")
      seasons = Plugin::Wikipedia.parse_episodelist_page_data(data)

      assert_equal("Das Gewitter", seasons[1][1])
      assert_equal("Nathan wird Vater", seasons[2][7])
    end
  end

  def test_inpage_episode_list_page_data_extraction

    VCR.use_cassette("wiki_#{method_name}") do
      wiki = MediaWiki::Gateway.new('http://de.wikipedia.org/w/api.php')

      data = wiki.get("The Glades")
      seasons = Plugin::Wikipedia.parse_inarticle_episodelist_page_data(data)

      assert_equal("Reptilien im Paradies", seasons[1][1])
      assert_equal("Doppelgänger", seasons[1][6])
      assert_equal("Unruhiges Blut", seasons[1][9])

      data = wiki.get("Dr. Dani Santino – Spiel des Lebens")
      seasons = Plugin::Wikipedia.parse_inarticle_episodelist_page_data(data)

      assert_equal("Touchdown", seasons[1][1])
      assert_equal("Zickenkrieg", seasons[1][7])

      # the following series have an old inarticle episodelist
      data = wiki.get("Prison Break")
      seasons = Plugin::Wikipedia.parse_inarticle_episodelist_page_data(data)

      assert_equal("Der große Plan", seasons[1][1])
      assert_equal("Seite 1213", seasons[2][5])
    end
  end


  def test_episode_information_generation

    VCR.use_cassette("wiki_#{method_name}") do
      two = Serienrenamer::Episode.new(@@files['two'])
      data = Plugin::Wikipedia.generate_episode_information(two)[0]
      two.add_episode_information(data, false) if data
      assert_equal("S09E07 - Das Tagebuch.avi", two.to_s)

      simp = Serienrenamer::Episode.new(@@files['simp'])
      data = Plugin::Wikipedia.generate_episode_information(simp)[0]
      simp.add_episode_information(data, false) if data
      assert_equal("S09E07 - Hochzeit auf Indisch.avi", simp.to_s)

      chuck = Serienrenamer::Episode.new(@@directories['chuck'])
      data = Plugin::Wikipedia.generate_episode_information(chuck)[0]
      chuck.add_episode_information(data, false) if data
      assert_equal("S01E01 - Pilot.avi", chuck.to_s)

      chuck2 = Serienrenamer::Episode.new(@@directories['chuck2'])
      data = Plugin::Wikipedia.generate_episode_information(chuck2)[0]
      chuck2.add_episode_information(data, false) if data
      assert_equal("S02E10 - Chuck gegen zehn Millionen.avi", chuck2.to_s)

      dani = Serienrenamer::Episode.new(@@files['dani'])
      data = Plugin::Wikipedia.generate_episode_information(dani)[0]
      dani.add_episode_information(data, false) if data
      assert_equal("S01E04 - Gewohnheiten.avi", dani.to_s)

      sea = Serienrenamer::Episode.new(@@files['sea'])
      data = Plugin::Wikipedia.generate_episode_information(sea)[0]
      sea.add_episode_information(data, false) if data
      assert_equal("S05E11 - Der Morgen danach.avi", sea.to_s)
    end
  end

  def test_check_for_series_main_page

    VCR.use_cassette("wiki_#{method_name}") do
      wiki = MediaWiki::Gateway.new('http://de.wikipedia.org/w/api.php')

      assert_equal(false, Plugin::Wikipedia.is_series_main_page?(wiki.get("Bones")))
      assert_equal(true,  Plugin::Wikipedia.is_series_main_page?(wiki.get("Bones – Die Knochenjägerin")))
      assert_equal(false, Plugin::Wikipedia.is_series_main_page?(wiki.get("Bones – Der Tod ist erst der Anfang")))
      assert_equal(false, Plugin::Wikipedia.is_series_main_page?(wiki.get("Chuck")))
      assert_equal(true,  Plugin::Wikipedia.is_series_main_page?(wiki.get("Chuck (Fernsehserie)")))
      assert_equal(false, Plugin::Wikipedia.is_series_main_page?(wiki.get("Chuck (Album)")))
    end
  end

  def test_check_for_disambiguation_page

    VCR.use_cassette("wiki_#{method_name}") do
      wiki = MediaWiki::Gateway.new('http://de.wikipedia.org/w/api.php')

      assert_equal(true,  Plugin::Wikipedia.is_disambiguation_site?(wiki.get("Bones")))
      assert_equal(false, Plugin::Wikipedia.is_disambiguation_site?(wiki.get("Bones – Die Knochenjägerin")))
      assert_equal(true,  Plugin::Wikipedia.is_disambiguation_site?(wiki.get("Chuck")))
      assert_equal(false, Plugin::Wikipedia.is_disambiguation_site?(wiki.get("Chuck (Fernsehserie)")))
      assert_equal(false, Plugin::Wikipedia.is_disambiguation_site?(wiki.get("Chuck (Album)")))
    end

  end

  def test_check_for_inarticle_episode_list

    VCR.use_cassette("wiki_#{method_name}") do
      wiki = MediaWiki::Gateway.new('http://de.wikipedia.org/w/api.php')

      assert_equal(false, Plugin::Wikipedia.contains_inarticle_episode_list?(wiki.get("Bones")))
      assert_equal(true,  Plugin::Wikipedia.contains_inarticle_episode_list?(wiki.get("The Glades")))
      assert_equal(false,  Plugin::Wikipedia.contains_inarticle_episode_list?(wiki.get("Royal Pains")))
    end

  end

end