1 package org.apache.maven.plugin.surefire.booterclient.lazytestprovider;
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 import org.apache.maven.surefire.api.booter.Command;
23 import org.apache.maven.surefire.api.booter.Shutdown;
24
25 import java.io.IOException;
26 import java.util.Queue;
27 import java.util.concurrent.ConcurrentLinkedQueue;
28 import java.util.concurrent.Semaphore;
29 import java.util.concurrent.atomic.AtomicBoolean;
30
31 import static org.apache.maven.surefire.api.booter.Command.BYE_ACK;
32 import static org.apache.maven.surefire.api.booter.Command.NOOP;
33 import static org.apache.maven.surefire.api.booter.Command.SKIP_SINCE_NEXT_TEST;
34 import static org.apache.maven.surefire.api.booter.Command.TEST_SET_FINISHED;
35 import static org.apache.maven.surefire.api.booter.Command.toRunClass;
36 import static org.apache.maven.surefire.api.booter.Command.toShutdown;
37
38
39
40
41
42
43
44
45
46
47
48
49 public final class TestProvidingInputStream
50 extends DefaultCommandReader
51 {
52 private final Semaphore barrier = new Semaphore( 0 );
53
54 private final Queue<Command> commands = new ConcurrentLinkedQueue<>();
55
56 private final AtomicBoolean closed = new AtomicBoolean();
57
58 private final Queue<String> testClassNames;
59
60
61
62
63
64
65 public TestProvidingInputStream( Queue<String> testClassNames )
66 {
67 this.testClassNames = testClassNames;
68 }
69
70
71
72
73 void testSetFinished()
74 {
75 if ( canContinue() )
76 {
77 commands.add( TEST_SET_FINISHED );
78 barrier.release();
79 }
80 }
81
82 @Override
83 public void skipSinceNextTest()
84 {
85 if ( canContinue() )
86 {
87 commands.add( SKIP_SINCE_NEXT_TEST );
88 barrier.release();
89 }
90 }
91
92 @Override
93 public void shutdown( Shutdown shutdownType )
94 {
95 if ( canContinue() )
96 {
97 commands.add( toShutdown( shutdownType ) );
98 barrier.release();
99 }
100 }
101
102 @Override
103 public void noop()
104 {
105 if ( canContinue() )
106 {
107 commands.add( NOOP );
108 barrier.release();
109 }
110 }
111
112 @Override
113 public void acknowledgeByeEventReceived()
114 {
115 if ( canContinue() )
116 {
117 commands.add( BYE_ACK );
118 barrier.release();
119 }
120 }
121
122 @Override
123 protected Command nextCommand()
124 {
125 Command cmd = commands.poll();
126 if ( cmd == null )
127 {
128 String cmdData = testClassNames.poll();
129 return cmdData == null ? TEST_SET_FINISHED : toRunClass( cmdData );
130 }
131 else
132 {
133 return cmd;
134 }
135 }
136
137 @Override
138 protected void beforeNextCommand()
139 throws IOException
140 {
141 awaitNextTest();
142 }
143
144 @Override
145 public boolean isClosed()
146 {
147 return closed.get();
148 }
149
150
151
152
153 @Override
154 public void provideNewTest()
155 {
156 if ( canContinue() )
157 {
158 barrier.release();
159 }
160 }
161
162 @Override
163 public void close()
164 {
165 if ( closed.compareAndSet( false, true ) )
166 {
167 barrier.drainPermits();
168 barrier.release();
169 }
170 }
171
172 private void awaitNextTest()
173 throws IOException
174 {
175 try
176 {
177 barrier.acquire();
178 }
179 catch ( InterruptedException e )
180 {
181 throw new IOException( e.getLocalizedMessage(), e );
182 }
183 }
184 }