#pragma once

#include <gtest/gtest.h>
#include <gmock/gmock-matchers.h>
#include "header.h"
#include "../../DesktopEditor/common/File.h"
#include "streamrw.h"

using namespace testing;
using namespace std;
using namespace CFCPP;


struct HeaderTest : testing::Test
{
    string filename;
    Stream stream;
    Header hd;

    HeaderTest() :
        filename("../../../data/ex.ppt"),
        stream(OpenFileStream(filename, false))
    {
    }
};

void test_header_state(const Header& hd)
{
    ASSERT_THAT(hd.headerSignature, ElementsAre(0xD0, 0xCF, 0x11, 0xE0, 0xA1, 0xB1, 0x1A, 0xE1));
    EXPECT_EQ(hd.minorVersion, 0x003E);
    EXPECT_EQ(hd.majorVersion, 0x0003);
    EXPECT_EQ(hd.byteOrder, 0xFFFE);
    EXPECT_EQ(hd.sectorShift, 0x0009);
    EXPECT_EQ(hd.miniSectorShift, 0x0006);
    EXPECT_EQ(hd.directorySectorsNumber, 0);
    EXPECT_EQ(hd.fatSectorsNumber, 1);
    EXPECT_EQ(hd.firstDirectorySectorID, 1);
    EXPECT_EQ(hd.minSizeStandardStream, 0x1000);
    EXPECT_EQ(hd.firstMiniFATSectorID, 2);
    EXPECT_EQ(hd.miniFATSectorsNumber, 1);
    EXPECT_EQ(hd.firstDIFATSectorID, -2);
    EXPECT_EQ(hd.difatSectorsNumber, 0);

    int difat[109];
    memset(reinterpret_cast<char*>(difat), 0xFF, sizeof(difat));
    difat[0] = 0;
    ASSERT_FALSE(memcmp(hd.difat, difat, sizeof(difat)));
}

TEST_F(HeaderTest, test_header_open)
{
    EXPECT_TRUE(IsOpen(stream));
}

TEST_F(HeaderTest, test_header_read)
{
    hd.Read(stream);
    test_header_state(hd);
}

TEST_F(HeaderTest, test_header_write)
{
    hd.Read(stream);

    std::string other_filename("../../../data/header.bin");
    stream = OpenFileStream(other_filename, true);
    hd.Write(stream);

    Header other;
    stream->seek(0, std::ios::beg);
    other.Read(stream);
    test_header_state(other);
    remove(other_filename.c_str());
}

TEST_F(HeaderTest, test_header_seek)
{
    hd.Read(stream);

    std::string other_filename("../../../data/sheader.bin");
    remove(other_filename.c_str());
    stream = OpenFileStream(other_filename, true);

    std::vector<char> zeroArray(512, 0);
    stream->write(zeroArray.data(), zeroArray.size());
    zeroArray.clear();

    EXPECT_EQ(Length(stream), 512);


    stream->seek(4);
    StreamRW rw(stream);
    auto pos = rw.Tell();
    EXPECT_EQ(pos, 4);

    hd.Write(stream);
    EXPECT_EQ(Length(stream), 512+4);


    Header other;
    stream->seek(0);
    int zeroValue = StreamRW(stream).Read<int>();
    EXPECT_EQ(zeroValue, 0);


    other.Read(stream);
    test_header_state(other);


}