#pragma once

#include <cxxtest/TestSuite.h>

#include "../xlnt.h"
#include "pugixml.hpp"

class PackageTestSuite : public CxxTest::TestSuite
{
public:
    PackageTestSuite()
    {
        template_zip = "../../source/tests/test_data/packaging/test.zip";
        test_zip = "../../source/tests/test_data/packaging/a.zip";
        existing_xlsx = "../../source/tests/test_data/packaging/existing.xlsx";
        new_xlsx = "../../source/tests/test_data/packaging/new.xlsx";

        xlnt::file::copy(template_zip, test_zip, true);
    }

    void test_existing_package()
    {
        //xlnt::package package;
        //package.open(existing_xlsx, xlnt::file_mode::Open, xlnt::file_access::Read);
    }

    void test_new_package()
    {
        xlnt::package package;
        package.open(new_xlsx, xlnt::file_mode::Create, xlnt::file_access::ReadWrite);

        //auto part_1 = package.create_part("workbook.xml", "type");
        //TS_ASSERT_DIFFERS(part_1, nullptr);

        //part_1.write("test");
    }

    void test_read_text()
    {
        xlnt::package package;
        package.open(test_zip, xlnt::file_mode::Open, xlnt::file_access::ReadWrite);

        auto part_1 = package.get_part("a.txt");
        TS_ASSERT_DIFFERS(part_1, nullptr);

        auto part_1_data = part_1.read();
        TS_ASSERT_EQUALS(part_1_data, "a.txt");
    }

    void test_write_text()
    {
        {
            xlnt::package package;
            package.open(test_zip, xlnt::file_mode::Open, xlnt::file_access::ReadWrite);

            auto part_1 = package.get_part("a.txt");
            TS_ASSERT_DIFFERS(part_1, nullptr);

            part_1.write("something else");
        }

        {
            xlnt::package package;
            package.open(test_zip, xlnt::file_mode::Open, xlnt::file_access::ReadWrite);

            auto part_1 = package.get_part("a.txt");
            TS_ASSERT_DIFFERS(part_1, nullptr);

            auto part_1_data = part_1.read();

            TS_ASSERT_EQUALS(part_1_data, "something else");
        }
    }

    void test_read_xml()
    {
        xlnt::package package;
        package.open(test_zip, xlnt::file_mode::Open, xlnt::file_access::ReadWrite);

        auto part_2 = package.get_part("a.xml");
        TS_ASSERT_DIFFERS(part_2, nullptr);

        auto part_2_data = part_2.read();
        TS_ASSERT_DIFFERS(part_2_data, "");

        pugi::xml_document part_2_doc;
        part_2_doc.load(part_2_data.c_str());

        auto root_element = part_2_doc.child("root");
        TS_ASSERT_DIFFERS(root_element, nullptr)

        auto child_element = root_element.child("child");
        TS_ASSERT_DIFFERS(child_element, nullptr)

        TS_ASSERT_EQUALS(std::string(child_element.attribute("attribute").as_string()), "attribute")

        auto element_element = root_element.child("element");
        TS_ASSERT_DIFFERS(element_element, nullptr)

        TS_ASSERT_EQUALS(std::string(element_element.text().as_string()), "Text")
    }

private:
    std::string template_zip;
    std::string test_zip;
    std::string existing_xlsx;
    std::string new_xlsx;
};