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.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 junit.framework.Assert;
import fj.data.List;
import fj.data.Option;
import fj.data.TreeMap;

public class AttributeIndexTest {

	@Test
	public void PutAttributeIndexTest(){
		DefaultJungle jungle = new DefaultJungle(null,"hoge",new DefaultTraverser());
		JungleTree tree = jungle.createNewTree("fuga");
		IndexJungleTreeEditor editor = tree.getIndexTreeEditor();
		TreeMap<String, TreeMap<String, List<TreeNode>>> 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 = editor3.success().b();
	
		editor3.putAttribute(path.add(0),"key", ByteBuffer.wrap("tatsuki".toString().getBytes())).b().success().b();	
		TreeNode targetNode = tree.getRootNode().getChildren().at(0).b();
		IndexJungleTreeEditor IndexEditor = tree.getIndexTreeEditor();
		TreeMap<String, TreeMap<String, List<TreeNode>>> index = IndexEditor.getIndex();
		List<TreeNode> NodeList = index.get("key").some().get("tatsuki").some();
		String attributeIndex = NodeList.head().getAttributes().getString("key");
		Assert.assertEquals(attributeIndex,"tatsuki");
		Assert.assertEquals(NodeList.head(),targetNode);
		
		JungleTreeEditor editor4 = tree.getIndexTreeEditor();
		Either<Error, JungleTreeEditor> either3 = editor4.deleteAttribute(path.add(0), "key").b().success();
		Assert.assertFalse(either3.isA());
		JungleTreeEditor editor5 = either3.b();
		editor5.success();
		
		IndexJungleTreeEditor IndexEditor2 = tree.getIndexTreeEditor();
		TreeMap<String, TreeMap<String, List<TreeNode>>> deleteIndexList = IndexEditor2.getIndex();		
		Option<TreeMap<String, List<TreeNode>>> deleteIndexOp = deleteIndexList.get("key");
		TreeMap<String, List<TreeNode>> deleteIndex = deleteIndexOp.some();
		Assert.assertTrue(deleteIndex.get("test").some().isEmpty());
		Assert.assertTrue(deleteIndex.get("tatsuki").some().isEmpty());
	}
}