Search in sources :

Example 6 with RemoteOnly

use of com.google.cloud.runtime.jetty.test.annotation.RemoteOnly in project jetty-runtime by GoogleCloudPlatform.

the class ForwardedIntegrationTest method testRemoteProto.

/**
   * Validate GFE proto header is handled.
   *
   * @throws Exception test in error
   */
@Test
@RemoteOnly
public void testRemoteProto() throws Exception {
    URI target = getUri().resolve("/dump/info");
    target = new URI(target.toASCIIString().replace("http:", "https:"));
    assertThat(target.getPath(), containsString("/dump/info"));
    // Check that GFE X-Forwarded-Proto is not ignored
    HttpURLConnection http = HttpUrlUtil.openTo(target);
    assertThat(http.getResponseCode(), is(200));
    String responseBody = HttpUrlUtil.getResponseBody(http);
    assertThat(responseBody, containsString("scheme=https(secure=true)"));
}
Also used : HttpURLConnection(java.net.HttpURLConnection) CoreMatchers.containsString(org.hamcrest.CoreMatchers.containsString) URI(java.net.URI) RemoteOnly(com.google.cloud.runtime.jetty.test.annotation.RemoteOnly) AbstractIntegrationTest(com.google.cloud.runtime.jetty.test.AbstractIntegrationTest) Test(org.junit.Test)

Example 7 with RemoteOnly

use of com.google.cloud.runtime.jetty.test.annotation.RemoteOnly in project jetty-runtime by GoogleCloudPlatform.

the class LoggingIntegrationTest method testLogging.

@Test
@RemoteOnly
public void testLogging() throws Exception {
    // Create unique ID to relate request with log entries
    String id = Long.toHexString(System.nanoTime());
    // Hit the DumpServlet that will log the request path
    URI target = getUri().resolve("/dump/info/" + id);
    HttpURLConnection http = HttpUrlUtil.openTo(target);
    assertThat(http.getResponseCode(), is(200));
    String responseBody = HttpUrlUtil.getResponseBody(http);
    List<String> lines = new BufferedReader(new StringReader(responseBody)).lines().collect(Collectors.toList());
    assertThat(lines.stream().filter(s -> s.startsWith("requestURI=")).findFirst().get(), containsString(id));
    String traceId = lines.stream().filter(s -> s.startsWith("X-Cloud-Trace-Context: ")).findFirst().get().split("[ /]")[1];
    assertThat(traceId, Matchers.notNullValue());
    LoggingOptions options = LoggingOptions.getDefaultInstance();
    String filter = "resource.type=gae_app AND resource.labels.module_id=smoke" + " AND textPayload:" + id;
    int expected = 2;
    try (Logging logging = options.getService()) {
        Page<LogEntry> entries = logging.listLogEntries(EntryListOption.filter(filter));
        for (LogEntry entry : entries.iterateAll()) {
            if (entry.getSeverity() == Severity.INFO) {
                assertThat(entry.getLogName(), is("gae_app.log"));
                assertThat(entry.getResource().getType(), is("gae_app"));
                assertThat(entry.getResource().getLabels().get("module_id"), is("smoke"));
                assertThat(entry.getResource().getLabels().get("zone"), Matchers.notNullValue());
                assertThat(entry.getLabels().get("appengine.googleapis.com/trace_id"), is(traceId));
                assertThat(entry.getLabels().get("appengine.googleapis.com/instance_name"), Matchers.notNullValue());
                if (entry.getPayload().toString().contains("JUL.info:/dump/info/")) {
                    expected--;
                    assertThat(entry.getPayload().toString(), Matchers.containsString(id));
                }
                if (entry.getPayload().toString().contains("ServletContext.log:/dump/info")) {
                    expected--;
                    assertThat(entry.getPayload().toString(), Matchers.containsString(id));
                }
            }
        }
        assertThat(expected, is(0));
    }
}
Also used : CoreMatchers.is(org.hamcrest.CoreMatchers.is) HttpURLConnection(java.net.HttpURLConnection) Page(com.google.api.gax.paging.Page) CoreMatchers.containsString(org.hamcrest.CoreMatchers.containsString) LogEntry(com.google.cloud.logging.LogEntry) HttpUrlUtil(com.google.cloud.runtime.jetty.util.HttpUrlUtil) Matchers(org.hamcrest.Matchers) Test(org.junit.Test) Severity(com.google.cloud.logging.Severity) Collectors(java.util.stream.Collectors) List(java.util.List) StringReader(java.io.StringReader) LoggingOptions(com.google.cloud.logging.LoggingOptions) EntryListOption(com.google.cloud.logging.Logging.EntryListOption) AbstractIntegrationTest(com.google.cloud.runtime.jetty.test.AbstractIntegrationTest) MatcherAssert.assertThat(org.hamcrest.MatcherAssert.assertThat) Logging(com.google.cloud.logging.Logging) BufferedReader(java.io.BufferedReader) URI(java.net.URI) Assert(org.junit.Assert) RemoteOnly(com.google.cloud.runtime.jetty.test.annotation.RemoteOnly) Logging(com.google.cloud.logging.Logging) HttpURLConnection(java.net.HttpURLConnection) BufferedReader(java.io.BufferedReader) StringReader(java.io.StringReader) CoreMatchers.containsString(org.hamcrest.CoreMatchers.containsString) LoggingOptions(com.google.cloud.logging.LoggingOptions) URI(java.net.URI) LogEntry(com.google.cloud.logging.LogEntry) RemoteOnly(com.google.cloud.runtime.jetty.test.annotation.RemoteOnly) Test(org.junit.Test) AbstractIntegrationTest(com.google.cloud.runtime.jetty.test.AbstractIntegrationTest)

Aggregations

RemoteOnly (com.google.cloud.runtime.jetty.test.annotation.RemoteOnly)7 AbstractIntegrationTest (com.google.cloud.runtime.jetty.test.AbstractIntegrationTest)6 HttpURLConnection (java.net.HttpURLConnection)6 URI (java.net.URI)6 Test (org.junit.Test)6 CoreMatchers.containsString (org.hamcrest.CoreMatchers.containsString)5 Page (com.google.api.gax.paging.Page)1 LogEntry (com.google.cloud.logging.LogEntry)1 Logging (com.google.cloud.logging.Logging)1 EntryListOption (com.google.cloud.logging.Logging.EntryListOption)1 LoggingOptions (com.google.cloud.logging.LoggingOptions)1 Severity (com.google.cloud.logging.Severity)1 HttpUrlUtil (com.google.cloud.runtime.jetty.util.HttpUrlUtil)1 BufferedReader (java.io.BufferedReader)1 StringReader (java.io.StringReader)1 List (java.util.List)1 Collectors (java.util.stream.Collectors)1 CoreMatchers.is (org.hamcrest.CoreMatchers.is)1 MatcherAssert.assertThat (org.hamcrest.MatcherAssert.assertThat)1 Matchers (org.hamcrest.Matchers)1