xlnt/source/tests/ZipFileTestSuite.h

#pragma once

#include <cxxtest/TestSuite.h>

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

class ZipFileTestSuite : public CxxTest::TestSuite
{
public:
    ZipFileTestSuite()
    {
        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::zip_file package(new_xlsx, xlnt::file_mode::create, xlnt::file_access::read_write);

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

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

    void test_read_text()
    {
        xlnt::zip_file package(test_zip, xlnt::file_mode::open, xlnt::file_access::read_write);
        auto contents = package.get_file_contents("a.txt");
        TS_ASSERT_EQUALS(contents, "a.txt");
    }

    void test_write_text()
    {
        {
            xlnt::zip_file package(test_zip, xlnt::file_mode::open, xlnt::file_access::read_write);
            package.set_file_contents("a.txt", "something else");
        }

        {
            xlnt::zip_file package(test_zip, xlnt::file_mode::open, xlnt::file_access::read_write);
            auto contents = package.get_file_contents("a.txt");
            TS_ASSERT_EQUALS(contents, "something else");
        }
    }

    void test_read_xml()
    {
        xlnt::zip_file package(test_zip, xlnt::file_mode::open, xlnt::file_access::read_write);
        auto contents = package.get_file_contents("a.txt");

        pugi::xml_document doc;
        doc.load(contents.c_str());

        auto root_element = 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;
};