| // Copyright 2019 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 "base/bind.h" |
| #include "base/macros.h" |
| #include "base/metrics/user_metrics.h" |
| #include "base/optional.h" |
| #include "base/test/scoped_feature_list.h" |
| #include "base/version.h" |
| #include "chrome/browser/browser_process.h" |
| #include "chrome/browser/metrics/chrome_metrics_service_accessor.h" |
| #include "chrome/browser/profiles/profile_impl.h" |
| #include "chrome/browser/ui/browser.h" |
| #include "chrome/browser/ui/hats/hats_service.h" |
| #include "chrome/browser/ui/hats/hats_service_factory.h" |
| #include "chrome/browser/ui/views/hats/hats_bubble_view.h" |
| #include "chrome/common/chrome_features.h" |
| #include "chrome/test/base/in_process_browser_test.h" |
| #include "components/metrics_services_manager/metrics_services_manager.h" |
| #include "components/version_info/version_info.h" |
| #include "content/public/test/browser_test.h" |
| |
| namespace { |
| |
| class ScopedSetMetricsConsent { |
| public: |
| // Enables or disables metrics consent based off of |consent|. |
| explicit ScopedSetMetricsConsent(bool consent) : consent_(consent) { |
| ChromeMetricsServiceAccessor::SetMetricsAndCrashReportingForTesting( |
| &consent_); |
| } |
| |
| ~ScopedSetMetricsConsent() { |
| ChromeMetricsServiceAccessor::SetMetricsAndCrashReportingForTesting( |
| nullptr); |
| } |
| |
| private: |
| const bool consent_; |
| |
| DISALLOW_COPY_AND_ASSIGN(ScopedSetMetricsConsent); |
| }; |
| |
| class HatsServiceBrowserTestBase : public InProcessBrowserTest { |
| protected: |
| HatsServiceBrowserTestBase() = default; |
| ~HatsServiceBrowserTestBase() override = default; |
| |
| HatsService* GetHatsService() { |
| return HatsServiceFactory::GetForProfile(browser()->profile(), true); |
| } |
| |
| void SetMetricsConsent(bool consent) { |
| scoped_metrics_consent_.emplace(consent); |
| } |
| |
| bool HatsBubbleShown() { |
| views::BubbleDialogDelegateView* bubble = HatsBubbleView::GetHatsBubble(); |
| if (!bubble) |
| return false; |
| |
| views::Widget* widget = bubble->GetWidget(); |
| return widget && widget->IsVisible(); |
| } |
| |
| private: |
| base::Optional<ScopedSetMetricsConsent> scoped_metrics_consent_; |
| |
| DISALLOW_COPY_AND_ASSIGN(HatsServiceBrowserTestBase); |
| }; |
| |
| } // namespace |
| |
| IN_PROC_BROWSER_TEST_F(HatsServiceBrowserTestBase, BubbleNotShownOnDefault) { |
| GetHatsService()->LaunchSatisfactionSurvey(); |
| EXPECT_FALSE(HatsBubbleShown()); |
| } |
| |
| namespace { |
| |
| class HatsServiceProbabilityZero : public HatsServiceBrowserTestBase { |
| protected: |
| HatsServiceProbabilityZero() = default; |
| ~HatsServiceProbabilityZero() override = default; |
| |
| private: |
| void SetUpOnMainThread() override { |
| HatsServiceBrowserTestBase::SetUpOnMainThread(); |
| |
| // The HatsService must not be created so that feature parameters can be |
| // injected below. |
| ASSERT_FALSE( |
| HatsServiceFactory::GetForProfile(browser()->profile(), false)); |
| scoped_feature_list_.InitAndEnableFeatureWithParameters( |
| features::kHappinessTrackingSurveysForDesktop, |
| {{"probability", "0.000"}}); |
| } |
| |
| base::test::ScopedFeatureList scoped_feature_list_; |
| |
| DISALLOW_COPY_AND_ASSIGN(HatsServiceProbabilityZero); |
| }; |
| |
| } // namespace |
| |
| IN_PROC_BROWSER_TEST_F(HatsServiceProbabilityZero, NoShow) { |
| GetHatsService()->LaunchSatisfactionSurvey(); |
| EXPECT_FALSE(HatsBubbleShown()); |
| } |
| |
| namespace { |
| |
| class HatsServiceProbabilityOne : public HatsServiceBrowserTestBase { |
| protected: |
| HatsServiceProbabilityOne() = default; |
| ~HatsServiceProbabilityOne() override = default; |
| |
| private: |
| void SetUpOnMainThread() override { |
| HatsServiceBrowserTestBase::SetUpOnMainThread(); |
| |
| // The HatsService must not be created so that feature parameters can be |
| // injected below. |
| ASSERT_FALSE( |
| HatsServiceFactory::GetForProfile(browser()->profile(), false)); |
| // TODO(weili): refactor to use constants from hats_service.cc for these |
| // parameters. |
| scoped_feature_list_.InitAndEnableFeatureWithParameters( |
| features::kHappinessTrackingSurveysForDesktop, |
| {{"probability", "1.000"}, |
| {"survey", "satisfaction"}, |
| {"en_site_id", "test_site_id"}}); |
| // Set the profile creation time to be old enough to ensure triggering. |
| browser()->profile()->SetCreationTimeForTesting( |
| base::Time::Now() - base::TimeDelta::FromDays(45)); |
| } |
| |
| void TearDownOnMainThread() override { |
| GetHatsService()->SetSurveyMetadataForTesting({}); |
| } |
| |
| base::test::ScopedFeatureList scoped_feature_list_; |
| |
| DISALLOW_COPY_AND_ASSIGN(HatsServiceProbabilityOne); |
| }; |
| |
| } // namespace |
| |
| IN_PROC_BROWSER_TEST_F(HatsServiceProbabilityOne, NoShowConsentNotGiven) { |
| SetMetricsConsent(false); |
| ASSERT_FALSE( |
| g_browser_process->GetMetricsServicesManager()->IsMetricsConsentGiven()); |
| GetHatsService()->LaunchSatisfactionSurvey(); |
| EXPECT_FALSE(HatsBubbleShown()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(HatsServiceProbabilityOne, AlwaysShow) { |
| SetMetricsConsent(true); |
| ASSERT_TRUE( |
| g_browser_process->GetMetricsServicesManager()->IsMetricsConsentGiven()); |
| GetHatsService()->LaunchSatisfactionSurvey(); |
| EXPECT_TRUE(HatsBubbleShown()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(HatsServiceProbabilityOne, |
| DoubleShowOnlyResultsInOneShow) { |
| SetMetricsConsent(true); |
| ASSERT_TRUE( |
| g_browser_process->GetMetricsServicesManager()->IsMetricsConsentGiven()); |
| |
| GetHatsService()->LaunchSatisfactionSurvey(); |
| EXPECT_TRUE(HatsBubbleShown()); |
| views::BubbleDialogDelegateView* bubble1 = HatsBubbleView::GetHatsBubble(); |
| |
| GetHatsService()->LaunchSatisfactionSurvey(); |
| EXPECT_TRUE(HatsBubbleShown()); |
| EXPECT_EQ(bubble1, HatsBubbleView::GetHatsBubble()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(HatsServiceProbabilityOne, SameMajorVersionNoShow) { |
| SetMetricsConsent(true); |
| HatsService::SurveyMetadata metadata; |
| metadata.last_major_version = version_info::GetVersion().components()[0]; |
| GetHatsService()->SetSurveyMetadataForTesting(metadata); |
| GetHatsService()->LaunchSatisfactionSurvey(); |
| EXPECT_FALSE(HatsBubbleShown()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(HatsServiceProbabilityOne, DifferentMajorVersionShow) { |
| SetMetricsConsent(true); |
| HatsService::SurveyMetadata metadata; |
| metadata.last_major_version = 42; |
| ASSERT_NE(42u, version_info::GetVersion().components()[0]); |
| GetHatsService()->SetSurveyMetadataForTesting(metadata); |
| GetHatsService()->LaunchSatisfactionSurvey(); |
| EXPECT_TRUE(HatsBubbleShown()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(HatsServiceProbabilityOne, |
| SurveyStartedBeforeRequiredElapsedTimeNoShow) { |
| SetMetricsConsent(true); |
| HatsService::SurveyMetadata metadata; |
| metadata.last_survey_started_time = base::Time::Now(); |
| GetHatsService()->SetSurveyMetadataForTesting(metadata); |
| GetHatsService()->LaunchSatisfactionSurvey(); |
| EXPECT_FALSE(HatsBubbleShown()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(HatsServiceProbabilityOne, ProfileTooYoungToShow) { |
| SetMetricsConsent(true); |
| // Set creation time to only 15 days. |
| static_cast<ProfileImpl*>(browser()->profile()) |
| ->SetCreationTimeForTesting(base::Time::Now() - |
| base::TimeDelta::FromDays(15)); |
| GetHatsService()->LaunchSatisfactionSurvey(); |
| EXPECT_FALSE(HatsBubbleShown()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(HatsServiceProbabilityOne, ProfileOldEnoughToShow) { |
| SetMetricsConsent(true); |
| // Set creation time to 31 days. This is just past the threshold. |
| static_cast<ProfileImpl*>(browser()->profile()) |
| ->SetCreationTimeForTesting(base::Time::Now() - |
| base::TimeDelta::FromDays(31)); |
| GetHatsService()->LaunchSatisfactionSurvey(); |
| EXPECT_TRUE(HatsBubbleShown()); |
| } |