1 package org.apache.maven.surefire.booter;
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 import java.util.Collections;
23 import java.util.Iterator;
24
25 import org.apache.maven.surefire.api.booter.MasterProcessChannelEncoder;
26 import org.apache.maven.surefire.api.provider.SurefireProvider;
27 import org.apache.maven.surefire.api.util.CloseableIterator;
28 import org.apache.maven.surefire.api.util.TestsToRun;
29
30 import static org.apache.maven.surefire.api.util.ReflectionUtils.loadClass;
31
32
33
34
35
36
37
38
39
40
41
42
43
44 final class LazyTestsToRun
45 extends TestsToRun
46 {
47 private final MasterProcessChannelEncoder eventChannel;
48 private final CommandReader commandReader;
49
50
51
52
53
54
55 LazyTestsToRun( MasterProcessChannelEncoder eventChannel, CommandReader commandReader )
56 {
57 super( Collections.<Class<?>>emptySet() );
58 this.eventChannel = eventChannel;
59 this.commandReader = commandReader;
60 }
61
62 private final class BlockingIterator
63 implements Iterator<Class<?>>
64 {
65 private final Iterator<String> it = commandReader.getIterableClasses( eventChannel ).iterator();
66
67 @Override
68 public boolean hasNext()
69 {
70 return it.hasNext();
71 }
72
73 @Override
74 public Class<?> next()
75 {
76 return findClass( it.next() );
77 }
78
79 @Override
80 public void remove()
81 {
82 throw new UnsupportedOperationException();
83 }
84 }
85
86
87
88
89 @Override
90 public Iterator<Class<?>> iterated()
91 {
92 return newWeakIterator();
93 }
94
95
96
97
98
99
100 @Override
101 public Iterator<Class<?>> iterator()
102 {
103 return new BlockingIterator();
104 }
105
106
107
108
109
110 @Override
111 public String toString()
112 {
113 return "LazyTestsToRun";
114 }
115
116
117
118
119
120 @Override
121 public boolean allowEagerReading()
122 {
123 return false;
124 }
125
126 private static Class<?> findClass( String clazz )
127 {
128 return loadClass( Thread.currentThread().getContextClassLoader(), clazz );
129 }
130
131
132
133
134
135 private Iterator<Class<?>> newWeakIterator()
136 {
137 final Iterator<String> it = commandReader.iterated();
138 return new CloseableIterator<Class<?>>()
139 {
140 @Override
141 protected boolean isClosed()
142 {
143 return LazyTestsToRun.this.isFinished();
144 }
145
146 @Override
147 protected boolean doHasNext()
148 {
149 return it.hasNext();
150 }
151
152 @Override
153 protected Class<?> doNext()
154 {
155 return findClass( it.next() );
156 }
157
158 @Override
159 protected void doRemove()
160 {
161 }
162
163 @Override
164 public void remove()
165 {
166 throw new UnsupportedOperationException( "unsupported remove" );
167 }
168 };
169 }
170 }