# frozen_string_literal: true

require 'tc_helper'

class TestContentType < Test::Unit::TestCase
  def setup
    @package = Axlsx::Package.new
    @doc = Nokogiri::XML(@package.send(:content_types).to_xml_string)
  end

  def test_valid_document
    schema = Nokogiri::XML::Schema(File.open(Axlsx::CONTENT_TYPES_XSD))

    assert_empty(schema.validate(@doc).map { |e| puts e.message; e.message })
  end

  def test_pre_built_types
    o_path = "//xmlns:Override[@ContentType='%s']"
    d_path = "//xmlns:Default[@ContentType='%s']"

    # default
    assert_equal(2, @doc.xpath("//xmlns:Default").size, "There should be 2 default types")

    node = @doc.xpath(format(d_path, Axlsx::XML_CT)).first

    assert_equal(node["Extension"], Axlsx::XML_EX.to_s, "xml content type invalid")

    node = @doc.xpath(format(d_path, Axlsx::RELS_CT)).first

    assert_equal(node["Extension"], Axlsx::RELS_EX.to_s, "relationships content type invalid")

    # overrride
    assert_equal(4, @doc.xpath("//xmlns:Override").size, "There should be 4 Override types")

    node = @doc.xpath(format(o_path, Axlsx::APP_CT)).first

    assert_equal(node["PartName"], "/#{Axlsx::APP_PN}", "App part name invalid")

    node = @doc.xpath(format(o_path, Axlsx::CORE_CT)).first

    assert_equal(node["PartName"], "/#{Axlsx::CORE_PN}", "Core part name invalid")

    node = @doc.xpath(format(o_path, Axlsx::STYLES_CT)).first

    assert_equal(node["PartName"], "/xl/#{Axlsx::STYLES_PN}", "Styles part name invalid")

    node = @doc.xpath(format(o_path, Axlsx::WORKBOOK_CT)).first

    assert_equal(node["PartName"], "/#{Axlsx::WORKBOOK_PN}", "Workbook part invalid")
  end

  def test_should_get_worksheet_for_worksheets
    o_path = "//xmlns:Override[@ContentType='%s']"

    ws = @package.workbook.add_worksheet
    doc = Nokogiri::XML(@package.send(:content_types).to_xml_string)

    assert_equal(5, doc.xpath("//xmlns:Override").size, "adding a worksheet should add another type")
    assert_equal(doc.xpath(format(o_path, Axlsx::WORKSHEET_CT)).last["PartName"], "/xl/#{ws.pn}", "Worksheet part invalid")

    ws = @package.workbook.add_worksheet
    doc = Nokogiri::XML(@package.send(:content_types).to_xml_string)

    assert_equal(6, doc.xpath("//xmlns:Override").size, "adding workship should add another type")
    assert_equal(doc.xpath(format(o_path, Axlsx::WORKSHEET_CT)).last["PartName"], "/xl/#{ws.pn}", "Worksheet part invalid")
  end

  def test_drawings_and_charts_need_content_types
    o_path = "//xmlns:Override[@ContentType='%s']"
    ws = @package.workbook.add_worksheet

    c = ws.add_chart Axlsx::Pie3DChart
    doc = Nokogiri::XML(@package.send(:content_types).to_xml_string)

    assert_equal(7, doc.xpath("//xmlns:Override").size, "expected 7 types got #{doc.css('Types Override').size}")
    assert_equal(doc.xpath(format(o_path, Axlsx::DRAWING_CT)).first["PartName"], "/xl/#{ws.drawing.pn}", "Drawing part name invlid")
    assert_equal(doc.xpath(format(o_path, Axlsx::CHART_CT)).last["PartName"], "/xl/#{c.pn}", "Chart part name invlid")

    c = ws.add_chart Axlsx::Pie3DChart
    doc = Nokogiri::XML(@package.send(:content_types).to_xml_string)

    assert_equal(8, doc.xpath("//xmlns:Override").size, "expected 7 types got #{doc.css('Types Override').size}")
    assert_equal(doc.xpath(format(o_path, Axlsx::CHART_CT)).last["PartName"], "/xl/#{c.pn}", "Chart part name invlid")
  end
end