Add a unit test for translation event logging.

chrome/browser/translate/chrome_translate_client_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 "chrome/browser/translate/chrome_translate_client.h"
 
 #include <memory>
+#include <string>
 
 #include "base/command_line.h"
 #include "base/memory/ref_counted.h"
 #include "base/run_loop.h"
 #include "base/test/scoped_feature_list.h"
 #include "chrome/browser/sync/profile_sync_service_factory.h"
 #include "chrome/browser/sync/user_event_service_factory.h"
 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
+#include "components/metrics/proto/translate_event.pb.h"
 #include "components/sync/driver/sync_driver_switches.h"
 #include "components/sync/user_events/fake_user_event_service.h"
 #include "components/translate/core/common/language_detection_details.h"
 #include "content/public/browser/web_contents.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 
+using ::metrics::TranslateEventProto;
+
 std::unique_ptr<KeyedService> BuildFakeUserEventService(
     content::BrowserContext* context) {
   return base::MakeUnique<syncer::FakeUserEventService>();
 }
 
+TranslateEventProto BuildTranslateEventProto(
+    const std::string& from,
+    const std::string& to,
+    const TranslateEventProto::EventType type) {
+  TranslateEventProto event;
+  event.set_source_language(from);
+  event.set_target_language(to);
+  event.set_event_type(type);
+  return event;
+}
+
 class ChromeTranslateClientTest : public ChromeRenderViewHostTestHarness {
  public:
   void SetUp() override {
     ChromeRenderViewHostTestHarness::SetUp();
     fake_user_event_service_ = static_cast<syncer::FakeUserEventService*>(
         browser_sync::UserEventServiceFactory::GetInstance()
             ->SetTestingFactoryAndUse(browser_context(),
                                       &BuildFakeUserEventService));
     scoped_feature_list_ = base::MakeUnique<base::test::ScopedFeatureList>();
-    scoped_feature_list_->InitAndEnableFeature(
-        switches::kSyncUserLanguageDetectionEvents);
   }
 
   void TearDown() override { ChromeRenderViewHostTestHarness::TearDown(); }
 
  protected:
   syncer::FakeUserEventService* GetUserEventService() {
     return fake_user_event_service_;
   }
 
+  std::unique_ptr<base::test::ScopedFeatureList> scoped_feature_list_;
+
  private:
-  std::unique_ptr<base::test::ScopedFeatureList> scoped_feature_list_;
   syncer::FakeUserEventService* fake_user_event_service_;
 };
 
 TEST_F(ChromeTranslateClientTest, LanguageEventShouldRecord) {
+  scoped_feature_list_->InitAndEnableFeature(
+      switches::kSyncUserLanguageDetectionEvents);
   GURL url("http://yahoo.com");
   NavigateAndCommit(url);
   ChromeTranslateClient client(web_contents());
   translate::LanguageDetectionDetails details;
   details.cld_language = "en";
   details.is_cld_reliable = true;
   details.adopted_language = "en";
   client.OnLanguageDetermined(details);
   EXPECT_EQ(1ul, GetUserEventService()->GetRecordedUserEvents().size());
 }
 
 TEST_F(ChromeTranslateClientTest, LanguageEventShouldNotRecord) {
+  scoped_feature_list_->InitAndEnableFeature(
+      switches::kSyncUserLanguageDetectionEvents);
   GURL url("about://blank");
   NavigateAndCommit(url);
   ChromeTranslateClient client(web_contents());
   translate::LanguageDetectionDetails details;
   details.cld_language = "en";
   details.is_cld_reliable = true;
   details.adopted_language = "en";
   client.OnLanguageDetermined(details);
   EXPECT_EQ(0u, GetUserEventService()->GetRecordedUserEvents().size());
 }
+
+TEST_F(ChromeTranslateClientTest, TranslationEventShouldRecord) {
+  scoped_feature_list_->InitAndEnableFeature(

[napper, 2017/06/26 00:32:04]

Can't you just do this once in SetUp()?

[renjieliu1, 2017/06/26 01:04:42]

Done.

+      switches::kSyncUserTranslationEvents);
+  GURL url("http://yahoo.com");

[napper, 2017/06/26 00:32:04]

const?

[renjieliu1, 2017/06/26 01:04:42]

Done.

+  NavigateAndCommit(url);
+  ChromeTranslateClient client(web_contents());
+  // Event we care.
+  const TranslateEventProto& event_proto =
+      BuildTranslateEventProto("ja", "en", TranslateEventProto::USER_ACCEPT);
+  client.RecordTranslateEvent(event_proto);
+  EXPECT_EQ(1ul, GetUserEventService()->GetRecordedUserEvents().size());

[napper, 2017/06/26 00:32:04]

I think 1 instead of 1ul

[renjieliu1, 2017/06/26 01:04:42]

.size() returns a unsigned long and it seems EXPECT_EQ does not handle implicit
conversion :(

+}

[napper, 2017/06/26 00:32:04]

Can you also test the content of the event?

[renjieliu1, 2017/06/26 01:04:42]

Done.

+
+TEST_F(ChromeTranslateClientTest, TranslationEventShouldNotRecord) {
+  scoped_feature_list_->InitAndEnableFeature(
+      switches::kSyncUserTranslationEvents);
+  GURL url("http://yahoo.com");
+  NavigateAndCommit(url);
+  ChromeTranslateClient client(web_contents());
+  // Event we don't care.
+  const TranslateEventProto& event_proto = BuildTranslateEventProto(
+      "ja", "en", TranslateEventProto::UNSUPPORTED_URL);
+  client.RecordTranslateEvent(event_proto);
+  EXPECT_EQ(0u, GetUserEventService()->GetRecordedUserEvents().size());
+}