1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/browser/upgrade_detector_impl.h"
6
7 #include <vector>
8
9 #include "content/public/browser/notification_details.h"
10 #include "content/public/browser/notification_observer.h"
11 #include "content/public/browser/notification_registrar.h"
12 #include "content/public/browser/notification_service.h"
13 #include "content/public/test/test_browser_thread_bundle.h"
14 #include "testing/gtest/include/gtest/gtest.h"
15
16 class TestUpgradeDetectorImpl : public UpgradeDetectorImpl {
17 public:
TestUpgradeDetectorImpl()18 TestUpgradeDetectorImpl() : trigger_critical_update_call_count_(0) {}
~TestUpgradeDetectorImpl()19 virtual ~TestUpgradeDetectorImpl() {}
20
21 // Methods exposed for testing.
22 using UpgradeDetectorImpl::OnExperimentChangesDetected;
23 using UpgradeDetectorImpl::NotifyOnUpgradeWithTimePassed;
24
25 // UpgradeDetector:
TriggerCriticalUpdate()26 virtual void TriggerCriticalUpdate() OVERRIDE {
27 trigger_critical_update_call_count_++;
28 }
29
trigger_critical_update_call_count() const30 int trigger_critical_update_call_count() const {
31 return trigger_critical_update_call_count_;
32 }
33
34 private:
35 // How many times TriggerCriticalUpdate() has been called. Expected to either
36 // be 0 or 1.
37 int trigger_critical_update_call_count_;
38
39 DISALLOW_COPY_AND_ASSIGN(TestUpgradeDetectorImpl);
40 };
41
42 class TestUpgradeNotificationListener : public content::NotificationObserver {
43 public:
TestUpgradeNotificationListener()44 TestUpgradeNotificationListener() {
45 registrar_.Add(this, chrome::NOTIFICATION_UPGRADE_RECOMMENDED,
46 content::NotificationService::AllSources());
47 }
~TestUpgradeNotificationListener()48 virtual ~TestUpgradeNotificationListener() {
49 }
50
notifications_received() const51 const std::vector<int>& notifications_received() const {
52 return notifications_received_;
53 }
54
55 private:
56 // content::NotificationObserver:
Observe(int type,const content::NotificationSource & source,const content::NotificationDetails & details)57 virtual void Observe(int type,
58 const content::NotificationSource& source,
59 const content::NotificationDetails& details) OVERRIDE {
60 notifications_received_.push_back(type);
61 }
62
63 // Registrar for listening to notifications.
64 content::NotificationRegistrar registrar_;
65
66 // Keeps track of the number and types of notifications that were received.
67 std::vector<int> notifications_received_;
68
69 DISALLOW_COPY_AND_ASSIGN(TestUpgradeNotificationListener);
70 };
71
TEST(UpgradeDetectorImplTest,VariationsChanges)72 TEST(UpgradeDetectorImplTest, VariationsChanges) {
73 content::TestBrowserThreadBundle bundle;
74
75 TestUpgradeNotificationListener notifications_listener;
76 TestUpgradeDetectorImpl detector;
77 EXPECT_FALSE(detector.notify_upgrade());
78 EXPECT_TRUE(notifications_listener.notifications_received().empty());
79
80 detector.OnExperimentChangesDetected(
81 chrome_variations::VariationsService::Observer::BEST_EFFORT);
82 EXPECT_FALSE(detector.notify_upgrade());
83 EXPECT_TRUE(notifications_listener.notifications_received().empty());
84
85 detector.NotifyOnUpgradeWithTimePassed(base::TimeDelta::FromDays(30));
86 EXPECT_TRUE(detector.notify_upgrade());
87 ASSERT_EQ(1U, notifications_listener.notifications_received().size());
88 EXPECT_EQ(chrome::NOTIFICATION_UPGRADE_RECOMMENDED,
89 notifications_listener.notifications_received().front());
90 EXPECT_EQ(0, detector.trigger_critical_update_call_count());
91 }
92
TEST(UpgradeDetectorImplTest,VariationsCriticalChanges)93 TEST(UpgradeDetectorImplTest, VariationsCriticalChanges) {
94 content::TestBrowserThreadBundle bundle;
95
96 TestUpgradeNotificationListener notifications_listener;
97 TestUpgradeDetectorImpl detector;
98 EXPECT_FALSE(detector.notify_upgrade());
99 EXPECT_TRUE(notifications_listener.notifications_received().empty());
100
101 detector.OnExperimentChangesDetected(
102 chrome_variations::VariationsService::Observer::CRITICAL);
103 EXPECT_FALSE(detector.notify_upgrade());
104 EXPECT_TRUE(notifications_listener.notifications_received().empty());
105
106 detector.NotifyOnUpgradeWithTimePassed(base::TimeDelta::FromDays(30));
107 EXPECT_TRUE(detector.notify_upgrade());
108 ASSERT_EQ(1U, notifications_listener.notifications_received().size());
109 EXPECT_EQ(chrome::NOTIFICATION_UPGRADE_RECOMMENDED,
110 notifications_listener.notifications_received().front());
111 EXPECT_EQ(1, detector.trigger_critical_update_call_count());
112 }
113