# Copyright 2012 Red Hat, Inc.
#
# This software is licensed to you under the GNU General Public
# License as published by the Free Software Foundation; either version
# 2 of the License (GPLv2) or (at your option) any later version.
# There is NO WARRANTY for this software, express or implied,
# including the implied warranties of MERCHANTABILITY,
# NON-INFRINGEMENT, or FITNESS FOR A PARTICULAR PURPOSE. You should
# have received a copy of GPLv2 along with this software; if not, see
# http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt.

require 'rubygems'
require 'minitest/autorun'

require './test/integration/resources/helpers/repository_helper'
require './lib/runcible/resources/repository'
require './lib/runcible/extensions/repository'


module TestPulpRepositoryBase
  include RepositoryHelper

  def setup
    @resource = Runcible::Pulp::Repository
    @extension = Runcible::Pulp::RepositoryExtension
    VCR.insert_cassette('pulp_repository')
  end

  def teardown
    VCR.eject_cassette
  end

end

=begin
class TestPulpRepositoryCreate < MiniTest::Unit::TestCase
  include TestPulpRepositoryBase

  def teardown
    RepositoryHelper.destroy_repo
    super
  end

  def test_create
    response = RepositoryHelper.create_repo
    assert response.code == 201
    assert response['id'] == RepositoryHelper.repo_id
  end
end


class TestPulpRepositoryDelete < MiniTest::Unit::TestCase
  include TestPulpRepositoryBase

  def setup
    super
    RepositoryHelper.create_repo
  end

  def test_delete
    response = @resource.delete(RepositoryHelper.repo_id)
    assert response.code == 200
  end

end


class TestPulpRepository < MiniTest::Unit::TestCase
  include TestPulpRepositoryBase
  
  def self.before_suite
    RepositoryHelper.create_repo(:importer => true)
  end

  def self.after_suite
    RepositoryHelper.destroy_repo
  end

  def test_repository_path
    path = @resource.path
    assert_match("repositories/", path)
  end

  def test_repository_path_with_id
    path = @resource.path(RepositoryHelper.repo_id)
    assert_match("repositories/#{RepositoryHelper.repo_id}", path)
  end

  def test_update
    response = @resource.update(RepositoryHelper.repo_id, { :description => "updated_description_" + RepositoryHelper.repo_id })
    assert response.code == 200
    assert response["description"] == "updated_description_" + RepositoryHelper.repo_id
  end

  def test_retrieve
    response = @resource.retrieve(RepositoryHelper.repo_id)
    assert response.code == 200
    assert response["display_name"] == RepositoryHelper.repo_id
  end

  def test_retrieve_with_details
    response = @resource.retrieve(RepositoryHelper.repo_id, {:details => true})
    assert response.code == 200
    assert response.has_key?("distributors")
    assert response.has_key?("importers")
  end

  def test_retrieve_all
    response = @resource.retrieve_all()
    assert response.code == 200
    assert response.collect{ |repo| repo["display_name"] == RepositoryHelper.repo_id }.length > 0
  end

  def test_search
    response = @resource.search({})
    assert response.code == 200
    assert response.collect{ |repo| repo["display_name"] == RepositoryHelper.repo_id }.length > 0
  end

  def test_associate_importer
    response = @resource.associate_importer(RepositoryHelper.repo_id, "yum_importer", {})
    assert response.code == 201
    assert response['importer_type_id'] == "yum_importer"
  end

  def test_associate_distributor
    distributor_config = {"relative_url" => "/", "http" => true, "https" => true}

    response = @resource.associate_distributor(RepositoryHelper.repo_id, "yum_distributor", distributor_config)
    assert response.code == 201
    assert response['distributor_type_id'] == "yum_distributor"
  end

end


class TestPulpRepositorySync < MiniTest::Unit::TestCase
  include TestPulpRepositoryBase
  
  def setup
    super
    VCR.eject_cassette
    VCR.insert_cassette('pulp_repository_sync')
  end

  def teardown
    RepositoryHelper.destroy_repo
    super
  end

  def test_sync_repo
    RepositoryHelper.create_repo
    response = @resource.sync(RepositoryHelper.repo_id)
    RepositoryHelper.task = response[0]

    assert response.code == 202
    assert response.length == 1
    assert response[0]["tags"].include?('pulp:action:sync')
  end

  def test_sync_repo_with_yum_importer
    RepositoryHelper.create_repo(:importer => true)
    response = @resource.sync(RepositoryHelper.repo_id)
    RepositoryHelper.task = response.first

    assert response.code == 202
    assert response.length == 1
    assert response.first["tags"].include?('pulp:action:sync')
  end
