临时工 - PowerMock系列4

import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
public class ReaderDemo {
public static String read(String path) throws IOException {
String result = "";
InputStream in = new FileInputStream(path);
InputStreamReader isr = new InputStreamReader(in);
BufferedReader br = new BufferedReader(isr);
for (String line = br.readLine(); line != null; line = br.readLine()) {
result+=line;
}
isr.close();
return result;
}
}

import static org.junit.Assert.assertEquals;
import static org.powermock.api.mockito.PowerMockito.doReturn;
import static org.powermock.api.mockito.PowerMockito.mock;
import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.InputStreamReader;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
@RunWith(PowerMockRunner.class)
@PrepareForTest({ ReaderDemo.class })
public class ReaderDemoTest {
    @Test
    public void doSomethingReturnsValueFromBufferedReader() throws Exception {
        InputStreamReader inputStreamReaderMock = mock(InputStreamReader.class);
        BufferedReader bufferedReaderMock = mock(BufferedReader.class);
        FileInputStream inputStreamMock= mock(FileInputStream.class);
        PowerMockito.whenNew(BufferedReader.class).
                    withArguments(inputStreamReaderMock).
                    thenReturn(bufferedReaderMock);
        PowerMockito.whenNew(FileInputStream.class).
        withAnyArguments().
        thenReturn(inputStreamMock);

      PowerMockito.whenNew(InputStreamReader.class).
                    withAnyArguments().thenReturn(inputStreamReaderMock);
        final String string = "key=value\n";
      doReturn(string).doReturn(null).when(bufferedReaderMock).readLine();
        // read && Assert
        assertEquals(string, ReaderDemo.read("anything"));
    }
}