项目:digdag
文件:DigdagClientTest.java
@Test
public void getLogFileFailsAfter10Attempts()
throws Exception
{
QueueDispatcher dispatcher = new QueueDispatcher();
dispatcher.setFailFast(new MockResponse().setResponseCode(500));
mockWebServer.setDispatcher(dispatcher);
try {
client.getLogFile(Id.of("17"),RestLogFileHandle.builder()
.agentId("test-agent")
.fileName("test-task-1.log")
.fileSize(4711)
.fileTime(Instant.now().truncatedTo(SECONDS))
.taskName("test-task-1")
.build());
fail();
}
catch (InternalServerErrorException ignore) {
}
assertThat(mockWebServer.getRequestCount(),is(10));
}
项目:mockwebserverplus
文件:MockWebServerPlusTest.java
@Test public void enqueueSocketPolicy() throws IOException {
server.enqueue(SocketPolicy.KEEP_OPEN);
QueueDispatcher dispatcher = new QueueDispatcher();
server.setDispatcher(dispatcher);
MockResponse mockResponse = dispatcher.peek();
assertThat(mockResponse.getSocketPolicy()).isEqualTo(SocketPolicy.KEEP_OPEN);
}
项目:mockwebserverplus
文件:MockWebServerPlusTest.java
@Test public void interceptDispatch() throws InterruptedException,IOException {
Dispatcher dispatcher = spy(new QueueDispatcher());
server.setDispatcher(dispatcher);
server.enqueue(new MockResponse());
execute();
verify(dispatcher).dispatch(any(RecordedRequest.class));
}
项目:digdag
文件:SlaIT.java
@Test
public void verifyAlertIsRetried()
throws Exception
{
mockWebServer.setDispatcher(new QueueDispatcher());
mockWebServer.enqueue(new MockResponse().setResponseCode(500).setBody("FAIL"));
mockWebServer.enqueue(new MockResponse().setResponseCode(200));
Id attemptId = pushAndStart("duration_alert_enabled.dig",Duration.ofSeconds(5));
RecordedRequest recordedRequest1 = mockWebServer.takeRequest(30,TimeUnit.SECONDS);
RecordedRequest recordedRequest2 = mockWebServer.takeRequest(30,TimeUnit.SECONDS);
verifyNotification(attemptId,recordedRequest1);
verifyNotification(attemptId,recordedRequest2);
}
项目:digdag
文件:HttpIT.java
@Test
public void testForEach()
throws Exception
{
String uri = "http://localhost:" + httpMockWebServer.getPort() + "/";
httpMockWebServer.setDispatcher(new QueueDispatcher());
String content = DigdagClient.objectMapper().writeValueAsString(
ImmutableList.of("foo","bar","baz"));
httpMockWebServer.enqueue(new MockResponse().setBody(content));
runWorkflow(folder,"acceptance/http/http_for_each.dig",ImmutableMap.of("test_uri",uri));
assertThat(httpMockWebServer.getRequestCount(),is(1));
}
项目:mockstar
文件:BaseLogicTest.java
protected MockWebServer getErrorMockWebServer() {
mockWebServer.setDispatcher(new QueueDispatcher());
return mockWebServer;
}
项目:datacollector
文件:WebServerBuilder.java
public WebServerBuilder() {
server = new MockWebServer();
QueueDispatcher dispatcher = new QueueDispatcher();
dispatcher.setFailFast(new MockResponse().setResponseCode(400));
server.setDispatcher(dispatcher);
}