1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21 package org.apache.directory.mavibot.btree;
22
23
24 import static org.junit.Assert.assertEquals;
25
26 import java.io.IOException;
27 import java.util.ArrayList;
28 import java.util.List;
29
30 import org.apache.directory.mavibot.btree.exception.KeyNotFoundException;
31 import org.apache.directory.mavibot.btree.serializer.IntSerializer;
32 import org.junit.Test;
33
34
35
36
37
38
39
40 public class InMemoryBTreeBuilderTest
41 {
42 @Test
43 public void testIntegerTree() throws IOException, KeyNotFoundException
44 {
45 List<Tuple<Integer, Integer>> sortedTuple = new ArrayList<Tuple<Integer, Integer>>();
46
47 for ( int i = 1; i < 8; i++ )
48 {
49 Tuple<Integer, Integer> t = new Tuple<Integer, Integer>( i, i );
50 sortedTuple.add( t );
51 }
52
53 IntSerializer ser = IntSerializer.INSTANCE;
54 InMemoryBTreeBuilder<Integer, Integer> bb = new InMemoryBTreeBuilder<Integer, Integer>( "master", 4, ser, ser );
55
56
57 BTree<Integer, Integer> btree = bb.build( sortedTuple.iterator() );
58
59 assertEquals( 1, btree.getRootPage().getNbElems() );
60
61 assertEquals( 7, btree.getRootPage().findRightMost().getKey().intValue() );
62
63 assertEquals( 1, btree.getRootPage().findLeftMost().getKey().intValue() );
64
65 TupleCursor<Integer, Integer> cursor = btree.browse();
66 int i = 0;
67
68 while ( cursor.hasNext() )
69 {
70 Tuple<Integer, Integer> expected = sortedTuple.get( i++ );
71 Tuple<Integer, Integer> actual = cursor.next();
72 assertEquals( expected.getKey(), actual.getKey() );
73 assertEquals( expected.getValue(), actual.getValue() );
74 }
75
76 cursor.close();
77 btree.close();
78 }
79
80 }