1 package com.atlassian.user.search.page;
2
3 import junit.framework.TestCase;
4
5 import java.util.List;
6 import java.util.ArrayList;
7 import java.util.Iterator;
8
9 import com.atlassian.user.EntityException;
10
11 public class TestMergedPager extends TestCase
12 {
13 public void setUp() throws Exception
14 {
15 super.setUp();
16 }
17
18 public void tearDown() throws Exception
19 {
20 super.tearDown();
21 }
22
23 public void testMergePagers() throws EntityException
24 {
25 List<String> one = new ArrayList<String>();
26 List<String> two = new ArrayList<String>();
27
28 one.add("1");
29 two.add("2");
30
31 Pager<String> firstPager = new DefaultPager<String>(one);
32 Pager<String> secondPager = new DefaultPager<String>(two);
33
34 MergedPager merged = new MergedPager(firstPager, secondPager);
35
36
37 Iterator iter = merged.iterator();
38 List<String> results = new ArrayList<String>();
39
40 while (iter.hasNext())
41 {
42 String s = (String) iter.next();
43 results.add(s);
44 }
45
46 assertEquals(2, results.size());
47 assertTrue(results.contains("1"));
48 assertTrue(results.contains("2"));
49 }
50
51 public void testMergePagerWithNullPager() throws EntityException
52 {
53 List<String> one = new ArrayList<String>();
54 List two = null;
55
56 one.add("1");
57
58 Pager<String> firstPager = new DefaultPager<String>(one);
59
60 MergedPager merged = new MergedPager(firstPager, null);
61
62 Iterator iter = merged.iterator();
63 List<String> results = new ArrayList<String>();
64
65 while (iter.hasNext())
66 {
67 String s = (String) iter.next();
68 results.add(s);
69 }
70
71 assertEquals(1, results.size());
72 assertTrue(results.contains("1"));
73 }
74 }