| package org.apache.maven.plugin.surefire.extensions; |
| |
| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| |
| import org.apache.maven.plugin.surefire.log.api.ConsoleLogger; |
| import org.apache.maven.surefire.api.booter.Command; |
| import org.apache.maven.surefire.extensions.CommandReader; |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.mockito.invocation.InvocationOnMock; |
| import org.mockito.stubbing.Answer; |
| |
| import java.io.ByteArrayOutputStream; |
| import java.io.IOException; |
| import java.nio.ByteBuffer; |
| import java.nio.channels.WritableByteChannel; |
| import java.util.Iterator; |
| |
| import static java.util.Arrays.asList; |
| import static org.apache.maven.surefire.api.booter.Command.TEST_SET_FINISHED; |
| import static org.apache.maven.surefire.api.booter.MasterProcessCommand.NOOP; |
| import static org.apache.maven.surefire.api.booter.MasterProcessCommand.RUN_CLASS; |
| import static org.fest.assertions.Assertions.assertThat; |
| import static org.mockito.ArgumentMatchers.any; |
| import static org.mockito.Mockito.mock; |
| import static org.mockito.Mockito.times; |
| import static org.mockito.Mockito.verify; |
| import static org.mockito.Mockito.verifyZeroInteractions; |
| import static org.mockito.Mockito.when; |
| |
| /** |
| * Tests for {@link StreamFeeder}. |
| */ |
| public class StreamFeederTest |
| { |
| private final ByteArrayOutputStream out = new ByteArrayOutputStream(); |
| private final WritableByteChannel channel = mock( WritableByteChannel.class ); |
| private final CommandReader commandReader = mock( CommandReader.class ); |
| private StreamFeeder streamFeeder; |
| |
| @Before |
| public void setup() throws IOException |
| { |
| final Iterator<Command> it = asList( new Command( RUN_CLASS, "pkg.ATest" ), TEST_SET_FINISHED ).iterator(); |
| when( commandReader.readNextCommand() ) |
| .thenAnswer( new Answer<Command>() |
| { |
| @Override |
| public Command answer( InvocationOnMock invocation ) |
| { |
| return it.hasNext() ? it.next() : null; |
| } |
| } ); |
| } |
| |
| @After |
| public void close() throws IOException |
| { |
| if ( streamFeeder != null ) |
| { |
| streamFeeder.disable(); |
| streamFeeder.close(); |
| } |
| } |
| |
| @Test |
| public void shouldEncodeCommandToStream() throws Exception |
| { |
| when( channel.write( any( ByteBuffer.class ) ) ) |
| .thenAnswer( new Answer<Object>() |
| { |
| @Override |
| public Object answer( InvocationOnMock invocation ) throws IOException |
| { |
| ByteBuffer bb = invocation.getArgument( 0 ); |
| bb.flip(); |
| out.write( bb.array() ); |
| return 0; |
| } |
| } ); |
| |
| ConsoleLogger logger = mock( ConsoleLogger.class ); |
| streamFeeder = new StreamFeeder( "t", channel, commandReader, logger ); |
| streamFeeder.start(); |
| |
| streamFeeder.join(); |
| String commands = out.toString(); |
| |
| assertThat( commands ) |
| .isEqualTo( ":maven-surefire-command:run-testclass:pkg.ATest::maven-surefire-command:testset-finished:" ); |
| |
| verify( channel, times( 1 ) ) |
| .close(); |
| |
| assertThat( streamFeeder.getException() ) |
| .isNull(); |
| |
| verifyZeroInteractions( logger ); |
| } |
| |
| @Test |
| public void shouldFailThread() throws Exception |
| { |
| when( channel.write( any( ByteBuffer.class ) ) ) |
| .thenAnswer( new Answer<Object>() |
| { |
| @Override |
| public Object answer( InvocationOnMock invocation ) throws IOException |
| { |
| throw new IOException(); |
| } |
| } ); |
| |
| ConsoleLogger logger = mock( ConsoleLogger.class ); |
| streamFeeder = new StreamFeeder( "t", channel, commandReader, logger ); |
| streamFeeder.start(); |
| |
| streamFeeder.join(); |
| |
| assertThat( out.size() ) |
| .isZero(); |
| |
| verify( channel, times( 1 ) ) |
| .close(); |
| |
| assertThat( streamFeeder.getException() ) |
| .isNotNull() |
| .isInstanceOf( IOException.class ); |
| |
| verifyZeroInteractions( logger ); |
| } |
| |
| @Test( expected = IllegalArgumentException.class ) |
| public void shouldFailWithoutData() |
| { |
| StreamFeeder.encode( RUN_CLASS ); |
| } |
| |
| @Test( expected = IllegalArgumentException.class ) |
| public void shouldFailWithData() |
| { |
| StreamFeeder.encode( NOOP, "" ); |
| } |
| } |