Unit tests

Author: smalyshev

server/src/test/java/org/elasticsearch/action/admin/cluster/stats/CCSTelemetrySnapshotTests.java

@@ -352,4 +352,20 @@ public void testRanges() throws IOException {
         assertThat(value2Read.count(), equalTo(count1 + count2));
         assertThat(value2Read.max(), equalTo(max1));
     }
+
+    public void testUseMRTFalse() {
+        CCSTelemetrySnapshot empty = new CCSTelemetrySnapshot();
+        // Ignore MRT data
+        empty.setUseMRT(false);
+
+        var randomWithMRT = randomValueOtherThanMany(
+            v -> v.getTookMrtTrue().count() == 0 || v.getTookMrtFalse().count() == 0,
+            this::randomCCSTelemetrySnapshot
+        );
+
+        empty.add(randomWithMRT);
+        assertThat(empty.getTook().count(), equalTo(randomWithMRT.getTook().count()));
+        assertThat(empty.getTookMrtFalse().count(), equalTo(0L));
+        assertThat(empty.getTookMrtTrue().count(), equalTo(0L));
+    }
 }

server/src/test/java/org/elasticsearch/action/admin/cluster/stats/CCSUsageTelemetryTests.java

@@ -340,4 +340,23 @@ public void testConcurrentUpdates() throws InterruptedException {
         CCSTelemetrySnapshot expectedSnapshot = ccsUsageHolder.getCCSTelemetrySnapshot();
         assertThat(snapshot, equalTo(expectedSnapshot));
     }
+
+    public void testUseMRTFalse() {
+        // Ignore MRT counters if instructed.
+        CCSUsageTelemetry ccsUsageHolder = new CCSUsageTelemetry(false);
+
+        CCSUsage.Builder builder = new CCSUsage.Builder();
+        builder.took(10L).setRemotesCount(1).setClient("kibana");
+        builder.setFeature(MRT_FEATURE);
+        ccsUsageHolder.updateUsage(builder.build());
+
+        builder = new CCSUsage.Builder();
+        builder.took(11L).setRemotesCount(1).setClient("kibana");
+        ccsUsageHolder.updateUsage(builder.build());
+
+        CCSTelemetrySnapshot snapshot = ccsUsageHolder.getCCSTelemetrySnapshot();
+        assertThat(snapshot.getTook().count(), equalTo(2L));
+        assertThat(snapshot.getTookMrtFalse().count(), equalTo(0L));
+        assertThat(snapshot.getTookMrtTrue().count(), equalTo(0L));
+    }
 }