require 'test_helper'

class CanadaPostPwsRatingTest < Minitest::Test
  include ActiveShipping::Test::Fixtures

  def setup
    # 100 grams, 93 cm long, 10 cm diameter, cylinders have different volume calculations
    @pkg1 = Package.new(25, [93, 10], :cylinder => true)
    # 7.5 lbs, times 16 oz/lb., 15x10x4.5 inches, not grams, not centimetres
    @pkg2 = Package.new((7.5 * 16), [15, 10, 4.5], :units => :imperial)

    @home_params = {
      :name        => "John Smith",
      :company     => "test",
      :phone       => "613-555-1212",
      :address1    => "123 Elm St.",
      :city        => 'Ottawa',
      :province    => 'ON',
      :country     => 'CA',
      :postal_code => 'K1P 1J1'
    }
    @home = Location.new(@home_params)

    @dest_params = {
      :name     => "Frank White",
      :address1 => '999 Wiltshire Blvd',
      :city     => 'Beverly Hills',
      :state    => 'CA',
      :country  => 'US',
      :zip      => '90210'
    }
    @dest = Location.new(@dest_params)

    @shipping_opts1 = {:dc => true, :cod => :true, :cod_amount => 50.00, :cod_includes_shipping => true,
                       :cod_method_of_payment => 'CSH', :cov => true, :cov_amount => 100.00,
                       :so => true, :pa18 => true}

    @customer_number = '654321'

    credentials = { platform_id: 123, api_key: '456', secret: '789' }
    @cp = CanadaPostPWS.new(credentials)
    @cp.logger = Logger.new(StringIO.new)
    @french_cp = CanadaPostPWS.new(credentials.merge(language: 'fr'))
    @cp_customer_number = CanadaPostPWS.new(credentials.merge(customer_number: @customer_number))

    @default_options = {:customer_number => '123456'}
  end

  # rating

  def test_language_header
    assert_equal 'fr-CA', @french_cp.language
    assert_equal 'en-CA', @cp.language
  end

  def test_name_accessor
    assert_equal 'Canada Post PWS', @cp.name
  end

  def test_find_rates
    response = xml_fixture('canadapost_pws/rates_info')
    expected_headers = {
      'Authorization'   => "#{@cp.send(:encoded_authorization)}",
      'Accept-Language' => 'en-CA',
      'Accept'          => 'application/vnd.cpc.ship.rate+xml',
      'Content-Type'    => 'application/vnd.cpc.ship.rate+xml'
    }
    CanadaPostPWS.any_instance.expects(:ssl_post).with(anything, anything, expected_headers).returns(response)

    rates_response = @cp.find_rates(@home_params, @dest_params, [@pkg1, @pkg2], @default_options)

    assert_equal 4, rates_response.rates.size
    rate = rates_response.rates.first
    assert_equal RateEstimate, rate.class
    assert_equal "Canada Post PWS", rate.carrier
    assert_equal @home.to_s, Location.new(rate.origin).to_s
    assert_equal @dest.to_s, Location.new(rate.destination).to_s
  end

  def test_find_rates_with_error
    response = xml_fixture('canadapost_pws/rates_info_error')
    http_response = mock
    http_response.stubs(:code).returns('400')
    http_response.stubs(:body).returns(response)
    response_error = ActiveUtils::ResponseError.new(http_response)
    @cp.expects(:ssl_post).raises(response_error)

    exception = assert_raises ActiveShipping::ResponseError do
      @cp.find_rates(@home_params, @dest_params, [@pkg1, @pkg2], @default_options)
    end

    assert_equal "You cannot mail on behalf of the requested customer.", exception.message
  end

  def test_find_rates_line_items_single_object
    response = xml_fixture('canadapost_pws/rates_info')
    expected_headers = {
      'Authorization'   => "#{@cp.send(:encoded_authorization)}",
      'Accept-Language' => 'en-CA',
      'Accept'          => 'application/vnd.cpc.ship.rate+xml',
      'Content-Type'    => 'application/vnd.cpc.ship.rate+xml'
    }
    CanadaPostPWS.any_instance.expects(:ssl_post).with(anything, anything, expected_headers).returns(response)

    rates_response = @cp.find_rates(@home_params, @dest_params, @pkg1, @default_options)

    assert_equal 4, rates_response.rates.size
    rate = rates_response.rates.first
    assert_equal RateEstimate, rate.class
    assert_equal "Canada Post PWS", rate.carrier
    assert_equal @home.to_s, Location.new(rate.origin).to_s
    assert_equal @dest.to_s, Location.new(rate.destination).to_s
  end

  def test_find_rates_excluding_tax
    response = xml_fixture('canadapost_pws/rates_info')
    expected_headers = {
      'Authorization'   => "#{@cp.send(:encoded_authorization)}",
      'Accept-Language' => 'en-CA',
      'Accept'          => 'application/vnd.cpc.ship.rate+xml',
      'Content-Type'    => 'application/vnd.cpc.ship.rate+xml'
    }
    CanadaPostPWS.any_instance.expects(:ssl_post).with(anything, anything, expected_headers).returns(response).twice

    rates_tax_response = @cp.find_rates(@home_params, @dest_params, [@pkg1, @pkg2], {})
    rates_no_tax_response = @cp.find_rates(@home_params, @dest_params, [@pkg1, @pkg2], {exclude_tax: true})

    expected = [139, 407, 139, 240] #taken from fixture, tax in centsÏ
    assert_equal rates_no_tax_response.rates.size, rates_tax_response.rates.size
    expected.zip(rates_no_tax_response.rates, rates_tax_response.rates).each do |expected, no_tax, tax|
      assert_equal expected, tax.total_price - no_tax.total_price
    end
  end
  # build rates

  def test_build_rates_request
    xml = @cp.build_rates_request(@home_params, @dest_params, [@pkg1, @pkg2], @default_options)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!

    assert_equal @default_options[:customer_number], doc.at('//customer-number').text
    assert_equal 'K1P1J1', doc.at('//origin-postal-code').text
    assert_equal 'united-states', doc.at('//destination/*').name
    assert !doc.xpath('//parcel-characteristics').empty?
    assert_equal "3.427", doc.at('//parcel-characteristics//weight').text
  end

  def test_build_rates_request_use_carrier_customer_number
    xml = @cp_customer_number.build_rates_request(@home_params, @dest_params, [@pkg1, @pkg2])
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!

    assert_equal @customer_number, doc.at('//customer-number').text
    assert_equal 'K1P1J1', doc.at('//origin-postal-code').text
    assert_equal 'united-states', doc.at('//destination/*').name
    assert !doc.xpath('//parcel-characteristics').empty?
    assert_equal "3.427", doc.at('//parcel-characteristics//weight').text
  end

  def test_build_rates_request_override_carrier_customer_number
    xml = @cp_customer_number.build_rates_request(@home_params, @dest_params, [@pkg1, @pkg2], @default_options)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!

    assert_equal @default_options[:customer_number], doc.at('//customer-number').text
    assert_equal 'K1P1J1', doc.at('//origin-postal-code').text
    assert_equal 'united-states', doc.at('//destination/*').name
    assert doc.at('//parcel-characteristics')
    assert_equal "3.427", doc.at('//parcel-characteristics//weight').text
  end

  def test_build_rates_request_location_object
    xml = @cp.build_rates_request(Location.new(@home_params), Location.new(@dest_params), [@pkg1, @pkg2], @default_options)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!

    assert_equal @default_options[:customer_number], doc.at('//customer-number').text
    assert_equal 'K1P1J1', doc.at('//origin-postal-code').text
    assert_equal 'united-states', doc.at('//destination/*').name
    assert !doc.xpath('//parcel-characteristics').empty?
    assert_equal "3.427", doc.at('//parcel-characteristics//weight').text
  end

  def test_build_rates_request_domestic
    @dest_params = {
      :name        => "John Smith",
      :company     => "test",
      :phone       => "613-555-1212",
      :address1    => "123 Oak St.",
      :city        => 'Vanncouver',
      :province    => 'BC',
      :country     => 'CA',
      :postal_code => 'V5J 1J1'
    }
    xml = @cp.build_rates_request(@home_params, @dest_params, [@pkg1, @pkg2], @default_options)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!

    assert_equal 'K1P1J1', doc.at('//origin-postal-code').text
    assert_equal 'domestic', doc.at('//destination/*').name
    assert_equal 'V5J1J1', doc.at('//destination//postal-code').text
  end

  def test_build_rates_request_international
    @dest_params = {
      :name        => "John Smith",
      :company     => "test",
      :phone       => "613-555-1212",
      :address1    => "123 Tokyo St",
      :city        => 'Tokyo',
      :country     => 'JP'
    }
    xml = @cp.build_rates_request(@home_params, @dest_params, [@pkg1, @pkg2], @default_options)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!

    assert_equal 'K1P1J1', doc.xpath('//origin-postal-code').text
    assert_equal 'international', doc.at('//destination/*').name
    assert_equal 'JP', doc.at('//destination//country-code').text
  end

  def test_build_rates_request_with_cod_option
    opts = @default_options.merge(:cod => true, :cod_amount => 12.05)
    xml = @cp.build_rates_request(@home_params, @dest_params, [@pkg1, @pkg2], opts)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!
    assert_equal 'COD', doc.xpath('//options/option/option-code').text
    assert_equal '12.05', doc.xpath('//options/option/option-amount').text
  end

  def test_build_rates_request_with_signature_option
    opts = @default_options.merge(:so => true)
    xml = @cp.build_rates_request(@home_params, @dest_params, [@pkg1, @pkg2], opts)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!
    assert_equal 'SO', doc.at('//options/option/option-code').text
  end

  def test_build_rates_request_with_insurance_option
    opts = @default_options.merge(:cov => true, :cov_amount => 122.05)
    xml = @cp.build_rates_request(@home_params, @dest_params, [@pkg1, @pkg2], opts)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!
    assert_equal 'COV', doc.at('//options/option/option-code').text
    assert_equal '122.05', doc.at('//options/option/option-amount').text
  end

  def test_build_rates_request_with_other_options
    opts = @default_options.merge(:pa18 => true, :pa19 => true, :hfp => true, :dns => true, :lad => true)
    xml = @cp.build_rates_request(@home_params, @dest_params, [@pkg1, @pkg2], opts)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!
    options = doc.xpath('//options/option/option-code').map(&:text).sort
    assert_equal %w(PA18 PA19 HFP DNS LAD).sort, options
  end

  def test_build_rates_request_with_single_item
    opts = @default_options
    xml = @cp.build_rates_request(@home_params, @dest_params, [@pkg1], opts)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!
    assert_equal '0.025', doc.at('//parcel-characteristics/weight').text
  end

  def test_build_rates_request_with_mailing_tube
    pkg = Package.new(25, [93, 10], :cylinder => true)
    opts = @default_options
    xml = @cp.build_rates_request(@home_params, @dest_params, [pkg], opts)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!
    assert_equal 'true', doc.at('//parcel-characteristics/mailing-tube').text
  end

  def test_build_rates_request_with_oversize
    pkg = Package.new(25, [93, 10], :oversized => true)
    opts = @default_options
    xml = @cp.build_rates_request(@home_params, @dest_params, [pkg], opts)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!
    assert_equal 'true', doc.at('//parcel-characteristics/oversized').text
  end

  def test_build_rates_request_with_unpackaged
    pkg = Package.new(25, [93, 10], :unpackaged => true)
    opts = @default_options
    xml = @cp.build_rates_request(@home_params, @dest_params, [pkg], opts)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!
    assert_equal 'true', doc.at('//parcel-characteristics/unpackaged').text
  end

  def test_build_rates_request_with_zero_weight
    options = @default_options.merge(@shipping_opts1)
    line_items = [Package.new(0, [93, 10]), Package.new(0, [10, 10])]
    xml = @cp.build_rates_request(@home_params, @dest_params, line_items, options)
    doc = Nokogiri.XML(xml)
    doc.remove_namespaces!
    assert_equal '0.001', doc.at('//parcel-characteristics/weight').text
  end

  # parse response

  def test_parse_rates_response
    body = xml_fixture('canadapost_pws/rates_info')
    rates_response = @cp.parse_rates_response(body, @home, @dest, false)

    assert_equal 4, rates_response.rates.count
    rate = rates_response.rates.first
    assert_equal "DOM.EP", rate.service_code
    assert_equal 'Expedited Parcel', rate.service_name
    assert_equal @home, rate.origin
    assert_equal @dest, rate.destination
    assert_equal 1301, rate.total_price
  end

  def test_parse_rates_response_with_invalid_response_raises
    body = xml_fixture('canadapost_pws/rates_info_error')
    exception = assert_raises ActiveShipping::ResponseError do
      @response = @cp.parse_rates_response(body, @home, @dest, false)
    end
    assert_equal "No Quotes", exception.message
  end

  def test_parse_services_response
    body = xml_fixture('canadapost_pws/services_response')
    response = @cp.parse_services_response(body)
    assert_equal 6, response.size
    assert service = response['INT.PW.ENV']
    assert_equal "Priority Worldwide envelope INT'L", service[:name]
  end

  def test_parse_find_service_options_response
    body = xml_fixture('canadapost_pws/service_options_response')
    response = @cp.parse_service_options_response(body)
    assert_equal 3, response[:options].size
    assert_equal 0, response[:restrictions][:min_weight]
    assert_equal 30000, response[:restrictions][:max_weight]
    assert_equal 0.1, response[:restrictions][:min_length]
    assert_equal 0.1, response[:restrictions][:min_height]
    assert_equal 0.1, response[:restrictions][:min_width]
    assert_equal 150, response[:restrictions][:max_length]
    assert_equal 150, response[:restrictions][:max_height]
    assert_equal 150, response[:restrictions][:max_width]
  end

  def test_parse_find_option_response
    body = xml_fixture('canadapost_pws/option_response')
    response = @cp.parse_option_response(body)
    assert_equal "SO", response[:code]
    assert_equal "Signature option", response[:name]
    assert_equal "FEAT", response[:class]
    assert_equal true, response[:prints_on_label]
    assert_equal false, response[:qualifier_required]
    assert_equal 1, response[:conflicting_options].size
    assert_equal "LAD", response[:conflicting_options][0]
    assert_equal 1, response[:prerequisite_options].size
    assert_equal "DC", response[:prerequisite_options][0]
  end

  def test_parse_find_option_response_no_conflicts_or_prereqs
    body = xml_fixture('canadapost_pws/option_response_no_conflicts')
    response = @cp.parse_option_response(body)
    assert_equal "SO", response[:code]
    assert_equal "Signature option", response[:name]
    assert_equal "FEAT", response[:class]
    assert_equal true, response[:prints_on_label]
    assert_equal false, response[:qualifier_required]
    assert response[:conflicting_options].blank?
    assert response[:prerequisite_options].blank?
  end

  def test_error_response_includes_error_code
    response = xml_fixture('canadapost_pws/rates_info_error')
    e = assert_raises ActiveShipping::ResponseError do
      @cp.error_response(response, CPPWSRateResponse)
    end
    assert_equal 'AA004', e.response.error_code
  end
end