Add SampleRate verification support to assertion framework and HttpClients test

Co-authored-by: trask <[email protected]>

Author: Copilot

smoke-tests/apps/HttpClients/src/smokeTest/java/com/microsoft/applicationinsights/smoketest/HttpClientTest.java

@@ -86,7 +86,8 @@ private static void verify(String successUrlWithQueryString) throws Exception {
                         request
                             .hasSuccess(true)
                             .hasProperty("_MS.ProcessedByMetricExtractors", "True")
-                            .hasNoParent())
+                            .hasNoParent()
+                            .hasNoSampleRate())
                 .hasDependencySatisying(
                     dependency ->
                         dependency
@@ -97,7 +98,8 @@ private static void verify(String successUrlWithQueryString) throws Exception {
                             .hasResultCode("200")
                             .hasSuccess(true)
                             .hasProperty("_MS.ProcessedByMetricExtractors", "True")
-                            .hasParent(trace.getRequestId(0)))
+                            .hasParent(trace.getRequestId(0))
+                            .hasNoSampleRate())
                 .hasDependencySatisying(
                     dependency ->
                         dependency
@@ -108,7 +110,8 @@ private static void verify(String successUrlWithQueryString) throws Exception {
                             .hasResultCode("404")
                             .hasSuccess(false)
                             .hasProperty("_MS.ProcessedByMetricExtractors", "True")
-                            .hasParent(trace.getRequestId(0)))
+                            .hasParent(trace.getRequestId(0))
+                            .hasNoSampleRate())
                 .hasDependencySatisying(
                     dependency ->
                         dependency
@@ -119,7 +122,8 @@ private static void verify(String successUrlWithQueryString) throws Exception {
                             .hasResultCode("500")
                             .hasSuccess(false)
                             .hasProperty("_MS.ProcessedByMetricExtractors", "True")
-                            .hasParent(trace.getRequestId(0))));
+                            .hasParent(trace.getRequestId(0))
+                            .hasNoSampleRate()));
   }
 
   @Environment(TOMCAT_8_JAVA_8)

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

@@ -87,6 +87,20 @@ public DependencyAssert hasResultCode(String resultCode) {
     return this;
   }
 
+  @CanIgnoreReturnValue
+  public DependencyAssert hasSampleRate(Float expectedSampleRate) {
+    isNotNull();
+    assertThat(actual.getSampleRate()).isEqualTo(expectedSampleRate);
+    return this;
+  }
+
+  @CanIgnoreReturnValue
+  public DependencyAssert hasNoSampleRate() {
+    isNotNull();
+    assertThat(actual.getSampleRate()).isNull();
+    return this;
+  }
+
   private RemoteDependencyData getDependencyData() {
     Data<?> data = (Data<?>) actual.getData();
     return (RemoteDependencyData) data.getBaseData();

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

@@ -58,6 +58,20 @@ public RequestAssert hasParent(String parentId) {
     return this;
   }
 
+  @CanIgnoreReturnValue
+  public RequestAssert hasSampleRate(Float expectedSampleRate) {
+    isNotNull();
+    assertThat(actual.getSampleRate()).isEqualTo(expectedSampleRate);
+    return this;
+  }
+
+  @CanIgnoreReturnValue
+  public RequestAssert hasNoSampleRate() {
+    isNotNull();
+    assertThat(actual.getSampleRate()).isNull();
+    return this;
+  }
+
   private RequestData getRequestData() {
     Data<?> data = (Data<?>) actual.getData();
     return (RequestData) data.getBaseData();