starting to write test, debug test server

Author: harsimar

smoke-tests/apps/LiveMetrics/build.gradle.kts

@@ -4,4 +4,5 @@ plugins {
 
 dependencies {
   smokeTestImplementation("org.awaitility:awaitility:4.2.0")
+  implementation("log4j:log4j:1.2.17")
 }

smoke-tests/apps/LiveMetrics/src/main/java/com/microsoft/applicationinsights/smoketestapp/TestServlet.java

@@ -9,13 +9,24 @@
 import javax.servlet.http.HttpServlet;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+import org.apache.log4j.LogManager;
+import org.apache.log4j.Logger;
 
 @WebServlet("/*")
 public class TestServlet extends HttpServlet {
 
+  private static final Logger logger = LogManager.getLogger("smoketestapp-livemetrics");
+
   @Override
   protected void doGet(HttpServletRequest req, HttpServletResponse resp)
       throws ServletException, IOException {
+    doWork();
     resp.getWriter().println("ok");
   }
+
+  private void doWork() {
+    System.out.println("Doing work to generate a dependency call, exception, and trace.");
+    logger.error("This message should generate an exception!", new Exception("Fake Exception"));
+    logger.info("This message should generate a trace");
+  }
 }

smoke-tests/apps/LiveMetrics/src/smokeTest/java/com/microsoft/applicationinsights/smoketest/LiveMetricsTest.java

@@ -30,11 +30,10 @@ abstract class LiveMetricsTest {
   @Test
   @TargetUri("/test")
   void doMostBasicTest() throws Exception {
-    testing.getTelemetry(0);
 
     Awaitility.await()
-        .atMost(Duration.ofSeconds(10))
-        .until(() -> testing.mockedIngestion.isLiveMetricsPingReceived());
+        .atMost(Duration.ofSeconds(10));
+        //.until(() -> testing.mockedIngestion.isLiveMetricsPingReceived());
   }
 
   @Environment(TOMCAT_8_JAVA_8)

smoke-tests/framework/src/main/java/com/microsoft/applicationinsights/smoketest/fakeingestion/MockedAppInsightsIngestionServer.java

@@ -24,6 +24,7 @@ public class MockedAppInsightsIngestionServer {
 
   private final MockedAppInsightsIngestionServlet servlet;
   private final MockedProfilerSettingsServlet profilerSettingsServlet;
+  private final MockedQuickPulseServlet quickPulseServlet;
   private final Server server;
 
   public MockedAppInsightsIngestionServer() {
@@ -33,9 +34,11 @@ public MockedAppInsightsIngestionServer() {
 
     servlet = new MockedAppInsightsIngestionServlet();
     profilerSettingsServlet = new MockedProfilerSettingsServlet();
+    quickPulseServlet = new MockedQuickPulseServlet();
 
     handler.addServletWithMapping(new ServletHolder(profilerSettingsServlet), "/profiler/*");
-    handler.addServletWithMapping(new ServletHolder(servlet), "/*");
+    handler.addServletWithMapping(new ServletHolder(quickPulseServlet), "/QuickPulseService.svc/*");
+    handler.addServletWithMapping(new ServletHolder(servlet), "/v2.1/track");
   }
 
   @SuppressWarnings("SystemOut")
@@ -276,8 +279,20 @@ public boolean test(Envelope input) {
     return items;
   }
 
-  public boolean isLiveMetricsPingReceived() {
-    return servlet.isLiveMetricsPingReceived();
+  public int getNumPingsReceived() {
+    return quickPulseServlet.getNumPingsReceived();
+  }
+
+  public int getNumPostsReceived() {
+    return quickPulseServlet.getNumPostsReceived();
+  }
+
+  public String getLastPingBody() {
+    return quickPulseServlet.getPingBody();
+  }
+
+  public String getLastPostBody() {
+    return quickPulseServlet.getLastPostBody();
   }
 
   @SuppressWarnings("SystemOut")

smoke-tests/framework/src/main/java/com/microsoft/applicationinsights/smoketest/fakeingestion/MockedAppInsightsIngestionServlet.java

@@ -32,8 +32,6 @@ class MockedAppInsightsIngestionServlet extends HttpServlet {
 
   private final Object multimapLock = new Object();
 
-  private volatile boolean liveMetricsPingReceived;
-
   private volatile boolean loggingEnabled;
 
   MockedAppInsightsIngestionServlet() {
@@ -51,7 +49,6 @@ void resetData() {
     synchronized (multimapLock) {
       type2envelope.clear();
     }
-    liveMetricsPingReceived = false;
   }
 
   boolean hasData() {
@@ -100,14 +97,11 @@ List<Envelope> waitForItems(
     throw new TimeoutException("timed out waiting for items");
   }
 
-  boolean isLiveMetricsPingReceived() {
-    return liveMetricsPingReceived;
-  }
 
   @Override
   protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws IOException {
-    if (!"/v2.1/track".equals(req.getPathInfo())
-        && !"/QuickPulseService.svc/ping".equals(req.getPathInfo())) {
+    logit("request path info: " + req.getPathInfo());
+    if (!"/v2.1/track".equals(req.getPathInfo())) {
       resp.sendError(404, "Unknown URI");
       return;
     }
@@ -126,12 +120,6 @@ protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws I
     resp.setContentType("application/json");
     logit("raw payload:\n\n" + body + "\n");
 
-    if ("/QuickPulseService.svc/ping".equals(req.getPathInfo())) {
-      liveMetricsPingReceived = true;
-      resp.setHeader("x-ms-qps-subscribed", "false");
-      return;
-    }
-
     String[] lines = body.split("\n");
     for (String line : lines) {
       Envelope envelope = JsonHelper.GSON.fromJson(line.trim(), Envelope.class);

smoke-tests/framework/src/main/java/com/microsoft/applicationinsights/smoketest/fakeingestion/MockedQuickPulseServlet.java

@@ -0,0 +1,76 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+package com.microsoft.applicationinsights.smoketest.fakeingestion;
+
+import com.google.common.io.CharStreams;
+import javax.servlet.http.HttpServlet;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.io.StringWriter;
+import java.util.concurrent.atomic.AtomicInteger;
+import java.util.concurrent.atomic.AtomicReference;
+
+public class MockedQuickPulseServlet extends HttpServlet {
+
+  private final AtomicInteger liveMetricsPingReceived = new AtomicInteger();
+
+  private final AtomicInteger liveMetricsPostReceived = new AtomicInteger();
+
+  private final AtomicReference<String> pingBody = new AtomicReference<>();
+  private final AtomicReference<String> lastPostBody = new AtomicReference<>();
+
+  private static final String MOCK_RESPONSE_JSON_DEFAULT_CONFIG =
+      "{\"ETag\":\"fake::etag\",\"Metrics\":[],\"QuotaInfo\":null,\"DocumentStreams\":[{\"Id\":\"all-types-default\",\"DocumentFilterGroups\":[{\"TelemetryType\":\"Request\",\"Filters\":{\"Filters\":[{\"FieldName\":\"Success\",\"Predicate\":\"Equal\",\"Comparand\":\"false\"}]}},{\"TelemetryType\":\"Dependency\",\"Filters\":{\"Filters\":[{\"FieldName\":\"Success\",\"Predicate\":\"Equal\",\"Comparand\":\"false\"}]}},{\"TelemetryType\":\"Exception\",\"Filters\":{\"Filters\":[]}},{\"TelemetryType\":\"Event\",\"Filters\":{\"Filters\":[]}},{\"TelemetryType\":\"Trace\",\"Filters\":{\"Filters\":[]}}]}]}";
+
+  public MockedQuickPulseServlet() {
+
+  }
+
+  @Override
+  protected void doPost(HttpServletRequest req, HttpServletResponse resp)
+    throws IOException {
+      Readable reader = req.getReader();
+      StringWriter sw = new StringWriter();
+      CharStreams.copy(reader, sw);
+      String body = sw.toString();
+
+      String path = req.getPathInfo();
+      System.out.println("QuickPulse received: " + path);
+      if (path.equals("QuickPulse.svc/ping")) {
+          liveMetricsPingReceived.incrementAndGet();
+          pingBody.set(body);
+          System.out.println("ping body: " + body);
+          // want the following request to be a post
+          resp.setHeader("x-ms-qps-subscribed", "true");
+          resp.setContentType("application/json");
+          resp.getWriter().write(MOCK_RESPONSE_JSON_DEFAULT_CONFIG);
+
+      } else if (path.equals("QuickPulse.svc/post")) {
+          liveMetricsPostReceived.incrementAndGet();
+          lastPostBody.set(body);
+          System.out.println("post body: " + body);
+          // continue to post
+          resp.setHeader("x-ms-qps-subscribed", "true");
+      } else {
+          resp.setStatus(404);
+      }
+  }
+
+  public int getNumPingsReceived() {
+    return liveMetricsPingReceived.get();
+  }
+
+  public int getNumPostsReceived() {
+    return liveMetricsPostReceived.get();
+  }
+
+  public String getPingBody() {
+    return pingBody.get();
+  }
+
+  public String getLastPostBody() {
+    return lastPostBody.get();
+  }
+
+}