package jp.ac.u_ryukyu.ie.cr.jungle.core;

import jp.ac.u_ryukyu.ie.cr.jungle.DefaultJungle;
import jp.ac.u_ryukyu.ie.cr.jungle.Jungle;
import jp.ac.u_ryukyu.ie.cr.jungle.JungleTree;
import jp.ac.u_ryukyu.ie.cr.jungle.JungleTreeEditor;
import jp.ac.u_ryukyu.ie.cr.jungle.store.impl.DefaultNodePath;
import jp.ac.u_ryukyu.ie.cr.jungle.store.impl.DefaultTreeEditor;
import jp.ac.u_ryukyu.ie.cr.jungle.store.impl.TreeNode;
import jp.ac.u_ryukyu.ie.cr.jungle.traverser.DefaultTraverser;
import jp.ac.u_ryukyu.ie.cr.jungle.traverser.InterfaceTraverser;
import junit.framework.Assert;
import org.junit.Test;

import java.nio.ByteBuffer;
import java.util.Iterator;

public class GetNodeOfPathTest {

    @Test
    public void getNodeOfPathTest() {
        Jungle jungle = new DefaultJungle(null, "hogehoge", new DefaultTreeEditor(new DefaultTraverser()));
        jungle.createNewTree("tree");
        JungleTree tree = jungle.getTreeByName("tree");
        JungleTreeEditor editor = tree.getTreeEditor();
        DefaultNodePath path = new DefaultNodePath();
        editor = editor.addNewChildAt(path, 0).b();
        path = path.add(0);
        editor = editor.putAttribute(path, "KEY", ByteBuffer.wrap("tatsuki".getBytes())).b();
        editor = editor.addNewChildAt(path, 0).b();
        path = path.add(0);
        editor = editor.putAttribute(path, "KEY", ByteBuffer.wrap("tatsuki".getBytes())).b();
        editor.success();
        InterfaceTraverser traverser = tree.getTraverser(true);
        Iterator<TreeNode> nodeIterator = traverser.find((TreeNode node) -> {
            String str = node.getAttributes().getString("KEY");
            if (str == null)
                return false;
            if (str.equals("tatsuki"))
                return true;
            return false;
        }, "KEY", "tatsuki");

        TreeNode node = tree.getNodeOfPath(path).b();
        Assert.assertTrue(nodeIterator.hasNext());
        Assert.assertEquals(node, nodeIterator.next());
    }
}