package jp.ac.u_ryukyu.ie.cr.tatsuki.jungle.index;

import java.nio.ByteBuffer;

import org.junit.Test;

import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.DefaultJungle;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTree;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.NodePath;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.DefaultNodePath;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.DefaultTreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.TreeNode;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.transaction.IndexJungleTreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.traverser.DefaultTraverser;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Either;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Error;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Pair;
import junit.framework.Assert;
import fj.data.List;
import fj.data.TreeMap;

public class AttributeIndexTest {

	@Test
	public void PutAttributeIndexTest(){
		DefaultJungle jungle = new DefaultJungle(null,"hoge",new DefaultTreeEditor(new DefaultTraverser()));
		JungleTree tree = jungle.createNewTree("fuga");
		IndexJungleTreeEditor editor = tree.getIndexTreeEditor();
		TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> emptyIndex = editor.getIndex();
		Assert.assertTrue(emptyIndex.isEmpty());
		NodePath path = new DefaultNodePath();
		Either<Error, JungleTreeEditor> either = editor.addNewChildAt(path, 0);
		Assert.assertFalse(either.isA());
		
		JungleTreeEditor editor2 = either.b();
		Either<Error, JungleTreeEditor> either2 = editor2.putAttribute(path.add(0),"key", ByteBuffer.wrap("test".toString().getBytes()));
		Assert.assertFalse(either2.isA());
		JungleTreeEditor editor3 = either2.b();
		editor3.success();
		
		IndexJungleTreeEditor IndexEditor = tree.getIndexTreeEditor();
		TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> index = IndexEditor.getIndex();
		
		Assert.assertEquals(index.get("key").some().get("test").some().head().left().getAttributes().getString("key"),"test");
		
		JungleTreeEditor editor4 = tree.getIndexTreeEditor();
		Either<Error, JungleTreeEditor> either3 = editor4.deleteAttribute(path.add(0), "key");
		Assert.assertFalse(either3.isA());
		JungleTreeEditor editor5 = either3.b();
		editor5.success();
		
		IndexJungleTreeEditor IndexEditor2 = tree.getIndexTreeEditor();
		TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> deleteIndex = IndexEditor2.getIndex();		
		Assert.assertTrue(deleteIndex.isEmpty());
	}
}