01: /*
02: * Copyright 2001-2004 The Apache Software Foundation
03: *
04: * Licensed under the Apache License, Version 2.0 (the "License");
05: * you may not use this file except in compliance with the License.
06: * You may obtain a copy of the License at
07: *
08: * http://www.apache.org/licenses/LICENSE-2.0
09: *
10: * Unless required by applicable law or agreed to in writing, software
11: * distributed under the License is distributed on an "AS IS" BASIS,
12: * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13: * See the License for the specific language governing permissions and
14: * limitations under the License.
15: */
16: package org.apache.commons.collections;
17:
18: import junit.framework.Test;
19: import junit.framework.TestSuite;
20:
21: import org.apache.commons.collections.bag.AbstractTestBag;
22:
23: /**
24: * Extension of {@link TestBag} for exercising the {@link TreeBag}
25: * implementation.
26: *
27: * @version $Revision: 155406 $ $Date: 2005-02-26 12:55:26 +0000 (Sat, 26 Feb 2005) $
28: *
29: * @author Chuck Burdick
30: */
31: public class TestTreeBag extends AbstractTestBag {
32:
33: public TestTreeBag(String testName) {
34: super (testName);
35: }
36:
37: public static Test suite() {
38: return new TestSuite(TestTreeBag.class);
39: }
40:
41: public static void main(String args[]) {
42: String[] testCaseName = { TestTreeBag.class.getName() };
43: junit.textui.TestRunner.main(testCaseName);
44: }
45:
46: public Bag makeBag() {
47: return new TreeBag();
48: }
49:
50: public SortedBag setupBag() {
51: SortedBag bag = (SortedBag) makeBag();
52: bag.add("C");
53: bag.add("A");
54: bag.add("B");
55: bag.add("D");
56: return bag;
57: }
58:
59: public void testOrdering() {
60: Bag bag = setupBag();
61: assertEquals("Should get elements in correct order", "A", bag
62: .toArray()[0]);
63: assertEquals("Should get elements in correct order", "B", bag
64: .toArray()[1]);
65: assertEquals("Should get elements in correct order", "C", bag
66: .toArray()[2]);
67: assertEquals("Should get first key", "A", ((SortedBag) bag)
68: .first());
69: assertEquals("Should get last key", "D", ((SortedBag) bag)
70: .last());
71: }
72: }
|