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.memory;
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.Tuple;
31 import org.apache.directory.mavibot.btree.TupleCursor;
32 import org.apache.directory.mavibot.btree.serializer.IntSerializer;
33 import org.junit.Test;
34
35
36
37
38
39
40
41 public class BTreeBuilderTest
42 {
43 @Test
44 public void testIntegerTree() throws IOException
45 {
46 List<Tuple<Integer, Integer>> sortedTuple = new ArrayList<Tuple<Integer, Integer>>();
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 = new IntSerializer();
54 BTreeBuilder<Integer, Integer> bb = new BTreeBuilder<Integer, Integer>( "master", 4, ser, ser );
55
56
57 BTree<Integer, Integer> btree = bb.build( sortedTuple.iterator() );
58
59 assertEquals( 1, btree.rootPage.getNbElems() );
60
61 assertEquals( 7, btree.rootPage.findRightMost().getKey().intValue() );
62
63 assertEquals( 1, btree.rootPage.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 }