[MediaRouter] Record DIAL device counts in DialMediaSinkServiceImpl

chrome/browser/media/router/media_router_metrics_unittest.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+#include "chrome/browser/media/router/media_router_metrics.h"
 #include "base/macros.h"
+#include "base/memory/ptr_util.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/test/histogram_tester.h"
+#include "base/test/simple_test_clock.h"
 #include "base/time/time.h"
-#include "chrome/browser/media/router/media_router_metrics.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using base::Bucket;
 using testing::ElementsAre;
 
 namespace media_router {
 
 TEST(MediaRouterMetricsTest, RecordMediaRouterDialogOrigin) {
   base::HistogramTester tester;
@@ skipping 66 matching lines @@
   MediaRouterMetrics::RecordRouteCreationOutcome(outcome1);
   MediaRouterMetrics::RecordRouteCreationOutcome(outcome2);
   tester.ExpectTotalCount(MediaRouterMetrics::kHistogramRouteCreationOutcome,
                           3);
   EXPECT_THAT(
       tester.GetAllSamples(MediaRouterMetrics::kHistogramRouteCreationOutcome),
       ElementsAre(Bucket(static_cast<int>(outcome1), 1),
                   Bucket(static_cast<int>(outcome2), 2)));
 }
 
+TEST(MediaRouterMetricsTest, RecordDialDeviceCounts) {
+  MediaRouterMetrics metrics;
+  base::SimpleTestClock* clock = new base::SimpleTestClock();
+  metrics.SetClockForTest(base::WrapUnique(clock));
+  base::HistogramTester tester;
+  tester.ExpectTotalCount(
+      MediaRouterMetrics::kHistogramDialAvailableDeviceCount, 0);
+  tester.ExpectTotalCount(MediaRouterMetrics::kHistogramDialKnownDeviceCount,
+                          0);
+
+  clock->SetNow(base::Time::Now());
+  metrics.RecordDialDeviceCounts(6, 10);
+  metrics.RecordDialDeviceCounts(7, 10);
+  tester.ExpectTotalCount(
+      MediaRouterMetrics::kHistogramDialAvailableDeviceCount, 1);
+  tester.ExpectTotalCount(MediaRouterMetrics::kHistogramDialKnownDeviceCount,
+                          1);
+  tester.ExpectBucketCount(
+      MediaRouterMetrics::kHistogramDialAvailableDeviceCount, 6, 1);
+  tester.ExpectBucketCount(MediaRouterMetrics::kHistogramDialKnownDeviceCount,
+                           10, 1);
+
+  clock->Advance(base::TimeDelta::FromHours(2));
+  metrics.RecordDialDeviceCounts(7, 10);
+
+  tester.ExpectTotalCount(
+      MediaRouterMetrics::kHistogramDialAvailableDeviceCount, 2);
+  tester.ExpectTotalCount(MediaRouterMetrics::kHistogramDialKnownDeviceCount,
+                          2);
+  tester.ExpectBucketCount(
+      MediaRouterMetrics::kHistogramDialAvailableDeviceCount, 6, 1);
+  tester.ExpectBucketCount(
+      MediaRouterMetrics::kHistogramDialAvailableDeviceCount, 7, 1);
+  tester.ExpectBucketCount(MediaRouterMetrics::kHistogramDialKnownDeviceCount,
+                           10, 2);
+}
+
 }  // namespace media_router