#pragma once

#include <algorithm>
#include <iostream>
#include <cxxtest/TestSuite.h>

#include <xlnt/xlnt.hpp>
#include "helpers/temporary_file.hpp"

//checked with 52f25d6

class test_workbook : public CxxTest::TestSuite
{
public:
    void test_active_sheet()
    {
        xlnt::workbook wb;
        TS_ASSERT_EQUALS(wb.active_sheet(), wb[0]);
    }

    void test_create_sheet()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet();
        auto last = std::distance(wb.begin(), wb.end()) - 1;
        TS_ASSERT_EQUALS(new_sheet, wb[last]);
    }

    void test_add_correct_sheet()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet();
        new_sheet.cell("A6").value(1.498);
        wb.copy_sheet(new_sheet);
        TS_ASSERT(wb[1].compare(wb[2], false));
        wb.create_sheet().cell("A6").value(1.497);
        TS_ASSERT(!wb[1].compare(wb[3], false));
    }

    void test_add_sheet_from_other_workbook()
    {
        xlnt::workbook wb1, wb2;
        auto new_sheet = wb1.active_sheet();
        TS_ASSERT_THROWS(wb2.copy_sheet(new_sheet), xlnt::invalid_parameter);
        TS_ASSERT_THROWS(wb2.index(new_sheet), std::runtime_error);
    }

    void test_add_sheet_at_index()
    {
        xlnt::workbook wb;
        auto ws = wb.active_sheet();
        ws.cell("B3").value(2);
        ws.title("Active");
        wb.copy_sheet(ws, 0);
        TS_ASSERT_EQUALS(wb.sheet_titles().at(0), "Sheet1");
        TS_ASSERT_EQUALS(wb.sheet_by_index(0).cell("B3").value<int>(), 2);
        TS_ASSERT_EQUALS(wb.sheet_titles().at(1), "Active");
        TS_ASSERT_EQUALS(wb.sheet_by_index(1).cell("B3").value<int>(), 2);
    }
    
    void test_remove_sheet()
    {
        xlnt::workbook wb, wb2;
        auto new_sheet = wb.create_sheet(0);
		new_sheet.title("removed");
        wb.remove_sheet(new_sheet);
        TS_ASSERT(!wb.contains("removed"));
        TS_ASSERT_THROWS(wb.remove_sheet(wb2.active_sheet()), std::runtime_error);
    }

    void test_get_sheet_by_title()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet();
        std::string title = "my sheet";
        new_sheet.title(title);
        auto found_sheet = wb.sheet_by_title(title);
        TS_ASSERT_EQUALS(new_sheet, found_sheet);
        TS_ASSERT_THROWS(wb.sheet_by_title("error"), xlnt::key_not_found);
        const auto &wb_const = wb;
        TS_ASSERT_THROWS(wb_const.sheet_by_title("error"), xlnt::key_not_found);
    }
    
    void test_get_sheet_by_title_const()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet();
        std::string title = "my sheet";
        new_sheet.title(title);
        const xlnt::workbook& wbconst = wb;
        auto found_sheet = wbconst.sheet_by_title(title);
        TS_ASSERT_EQUALS(new_sheet, found_sheet);
    }

    void test_index_operator() // test_getitem
    {
        xlnt::workbook wb;
        TS_ASSERT_THROWS_NOTHING(wb["Sheet1"]);
        TS_ASSERT_THROWS(wb["NotThere"], xlnt::key_not_found);
    }
    
    // void test_delitem() {} doesn't make sense in c++
    
    void test_contains()
    {
        xlnt::workbook wb;
        TS_ASSERT(wb.contains("Sheet1"));
        TS_ASSERT(!wb.contains("NotThere"));
    }
    
    void test_iter()
    {
        xlnt::workbook wb;
        
        for(auto ws : wb)
        {
            TS_ASSERT_EQUALS(ws.title(), "Sheet1");
        }
    }
    
    void test_get_index()
    {
        xlnt::workbook wb;
        wb.create_sheet().title("1");
        wb.create_sheet().title("2");

        auto sheet_index = wb.index(wb.sheet_by_title("1"));
        TS_ASSERT_EQUALS(sheet_index, 1);

        sheet_index = wb.index(wb.sheet_by_title("2"));
        TS_ASSERT_EQUALS(sheet_index, 2);
    }

    void test_get_sheet_names()
    {
        xlnt::workbook wb;
        wb.create_sheet().title("test_get_sheet_titles");

		const std::vector<std::string> expected_titles = { "Sheet1", "test_get_sheet_titles" };

		TS_ASSERT_EQUALS(wb.sheet_titles(), expected_titles);
    }
    
    void test_add_named_range()
    {
        xlnt::workbook wb, wb2;
        auto new_sheet = wb.create_sheet();
        wb.create_named_range("test_nr", new_sheet, "A1");
        TS_ASSERT(new_sheet.has_named_range("test_nr"));
        TS_ASSERT(wb.has_named_range("test_nr"));
        TS_ASSERT_THROWS(wb2.create_named_range("test_nr", new_sheet, "A1"), std::runtime_error);
    }
    
    void test_get_named_range()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet();
        wb.create_named_range("test_nr", new_sheet, "A1");
        auto found_range = wb.named_range("test_nr");
        auto expected_range = new_sheet.range("A1");
        TS_ASSERT_EQUALS(expected_range, found_range);
        TS_ASSERT_THROWS(wb.named_range("test_nr2"), std::runtime_error);
    }

    void test_remove_named_range()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet();
        wb.create_named_range("test_nr", new_sheet, "A1");
        wb.remove_named_range("test_nr");
        TS_ASSERT(!new_sheet.has_named_range("test_nr"));
        TS_ASSERT(!wb.has_named_range("test_nr"));
        TS_ASSERT_THROWS(wb.remove_named_range("test_nr2"), std::runtime_error);
    }

    void test_post_increment_iterator()
    {
        xlnt::workbook wb;

        wb.create_sheet().title("Sheet2");
        wb.create_sheet().title("Sheet3");

        auto iter = wb.begin();

        TS_ASSERT_EQUALS((*(iter++)).title(), "Sheet1");
        TS_ASSERT_EQUALS((*(iter++)).title(), "Sheet2");
        TS_ASSERT_EQUALS((*(iter++)).title(), "Sheet3");
        TS_ASSERT_EQUALS(iter, wb.end());

        const auto wb_const = wb;

        auto const_iter = wb_const.begin();

        TS_ASSERT_EQUALS((*(const_iter++)).title(), "Sheet1");
        TS_ASSERT_EQUALS((*(const_iter++)).title(), "Sheet2");
        TS_ASSERT_EQUALS((*(const_iter++)).title(), "Sheet3");
        TS_ASSERT_EQUALS(const_iter, wb_const.end());
    }

    void test_copy_iterator()
    {
        xlnt::workbook wb;

        wb.create_sheet().title("Sheet2");
        wb.create_sheet().title("Sheet3");

        auto iter = wb.begin();
		TS_ASSERT_EQUALS((*iter).title(), "Sheet1");

        iter++;
        TS_ASSERT_EQUALS((*iter).title(), "Sheet2");

        auto copy = wb.begin();
        copy = iter;
        TS_ASSERT_EQUALS((*iter).title(), "Sheet2");
        TS_ASSERT_EQUALS(iter, copy);

        iter++;
        TS_ASSERT_EQUALS((*iter).title(), "Sheet3");
        TS_ASSERT_DIFFERS(iter, copy);

        copy++;
        TS_ASSERT_EQUALS((*iter).title(), "Sheet3");
        TS_ASSERT_EQUALS(iter, copy);
    }
    
    void test_manifest()
    {
		xlnt::manifest m;
        TS_ASSERT(!m.has_default_type("xml"));
        TS_ASSERT_THROWS(m.default_type("xml"), xlnt::key_not_found);
        TS_ASSERT(!m.has_relationship(xlnt::path("/"), xlnt::relationship_type::office_document));
        TS_ASSERT(m.relationships(xlnt::path("xl/workbook.xml")).empty());
    }

    void test_memory()
    {
        xlnt::workbook wb, wb2;
        wb.active_sheet().title("swap");
        std::swap(wb, wb2);
        TS_ASSERT_EQUALS(wb.active_sheet().title(), "Sheet1");
        TS_ASSERT_EQUALS(wb2.active_sheet().title(), "swap");
        wb = wb2;
        TS_ASSERT_EQUALS(wb.active_sheet().title(), "swap");
    }

    void test_clear()
    {
        xlnt::workbook wb;
        xlnt::style s = wb.create_style("s");
        wb.active_sheet().cell("B2").value("B2");
        wb.active_sheet().cell("B2").style(s);
        TS_ASSERT(wb.active_sheet().cell("B2").has_style());
        wb.clear_styles();
        TS_ASSERT(!wb.active_sheet().cell("B2").has_style());
		xlnt::format format = wb.create_format();
        xlnt::font font;
        font.size(41);
        format.font(font, true);
        wb.active_sheet().cell("B2").format(format);
        TS_ASSERT(wb.active_sheet().cell("B2").has_format());
        wb.clear_formats();
        TS_ASSERT(!wb.active_sheet().cell("B2").has_format());
        wb.clear();
        TS_ASSERT(wb.sheet_titles().empty());
    }

    void test_comparison()
    {
        xlnt::workbook wb, wb2;
        TS_ASSERT(wb == wb);
        TS_ASSERT(!(wb != wb));
        TS_ASSERT(!(wb == wb2));
        TS_ASSERT(wb != wb2)
        
        const auto &wb_const = wb;
        //TODO these aren't tests...
        wb_const.manifest();
        
        TS_ASSERT(wb.has_theme());
        
        wb.create_style("style1");
        wb.style("style1");
        wb_const.style("style1");
    }
};