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.query.traverser.InterfaceTraverser;
import jp.ac.u_ryukyu.ie.cr.jungle.store.nodepath.DefaultNodePath;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.editor.jungleTreeEditor.JungleTreeEditor;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.node.TreeNode;
import jp.ac.u_ryukyu.ie.cr.jungle.tree.JungleTree;
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");
        jungle.createNewTree("tree");
        JungleTree tree = jungle.getTreeByName("tree");
        JungleTreeEditor editor = tree.getJungleTreeEditor();
        DefaultNodePath path = new DefaultNodePath();
        editor = editor.addNewChildAt(path, 0).b();
        path = path.add(0);
        editor = editor.putAttribute(path, "KEY", ByteBuffer.wrap("data".getBytes())).b();
        editor = editor.addNewChildAt(path, 0).b();
        path = path.add(0);
        editor = editor.putAttribute(path, "KEY", ByteBuffer.wrap("kanagawa".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("kanagawa"))
                return true;
            return false;
        }, "KEY", "kanagawa");

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