#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_get_active_sheet()
    {
        xlnt::workbook wb;
        TS_ASSERT_EQUALS(wb.get_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.get_cell("A6").set_value(1.498);
        wb.copy_sheet(new_sheet);
        TS_ASSERT(wb[1].compare(wb[2], false));
        wb.create_sheet().get_cell("A6").set_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.get_active_sheet();
        TS_ASSERT_THROWS(wb2.copy_sheet(new_sheet), xlnt::invalid_parameter);
        TS_ASSERT_THROWS(wb2.get_index(new_sheet), std::runtime_error);
    }

    void test_add_sheet_at_index()
    {
        xlnt::workbook wb;
        auto ws = wb.get_active_sheet();
        ws.get_cell("B3").set_value(2);
        ws.set_title("Active");
        wb.copy_sheet(ws, 0);
        TS_ASSERT_EQUALS(wb.get_sheet_titles().at(0), "Sheet1");
        TS_ASSERT_EQUALS(wb.get_sheet_by_index(0).get_cell("B3").get_value<int>(), 2);
        TS_ASSERT_EQUALS(wb.get_sheet_titles().at(1), "Active");
        TS_ASSERT_EQUALS(wb.get_sheet_by_index(1).get_cell("B3").get_value<int>(), 2);
    }
    
    void test_remove_sheet()
    {
        xlnt::workbook wb, wb2;
        auto new_sheet = wb.create_sheet(0);
		new_sheet.set_title("removed");
        wb.remove_sheet(new_sheet);
        TS_ASSERT(!wb.contains("removed"));
        TS_ASSERT_THROWS(wb.remove_sheet(wb2.get_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.set_title(title);
        auto found_sheet = wb.get_sheet_by_title(title);
        TS_ASSERT_EQUALS(new_sheet, found_sheet);
        TS_ASSERT_THROWS(wb.get_sheet_by_title("error"), xlnt::key_not_found);
        const auto &wb_const = wb;
        TS_ASSERT_THROWS(wb_const.get_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.set_title(title);
        const xlnt::workbook& wbconst = wb;
        auto found_sheet = wbconst.get_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.get_title(), "Sheet1");
        }
    }
    
    void test_get_index()
    {
        xlnt::workbook wb;
        wb.create_sheet().set_title("1");
        wb.create_sheet().set_title("2");

        auto sheet_index = wb.get_index(wb.get_sheet_by_title("1"));
        TS_ASSERT_EQUALS(sheet_index, 1);

        sheet_index = wb.get_index(wb.get_sheet_by_title("2"));
        TS_ASSERT_EQUALS(sheet_index, 2);
    }

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

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

		TS_ASSERT_EQUALS(wb.get_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.get_named_range("test_nr");
        auto expected_range = new_sheet.get_range("A1");
        TS_ASSERT_EQUALS(expected_range, found_range);
        TS_ASSERT_THROWS(wb.get_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().set_title("Sheet2");
        wb.create_sheet().set_title("Sheet3");

        auto iter = wb.begin();

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

        const auto wb_const = wb;

        auto const_iter = wb_const.begin();

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

    void test_copy_iterator()
    {
        xlnt::workbook wb;

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

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

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

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

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

        copy++;
        TS_ASSERT_EQUALS((*iter).get_title(), "Sheet3");
        TS_ASSERT_EQUALS(iter, copy);
    }
    
    void test_manifest()
    {
		xlnt::manifest m;
        TS_ASSERT(!m.has_default_type("xml"));
        TS_ASSERT_THROWS(m.get_default_type("xml"), xlnt::key_not_found);
        TS_ASSERT(!m.has_relationship(xlnt::path("/"), xlnt::relationship::type::office_document));
        TS_ASSERT(m.get_relationships(xlnt::path("xl/workbook.xml")).empty());
    }

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

    void test_clear()
    {
        xlnt::workbook wb;
        xlnt::style s = wb.create_style("s");
        wb.get_active_sheet().get_cell("B2").set_value("B2");
        wb.get_active_sheet().get_cell("B2").set_style(s);
        TS_ASSERT(wb.get_active_sheet().get_cell("B2").has_style());
        wb.clear_styles();
        TS_ASSERT(!wb.get_active_sheet().get_cell("B2").has_style());
		xlnt::format format = wb.create_format();
        xlnt::font font;
        font.size(41);
        format.font(font, true);
        wb.get_active_sheet().get_cell("B2").set_format(format);
        TS_ASSERT(wb.get_active_sheet().get_cell("B2").has_format());
        wb.clear_formats();
        TS_ASSERT(!wb.get_active_sheet().get_cell("B2").has_format());
        wb.clear();
        TS_ASSERT(wb.get_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.get_manifest();
        
        TS_ASSERT(wb.has_theme());
        
        wb.create_style("style1");
        wb.get_style("style1");
        wb_const.get_style("style1");
    }
};