UKM: Remove UkmService::RecordSource

components/autofill/core/browser/autofill_manager_unittest.cc

 // Copyright 2013 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/autofill/core/browser/autofill_manager.h"
 
 #include <stddef.h>
 
 #include <algorithm>
 #include <memory>
@@ skipping 774 matching lines @@
 const ukm::Entry_Metric* FindMetric(
     const char* name,
     const google::protobuf::RepeatedPtrField<ukm::Entry_Metric>& metrics) {
   for (const auto& metric : metrics) {
     if (metric.metric_hash() == base::HashMetricName(name))
       return &metric;
   }
   return nullptr;
 }
 
+// Get Ukm sources from the Ukm service.
+std::vector<const ukm::UkmSource*> GetUkmSources(ukm::TestUkmService* service) {
+  std::vector<const ukm::UkmSource*> sources;
+  for (const auto& kv : service->GetSources())
+    sources.push_back(kv.second.get());
+
+  return sources;
+}
+
 }  // namespace
 
 class AutofillManagerTest : public testing::Test {
  public:
   AutofillManagerTest() : field_trial_list_(nullptr) {}
 
   void SetUp() override {
     autofill_client_.SetPrefs(test::PrefServiceForTesting());
     personal_data_.set_database(autofill_client_.GetDatabase());
     personal_data_.SetPrefService(autofill_client_.GetPrefs());
@@ skipping 214 matching lines @@
   void EnableUkmLogging() {
     scoped_feature_list_.InitAndEnableFeature(kAutofillUkmLogging);
   }
 
   void ExpectUniqueCardUploadDecisionUkm(
       AutofillMetrics::CardUploadDecisionMetric upload_decision) {
     ukm::TestUkmService* ukm_service = autofill_client_.GetTestUkmService();
 
     // Check that one source is logged.
     ASSERT_EQ(1U, ukm_service->sources_count());
-    const ukm::UkmSource* source = ukm_service->GetSource(0);
+    const ukm::UkmSource* source = GetUkmSources(ukm_service)[0];
 
     // Check that one entry is logged.
     EXPECT_EQ(1U, ukm_service->entries_count());
     const ukm::UkmEntry* entry = ukm_service->GetEntry(0);
     EXPECT_EQ(source->id(), entry->source_id());
 
     ukm::Entry entry_proto;
     entry->PopulateProto(&entry_proto);
     EXPECT_EQ(source->id(), entry_proto.source_id());
 
@@ skipping 4790 matching lines @@
   FormsSeen({form});
 
   // Suggestions should be displayed.
   for (const FormFieldData& field : form.fields) {
     GetAutofillSuggestions(form, field);
     EXPECT_TRUE(external_delegate_->on_suggestions_returned_seen());
   }
 }
 
 }  // namespace autofill