#pragma once

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

#include "../xlnt.h"

class WorkbookTestSuite : public CxxTest::TestSuite
{
public:
    WorkbookTestSuite()
    {

    }

    void test_get_active_sheet()
    {
        xlnt::workbook wb;
        auto active_sheet = wb.get_active_sheet();
        TS_ASSERT_EQUALS(active_sheet, wb[0]);
    }

    void test_create_sheet()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet(0);
        TS_ASSERT_EQUALS(new_sheet, wb[0]);
    }

    void test_create_sheet_with_name()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet(0, "LikeThisName");
        TS_ASSERT_EQUALS(new_sheet, wb[0]);
    }

    void test_add_correct_sheet()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet(0);
        wb.add_sheet(new_sheet);
        TS_ASSERT_EQUALS(new_sheet, wb[2]);
    }

    void test_create_sheet_readonly()
    {
        xlnt::workbook wb(xlnt::optimized::read);
        TS_ASSERT_THROWS_ANYTHING(wb.create_sheet());
    }

    void test_remove_sheet()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet(0);
        wb.remove_sheet(new_sheet);
        for(auto worksheet : wb)
        {
            TS_ASSERT_DIFFERS(new_sheet, worksheet);
        }
    }

    void test_get_sheet_by_name()
    {
        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_name(title);
        TS_ASSERT_EQUALS(new_sheet, found_sheet);
    }
    
    void test_getitem()
    {
        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_name(title);
        TS_ASSERT_EQUALS(new_sheet, found_sheet);
        
        TS_ASSERT_THROWS_ANYTHING(wb["NotThere"]);
    }

    void test_get_index2()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet(0);
        auto sheet_index = wb.get_index(new_sheet);
        TS_ASSERT_EQUALS(sheet_index, 0);
    }

    void test_get_sheet_names()
    {
        xlnt::workbook wb;
        std::vector<std::string> names = {"Sheet", "Sheet1", "Sheet2", "Sheet3", "Sheet4", "Sheet5"};
        for(int count = 0; count < names.size(); count++)
        {
            wb.create_sheet(names[count]);
        }
        auto actual_names = wb.get_sheet_names();
        std::sort(actual_names.begin(), actual_names.end());
        std::sort(names.begin(), names.end());
        for(int count = 0; count < names.size(); count++)
        {
            TS_ASSERT_EQUALS(actual_names[count], names[count]);
        }
    }

    void test_get_active_sheet2()
    {
        xlnt::workbook wb;
        auto active_sheet = wb.get_active_sheet();
        TS_ASSERT_EQUALS(active_sheet, wb[0]);
    }

    void test_create_sheet2()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet(0);
        TS_ASSERT_EQUALS(new_sheet, wb[0]);
    }

    void test_create_sheet_with_name2()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet(0, "LikeThisName");
        TS_ASSERT_EQUALS(new_sheet, wb[0]);
    }

    void test_add_correct_sheet2()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet(0);
        wb.add_sheet(new_sheet);
        TS_ASSERT_EQUALS(new_sheet, wb[2]);
    }

    void test_create_sheet_readonly2()
    {
        xlnt::workbook wb(xlnt::optimized::read);
        TS_ASSERT_THROWS_ANYTHING(wb.create_sheet());
    }

    void test_remove_sheet2()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet(0);
        wb.remove_sheet(new_sheet);
        for(auto worksheet : wb)
        {
            TS_ASSERT_DIFFERS(new_sheet, worksheet);
        }
    }

    void test_get_sheet_by_name2()
    {
        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_name(title);
        TS_ASSERT_EQUALS(new_sheet, found_sheet);
    }

    void test_get_index()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet(0);
        auto sheet_index = wb.get_index(new_sheet);
        TS_ASSERT_EQUALS(sheet_index, 0);
    }

    void test_add_named_range()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet();
        xlnt::named_range named_range(new_sheet, "A1");
        wb.add_named_range("test_nr", named_range);
        auto named_ranges_list = wb.get_named_ranges();
        TS_ASSERT_DIFFERS(std::find(named_ranges_list.begin(), named_ranges_list.end(), named_range), named_ranges_list.end());
    }

    void test_get_named_range2()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet();
        xlnt::named_range named_range(new_sheet, "A1");
        wb.add_named_range("test_nr", named_range);
        auto found_named_range = wb.get_named_range("test_nr", new_sheet);
        TS_ASSERT_EQUALS(named_range, found_named_range);
    }

    void test_remove_named_range()
    {
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet();
        xlnt::named_range named_range(new_sheet, "A1");
        wb.add_named_range("test_nr", named_range);
        wb.remove_named_range(named_range);
        auto named_ranges_list = wb.get_named_ranges();
        TS_ASSERT_EQUALS(std::find(named_ranges_list.begin(), named_ranges_list.end(), named_range), named_ranges_list.end());
    }

    void test_add_local_named_range()
    {
        make_tmpdir();
        xlnt::workbook wb;
        auto new_sheet = wb.create_sheet();
        xlnt::named_range named_range(new_sheet, "A1");
        named_range.set_scope(new_sheet);
        wb.add_named_range("test_nr", named_range);
        auto dest_filename = PathHelper::GetTempDirectory() + "/local_named_range_book.xlsx";
        wb.save(dest_filename);
        clean_tmpdir();
    }
    
    void make_tmpdir()
    {
        
    }
    
    void clean_tmpdir()
    {
        
    }

    void test_write_regular_date()
    {
        /*make_tmpdir();
        auto today = datetime.datetime(2010, 1, 18, 14, 15, 20, 1600);

        xlnt::workbook wb;
        auto sheet = book.get_active_sheet();
        sheet.cell("A1") = today;
        dest_filename = osp.join(TMPDIR, "date_read_write_issue.xlsx");
        book.save(dest_filename);

        test_book = load_workbook(dest_filename);
        test_sheet = test_book.get_active_sheet();

        TS_ASSERT_EQUALS(test_sheet.cell("A1"), today);
        clean_tmpdir();*/
    }

    void test_write_regular_float()
    {
        make_tmpdir();
        float float_value = 1.0 / 3.0;
        xlnt::workbook book;
        auto sheet = book.get_active_sheet();
        sheet.cell("A1") = float_value;
        auto dest_filename = PathHelper::GetTempDirectory() + "float_read_write_issue.xlsx";
        book.save(dest_filename);

        xlnt::workbook test_book;
        test_book.load(dest_filename);
        auto test_sheet = test_book.get_active_sheet();

        TS_ASSERT_EQUALS(test_sheet.cell("A1"), float_value);
        clean_tmpdir();
    }
};