# frozen_string_literal: true

require 'tc_helper'

class TestBorderCreator < Test::Unit::TestCase
  def setup
    @p = Axlsx::Package.new
    @wb = @p.workbook
    @ws = @wb.add_worksheet
  end

  def test_initialize
    @ws.add_row [1, 2, 3]

    bc = Axlsx::BorderCreator.new(worksheet: @ws, cells: @ws["A1:B1"])

    assert_equal bc.instance_variable_get(:@edges), Axlsx::Border::EDGES
    assert_equal(:thin, bc.instance_variable_get(:@style))
    assert_equal("000000", bc.instance_variable_get(:@color))

    bc = Axlsx::BorderCreator.new(worksheet: @ws, cells: @ws["A1:B1"], edges: [:top], style: :thick, color: "ffffff")

    assert_equal([:top], bc.instance_variable_get(:@edges))
    assert_equal(:thick, bc.instance_variable_get(:@style))
    assert_equal("ffffff", bc.instance_variable_get(:@color))
  end

  def test_initialize_edges
    @ws.add_row [1, 2, 3]

    bc = Axlsx::BorderCreator.new(worksheet: @ws, cells: @ws["A1:B1"], edges: nil)

    assert_equal bc.instance_variable_get(:@edges), Axlsx::Border::EDGES

    bc = Axlsx::BorderCreator.new(worksheet: @ws, cells: @ws["A1:B1"], edges: :all)

    assert_equal bc.instance_variable_get(:@edges), Axlsx::Border::EDGES

    bc = Axlsx::BorderCreator.new(worksheet: @ws, cells: @ws["A1:B1"], edges: [])

    assert_empty(bc.instance_variable_get(:@edges))

    assert_raises(ArgumentError) do
      bc = Axlsx::BorderCreator.new(worksheet: @ws, cells: @ws["A1:B1"], edges: [:foo])
    end

    assert_raises(ArgumentError) do
      bc = Axlsx::BorderCreator.new(worksheet: @ws, cells: @ws["A1:B1"], edges: :foo)
    end

    assert_raises(ArgumentError) do
      bc = Axlsx::BorderCreator.new(worksheet: @ws, cells: @ws["A1:B1"], edges: [nil])
    end
  end

  def test_draw
    5.times do
      @ws.add_row [1, 2, 3, 4, 5]
    end

    bc = Axlsx::BorderCreator.new(worksheet: @ws, cells: @ws["A1:C3"], edges: ["top", :left], style: :thick, color: "ffffff")

    bc.draw

    assert_equal 2, @ws.styles.borders.size

    @wb.apply_styles

    assert_equal 5, @ws.styles.borders.size

    assert_equal 2, @ws.styles.borders[2].prs.size
    assert_equal ["FFFFFFFF"], @ws.styles.borders[2].prs.map(&:color).map(&:rgb).uniq
    assert_equal [:thick], @ws.styles.borders[2].prs.map(&:style).uniq
    assert_equal [:left, :top], @ws.styles.borders[2].prs.map(&:name)

    assert_equal 1, @ws.styles.borders[3].prs.size
    assert_equal ["FFFFFFFF"], @ws.styles.borders[3].prs.map(&:color).map(&:rgb).uniq
    assert_equal [:thick], @ws.styles.borders[3].prs.map(&:style).uniq
    assert_equal [:top], @ws.styles.borders[3].prs.map(&:name)

    assert_equal 1, @ws.styles.borders[4].prs.size
    assert_equal ["FFFFFFFF"], @ws.styles.borders[4].prs.map(&:color).map(&:rgb).uniq
    assert_equal [:thick], @ws.styles.borders[4].prs.map(&:style).uniq
    assert_equal [:left], @ws.styles.borders[4].prs.map(&:name)
  end
end