#include "BaseTestCase.h"

#include "System.h"
#include "DefinitionNode.h"
#include "IntNode.h"
#include "FloatNode.h"
#include "PackStream.h"
#include "DefinitionWatcher.h"

TEST(DefinitionNode, toString) {
    DefinitionNode root(NULL, "root");
    DefinitionNode branch(&root, "branch");
    DefinitionNode leaf1(&branch, "leaf1");
    DefinitionNode leaf2(&branch, "leaf2");

    EXPECT_EQ("root\n branch\n  leaf1\n  leaf2", root.toString());
    EXPECT_EQ("branch\n leaf1\n leaf2", branch.toString());
}

TEST(DefinitionNode, getPath) {
    DefinitionNode root(NULL, "root");
    DefinitionNode branch(&root, "branch");
    DefinitionNode leaf(&branch, "leaf");

    EXPECT_EQ("/", root.getPath());
    EXPECT_EQ("/branch", branch.getPath());
    EXPECT_EQ("/branch/leaf", leaf.getPath());
}

TEST(DefinitionNode, findByPath) {
    DefinitionNode root(NULL, "root");
    DefinitionNode branch(&root, "branch");
    DefinitionNode leaf(&branch, "leaf");

    EXPECT_EQ(&root, root.findByPath("/"));
    EXPECT_EQ(&branch, root.findByPath("/branch"));
    EXPECT_EQ(NULL, root.findByPath("/branche"));
    EXPECT_EQ(&leaf, root.findByPath("/branch/leaf"));
    EXPECT_EQ(NULL, root.findByPath("/branche/leaf"));
    EXPECT_EQ(NULL, root.findByPath("/branch/leave"));

    EXPECT_EQ(&branch, root.findByPath("branch"));
    EXPECT_EQ(&leaf, root.findByPath("branch/leaf"));
    EXPECT_EQ(&leaf, branch.findByPath("leaf"));
}

TEST(DefinitionNode, attachDetach) {
    DefinitionNode *root = new DefinitionNode(NULL, "root");
    DefinitionNode *child1 = new DefinitionNode(root, "child1");
    DefinitionNode *child2 = new DefinitionNode(root, "child2");

    EXPECT_EQ(root, child1->getParent());
    EXPECT_EQ(root, child2->getParent());
    EXPECT_EQ(2u, root->getChildrenCount());

    delete child1;

    EXPECT_EQ(1u, root->getChildrenCount());

    delete child2;

    EXPECT_EQ(0u, root->getChildrenCount());

    delete root;
}

TEST(DefinitionNode, saveLoadCopy) {
    PackStream *stream;
    int check_in = 42, check_out = 0;

    DefinitionNode *before = new DefinitionNode(NULL, "root");
    DefinitionNode *before1 = new DefinitionNode(before, "before1");
    new FloatNode(before1, "before11", 2.1);
    new FloatNode(before1, "before12", -4.3);
    DefinitionNode *before2 = new DefinitionNode(before, "before2");
    new DefinitionNode(before2, "before21");
    new FloatNode(before2, "before22");
    new FloatNode(before, "before3", 6.7);

    stream = new PackStream();
    stream->bindToFile(TMP_DIRECTORY "test_paysages_pack", true);
    before->save(stream);
    stream->write(&check_in);
    delete stream;

    // Same definition tree, but with missing nodes, and added ones
    DefinitionNode *after = new DefinitionNode(NULL, "root");
    DefinitionNode *after1 = new DefinitionNode(after, "before1");
    FloatNode *after12 = new FloatNode(after1, "before12");
    FloatNode *after13 = new FloatNode(after1, "before13");
    FloatNode *after3 = new FloatNode(after, "before3");
    FloatNode *after4 = new FloatNode(after, "before4");

    stream = new PackStream();
    stream->bindToFile(TMP_DIRECTORY "test_paysages_pack");
    after->load(stream);
    stream->read(&check_out);
    delete stream;

    EXPECT_DOUBLE_EQ(-4.3, after12->getValue());
    EXPECT_DOUBLE_EQ(0.0, after13->getValue());
    EXPECT_DOUBLE_EQ(6.7, after3->getValue());
    EXPECT_DOUBLE_EQ(0.0, after4->getValue());
    EXPECT_EQ(42, check_out);

    // Test copy, that should work like "save then load"
    after12->setValue(1.1);
    after13->setValue(1.1);
    after3->setValue(1.1);
    after4->setValue(1.1);
    before->copy(after);

    EXPECT_DOUBLE_EQ(-4.3, after12->getValue());
    EXPECT_DOUBLE_EQ(1.1, after13->getValue());
    EXPECT_DOUBLE_EQ(6.7, after3->getValue());
    EXPECT_DOUBLE_EQ(1.1, after4->getValue());

    delete before;
    delete after;
}

TEST(DefinitionNode, onChildChange) {
    class CapturingNode : public DefinitionNode {
      public:
        CapturingNode(DefinitionNode *parent, const string &name) : DefinitionNode(parent, name) {
        }
        virtual void onChildChanged(int depth, const string &relpath) override {
            changes.push_back(pair<int, string>(depth, relpath));
            DefinitionNode::onChildChanged(depth, relpath);
        }
        vector<pair<int, string>> changes;
    };

    CapturingNode root(NULL, "root");
    CapturingNode branch(&root, "branch");
    IntNode leaf(&branch, "leaf");

    leaf.setValue(5.0);

    ASSERT_EQ(1u, root.changes.size());
    EXPECT_EQ(1, root.changes[0].first);
    EXPECT_EQ("branch/leaf", root.changes[0].second);

    ASSERT_EQ(1u, branch.changes.size());
    EXPECT_EQ(0, branch.changes[0].first);
    EXPECT_EQ("leaf", branch.changes[0].second);
}