#pragma once

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

#include <helpers/path_helper.hpp>
#include <xlnt/workbook/workbook.hpp>

// Cryptographic key generation can take a few seconds, particularly in unoptomized builds.
// Set this to false to skip those tests that use cryptography.
#define TEST_CRYPTO true

#ifndef TEST_CRYPTO
#define TEST_CRYPTO false
#endif

class test_consume_xlsx : public CxxTest::TestSuite
{
public:
    void test_decrypt_agile()
    {
        xlnt::workbook wb;
#if TEST_CRYPTO
        wb.load(path_helper::get_data_directory("14_encrypted_excel_2016.xlsx"), "secret");
#endif
    }

    void test_decrypt_libre_office()
    {
        xlnt::workbook wb;
#if TEST_CRYPTO
        wb.load(path_helper::get_data_directory("15_encrypted_libre_office.xlsx"), "secret");
#endif
    }

    void test_decrypt_standard()
    {
        xlnt::workbook wb;
#if TEST_CRYPTO
        wb.load(path_helper::get_data_directory("16_encrypted_excel_2007.xlsx"), "password");
#endif
    }

    void test_decrypt_numbers()
    {
        xlnt::workbook wb;
#if TEST_CRYPTO
        wb.load(path_helper::get_data_directory("17_encrypted_numbers.xlsx"), "secret");
#endif
    }

    void test_comments()
    {
        xlnt::workbook wb;
        wb.load("data/18_basic_comments.xlsx");

        auto sheet1 = wb[0];
        TS_ASSERT_EQUALS(sheet1.get_cell("A1").get_value<std::string>(), "Sheet1!A1");
        TS_ASSERT_EQUALS(sheet1.get_cell("A1").comment().plain_text(), "Sheet1 comment");
        TS_ASSERT_EQUALS(sheet1.get_cell("A1").comment().author(), "Microsoft Office User");

        auto sheet2 = wb[1];
        TS_ASSERT_EQUALS(sheet2.get_cell("A1").get_value<std::string>(), "Sheet2!A1");
        TS_ASSERT_EQUALS(sheet2.get_cell("A1").comment().plain_text(), "Sheet2 comment");
        TS_ASSERT_EQUALS(sheet2.get_cell("A1").comment().author(), "Microsoft Office User");
    }

    void test_read_unicode_filename()
    {
#ifdef _MSC_VER
        xlnt::workbook wb;
        wb.load(L"data\\19_unicode_Λ.xlsx");
        TS_ASSERT_EQUALS(wb.get_active_sheet().get_cell("A1").get_value<std::string>(), "unicode!");
#endif
#ifndef __MINGW32__
        xlnt::workbook wb2;
        wb2.load(u8"data/19_unicode_Λ.xlsx");
        TS_ASSERT_EQUALS(wb2.get_active_sheet().get_cell("A1").get_value<std::string>(), "unicode!");
#endif
    }
};