end
=end

class TestPulpRepositoryClone < MiniTest::Unit::TestCase
  include TestPulpRepositoryBase

  def setup
    super
    @clone_name = RepositoryHelper.repo_id + "_clone"
    RepositoryHelper.destroy_repo(@clone_name)
    RepositoryHelper.destroy_repo
    RepositoryHelper.create_and_sync_repo(:importer => true)
    @extension.create_with_importer(@clone_name, "yum_importer", {})
  end

  def teardown
    RepositoryHelper.destroy_repo(@clone_name)
    RepositoryHelper.destroy_repo
    super
  end

  def test_unit_copy
    response = @resource.unit_copy(@clone_name, RepositoryHelper.repo_id)
    RepositoryHelper.task = response
    assert response.code == 202
    assert response['tags'].include?('pulp:action:associate')
  end

end


=begin

  def test_discovery
    response = @resource.start_discovery(RepositoryHelper.repo_url, 'yum')
    assert response.length > 0
  end

  def test_all
    response = @resource.all()
    assert response.length > 0
    assert response.select { |r| r["name"] == RepositoryHelper.repo_id }.length > 0
  end

  def test_update_schedule
    response = @resource.update_schedule(RepositoryHelper.repo_id, "R1/" << Time.now.advance(:years => 1).iso8601 << "/P1D")
    assert JSON.parse(response)["id"] == RepositoryHelper.repo_id
    @resource.delete_schedule(RepositoryHelper.repo_id)
  end

  def test_delete_schedule
    @resource.update_schedule(RepositoryHelper.repo_id, "R1/" << Time.now.advance(:years => 1).iso8601 << "/P1D")
    response = @resource.delete_schedule(RepositoryHelper.repo_id)
    assert JSON.parse(response)["id"] == RepositoryHelper.repo_id
  end

  def test_add_filters
    response = @resource.add_filters(RepositoryHelper.repo_id, [FilterHelper.filter_id])
    assert response == "true"
  end

  def test_remove_filters
    response = @resource.remove_filters(RepositoryHelper.repo_id, [FilterHelper.filter_id])
    assert response == "true"
  end

  def test_destroy
    response = @resource.destroy(RepositoryHelper.repo_id)
    assert response == 202
  end
end


class TestPulpRepositoryRequiresSync < MiniTest::Unit::TestCase
  include TestPulpRepositoryBase

  def self.before_suite
    RepositoryHelper.create_and_sync_repo
  end

  def self.after_suite
    RepositoryHelper.destroy_repo
  end

  def test_packages
    response = @resource.packages(RepositoryHelper.repo_id)
    assert response.length > 0
  end

  def test_packages_by_name
    response = @resource.packages_by_name(RepositoryHelper.repo_id, "cheetah")
    assert response.length > 0
    assert response.select { |r| r["name"] == "cheetah" }.length > 0
  end

  def test_packages_by_nvre
    response = @resource.packages_by_nvre(RepositoryHelper.repo_id, "cheetah", "0.3", "0.8", "")
    assert response.length > 0
    assert response.select { |r| r["name"] == "cheetah" }.length > 0
  end

  def test_errata
    response = @resource.errata(RepositoryHelper.repo_id)
    assert response.length > 0
  end

  def test_errata_with_filter
    response = @resource.errata(RepositoryHelper.repo_id, { :type => 'security' })
    assert response.length > 0
    assert response.select { |errata| errata['id'] == "RHEA-2010:0002" }.length > 0
  end

  def test_distributions
    response = @resource.distributions(RepositoryHelper.repo_id)
    assert response.kind_of?(Array)
  end

  def test_sync_history
    response = @resource.sync_history(RepositoryHelper.repo_id)
    assert response.length > 0
  end

  def test_add_packages
    response = @resource.add_packages(RepositoryHelper.repo_id, [])
    assert response == "[[], 0]"
  end

  def test_add_errata 
    response = @resource.add_errata(RepositoryHelper.repo_id, ["RHEA-2010:0002"])
    assert response == "[]"
  end

  def test_add_distribution
    response = @resource.add_distribution(RepositoryHelper.repo_id, "ks-Test Family-TestVariant-16-x86_64")
    assert response == "true"
  end
end
=end