Fix TraceSampleApplicationTests reading log format (#2377)

spring-attic · May 19, 2020 · 36d7c43 · 36d7c43
1 parent 514fcff
commit 36d7c43
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.../spring-cloud-gcp-trace-sample/src/test/java/com/example/TraceSampleApplicationTests.java b/.../spring-cloud-gcp-trace-sample/src/test/java/com/example/TraceSampleApplicationTests.java
@@ -27,7 +27,7 @@
 import com.google.cloud.logging.LogEntry;
 import com.google.cloud.logging.Logging;
 import com.google.cloud.logging.LoggingOptions;
-import com.google.cloud.logging.Payload.StringPayload;
+import com.google.cloud.logging.Payload.JsonPayload;
 import com.google.devtools.cloudtrace.v1.GetTraceRequest;
 import com.google.devtools.cloudtrace.v1.Trace;
 import com.google.devtools.cloudtrace.v1.TraceServiceGrpc;
@@ -155,7 +155,8 @@ public void testTracesAreLoggedCorrectly() {
 
 
 			List<String> logContents = logEntries.stream()
-					.map((logEntry) -> ((StringPayload) logEntry.getPayload()).getData())
+					.map((logEntry) -> (String) ((JsonPayload) logEntry.getPayload())
+							.getDataAsMap().get("message"))
 					.collect(Collectors.toList());
 
 			assertThat(logContents).contains("starting busy work");