Add network information to UKM

components/ukm/ukm_service_unittest.cc

 // Copyright 2017 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 "components/ukm/ukm_service.h"
 
 #include "base/test/test_simple_task_runner.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "components/metrics/proto/ukm/report.pb.h"
 #include "components/metrics/proto/ukm/source.pb.h"
+#include "components/metrics/test_metrics_provider.h"
 #include "components/metrics/test_metrics_service_client.h"
 #include "components/prefs/testing_pref_service.h"
 #include "components/ukm/persisted_logs_metrics_impl.h"
 #include "components/ukm/ukm_pref_names.h"
 #include "components/ukm/ukm_source.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/zlib/google/compression_utils.h"
 
 namespace ukm {
 
@@ skipping 94 matching lines @@
   EXPECT_EQ(GetPersistedLogCount(), 1);
 
   Report proto_report = GetPersistedReport();
   EXPECT_EQ(1, proto_report.sources_size());
   const Source& proto_source = proto_report.sources(0);
 
   EXPECT_EQ(GURL("https://google.com").spec(), proto_source.url());
   EXPECT_EQ(300, proto_source.first_contentful_paint_msec());
 }
 
+TEST_F(UkmServiceTest, MetricsProviderTest) {
+  UkmService service(&prefs_, &client_);
+
+  metrics::TestMetricsProvider* provider = new metrics::TestMetricsProvider();
+  service.RegisterMetricsProvider(
+      std::unique_ptr<metrics::MetricsProvider>(provider));
+
+  service.Initialize();
+
+  // Providers have not supplied system profile information yet.
+  EXPECT_FALSE(provider->provide_system_profile_metrics_called());
+
+  task_runner_->RunUntilIdle();
+  service.EnableReporting();
+
+  service.RecordSource(base::WrapUnique(new UkmSource()));
+  service.Flush();
+  EXPECT_EQ(GetPersistedLogCount(), 1);
+
+  Report proto_report = GetPersistedReport();
+  EXPECT_EQ(1, proto_report.sources_size());
+
+  // Providers have now supplied system profile information.
+  EXPECT_TRUE(provider->provide_system_profile_metrics_called());
+}
+
 }  // namespace ukm