1 package org.apache.maven.surefire.util;
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 import java.util.ArrayList;
23 import java.util.Arrays;
24 import java.util.Collections;
25 import java.util.HashSet;
26 import java.util.Iterator;
27 import java.util.List;
28 import java.util.Set;
29 import org.apache.maven.surefire.testset.TestSetFailedException;
30
31
32
33
34
35
36
37 public class TestsToRun
38 {
39 private final List locatedClasses;
40
41
42
43
44
45
46 public TestsToRun( List locatedClasses )
47 {
48 this.locatedClasses = Collections.unmodifiableList( locatedClasses );
49 Set testSets = new HashSet();
50
51 for ( Iterator iterator = locatedClasses.iterator(); iterator.hasNext(); )
52 {
53 Class testClass = (Class) iterator.next();
54 if ( testSets.contains( testClass ) )
55 {
56 throw new RuntimeException( "Duplicate test set '" + testClass.getName() + "'" );
57 }
58 testSets.add( testClass );
59 }
60 }
61
62 public static TestsToRun fromClass( Class clazz )
63 throws TestSetFailedException
64 {
65 return new TestsToRun( Arrays.asList( new Class[]{ clazz } ) );
66 }
67
68
69
70
71
72
73 public Iterator iterator()
74 {
75 return locatedClasses.iterator();
76 }
77
78 public String toString()
79 {
80 StringBuffer sb = new StringBuffer();
81 sb.append( "TestsToRun: [" );
82 Iterator it = iterator();
83 while ( it.hasNext() )
84 {
85 Class clazz = (Class) it.next();
86 sb.append( " " ).append( clazz.getName() );
87 }
88
89 sb.append( ']' );
90 return sb.toString();
91 }
92
93 public boolean containsAtLeast( int atLeast )
94 {
95 return containsAtLeast( iterator(), atLeast );
96 }
97
98 private boolean containsAtLeast( Iterator it, int atLeast )
99 {
100 for ( int i = 0; i < atLeast; i++ )
101 {
102 if ( !it.hasNext() )
103 {
104 return false;
105 }
106
107 it.next();
108 }
109
110 return true;
111 }
112
113 public boolean containsExactly( int items )
114 {
115 Iterator it = iterator();
116 return containsAtLeast( it, items ) && !it.hasNext();
117 }
118
119
120
121
122
123 public boolean allowEagerReading()
124 {
125 return true;
126 }
127
128 public Class[] getLocatedClasses()
129 {
130 if ( !allowEagerReading() )
131 {
132 throw new IllegalStateException( "Cannot eagerly read" );
133 }
134 List result = new ArrayList();
135 Iterator it = iterator();
136 while ( it.hasNext() )
137 {
138 result.add( it.next() );
139 }
140 return (Class[]) result.toArray( new Class[result.size()] );
141 }
142 }