我需要测试以下方法而不改变方法本身。
该方法向服务器发出POST方法。但我需要创建一个独立于服务器的测试用例。
我在将它重定向到本地文件之前测试了一个类似的方法。
但那是因为我将协议作为文件，主机名作为localhost，端口作为-1。
我的问题是这个方法做了一个帖子并转换为HttpURLConnection和wr = new DataOutputStream（conn.getOutputStream（））;不会通过http在本地txt文件上工作。
//构造函数
public HTTPConnector(String usr, String pwd, String protocol,
String hostname, int port) {
this.usr = usr;
this.pwd = pwd;
this.protocol = protocol;
this.hostname = hostname;
this.port = port;
// connect();
}
//我需要测试的方法
public String doPost(String reference, String data) throws IOException {
URL url = null;
HttpURLConnection conn = null;
BufferedReader rd = null;
DataOutputStream wr = null;
InputStream is = null;
String line = null;
StringBuffer response = null;
url = new URL(protocol, hostname, port, reference);
conn = (HttpURLConnection) url.openConnection();
conn.setRequestMethod("POST");
conn.setRequestProperty("Authorization", "Basic dGVzdDphc2Rm");
conn.setRequestProperty("Content-Type", "application/xml");
conn.setUseCaches(false);
conn.setDoInput(true);
conn.setDoOutput(true);
// Send response
wr = new DataOutputStream(conn.getOutputStream());
wr.writeBytes(data);
wr.flush();
wr.close();
// Get response
is = conn.getInputStream();
rd = new BufferedReader(new InputStreamReader(is));
response = new StringBuffer();
while ((line = rd.readLine()) != null) {
response.append(line);
response.append('\r');
}
rd.close();
return response.toString();
}
//我可以测试的方法
public String doGet(String reference) throws IOException {
connect();
URL url = new URL(protocol, hostname, port, reference);
InputStream content = (InputStream) url.getContent();
BufferedReader xml = new BufferedReader(new InputStreamReader(content));
return xml.readLine();
}
这是一个示例测试。请注意，我所做的断言仅用于演示目的，您需要适应您的需求。
@RunWith(PowerMockRunner.class)
@PrepareForTest({ toTest.class, URL.class, HttpURLConnection.class })
public class soTest {
/**
* test response.
*/
private static final String TEST_RESPONSE = "test\nresponse";
/**
* test data.
*/
private static final String DATA = RandomStringUtils.randomAscii(125);
/**
* test port.
*/
private static final int PORT = 8080;
/**
* test hosts.
*/
private static final String HOSTNAME = "hostname";
/**
* test protocol.
*/
private static final String PROTOCOL = "http";
/**
* test reference.
*/
private static final String REFERENCE = "REFERENCE";
/**
* URL mock.
*/
private URL url;
/**
* HttpURLConnection mock.
*/
private HttpURLConnection connection;
/**
* Our output.
*/
private ByteArrayOutputStream output;
/**
* Our input.
*/
private ByteArrayInputStream input;
/**
* Instance under tests.
*/
private toTest instance;
@Before
public void setUp() throws Exception
{
this.url = PowerMockito.mock(URL.class);
this.connection = PowerMockito.mock(HttpURLConnection.class);
this.output = new ByteArrayOutputStream();
this.input = new ByteArrayInputStream(TEST_RESPONSE.getBytes());
this.instance = new toTest(PROTOCOL, HOSTNAME, PORT);
PowerMockito.whenNew(URL.class).withArguments(PROTOCOL, HOSTNAME, PORT, REFERENCE).thenReturn(this.url);
}
@Test
public void testDoPost() throws Exception
{
PowerMockito.doReturn(this.connection).when(this.url).openConnection();
PowerMockito.doReturn(this.output).when(this.connection).getOutputStream();
PowerMockito.doReturn(this.input).when(this.connection).getInputStream();
final String response = this.instance.doPost(REFERENCE, DATA);
PowerMockito.verifyNew(URL.class);
new URL(PROTOCOL, HOSTNAME, PORT, REFERENCE);
// Mockito.verify(this.url).openConnection(); // cannot be verified (mockito limitation)
Mockito.verify(this.connection).getOutputStream();
Mockito.verify(this.connection).setRequestMethod("POST");
Mockito.verify(this.connection).setRequestProperty("Authorization", "Basic dGVzdDphc2Rm");
Mockito.verify(this.connection).setRequestProperty("Content-Type", "application/xml");
Mockito.verify(this.connection).setUseCaches(false);
Mockito.verify(this.connection).setDoInput(true);
Mockito.verify(this.connection).setDoOutput(true);
Mockito.verify(this.connection).getInputStream();
assertArrayEquals(DATA.getBytes(), this.output.toByteArray());
assertEquals(TEST_RESPONSE.replaceAll("\n", "\r") + "\r", response);
}
}
@Data
public class toTest {
private final String protocol, hostname;
private final int port;
public String doPost(String reference, String data) throws IOException
{
// your method, not modified
}
}
依赖关系：
- commons-lang 2.5
- powermock-api-mockito 1.4.11
- powermock-module-junit4 1.4.11
- junit 4.10
- lombok 0.11.0在测试类
如果你没有机会重构被测试的方法，那么一种新颖的方法就是嘲笑 HttpUrlConnection 它使用。第一手这似乎很难，因为 HttpUrlConnection 不作为参数传入。但是，您可以通过控制从中返回的连接来控制此操作 url.openConnection。
这由注册的协议处理程序控制 java.net.URL 对于传递给构造函数的协议。诀窍是注册一个新的协议处理程序（参见 Java - 注册自定义URL协议处理程序 详情）。
您的新协议处理程序应该返回一个模拟 HttpUrlConnection 然后，您可以在测试中使用。