Index: ios/chrome/browser/upgrade/upgrade_center_unittest.mm |
diff --git a/ios/chrome/browser/upgrade/upgrade_center_unittest.mm b/ios/chrome/browser/upgrade/upgrade_center_unittest.mm |
new file mode 100644 |
index 0000000000000000000000000000000000000000..b8b1b1ebe3ab944595a510b7502c64bbf229c105 |
--- /dev/null |
+++ b/ios/chrome/browser/upgrade/upgrade_center_unittest.mm |
@@ -0,0 +1,139 @@ |
+// Copyright (c) 2012 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. |
+ |
+#import "ios/chrome/browser/upgrade/upgrade_center.h" |
+ |
+#include "base/mac/scoped_nsobject.h" |
+#include "ios/chrome/browser/upgrade/upgrade_recommended_details.h" |
+#include "testing/platform_test.h" |
+ |
+namespace { |
+ |
+class UpgradeCenterTest : public PlatformTest { |
+ public: |
+ unsigned int count_; |
+ |
+ protected: |
+ void SetUp() override { |
+ [[UpgradeCenter sharedInstance] resetForTests]; |
+ count_ = 0; |
+ }; |
+ |
+ void TearDown() override { [[UpgradeCenter sharedInstance] resetForTests]; } |
+}; |
+ |
+} // namespace |
+ |
+@interface FakeUpgradeCenterClient : NSObject<UpgradeCenterClientProtocol> |
+- (instancetype)initWithTest:(UpgradeCenterTest*)test; |
+@end |
+ |
+@implementation FakeUpgradeCenterClient { |
+ UpgradeCenterTest* test_; // weak |
+} |
+ |
+- (instancetype)initWithTest:(UpgradeCenterTest*)test { |
+ self = [super init]; |
+ if (self) { |
+ test_ = test; |
+ } |
+ return self; |
+} |
+ |
+- (void)showUpgrade:(UpgradeCenter*)center { |
+ test_->count_ += 1; |
+} |
+ |
+@end |
+ |
+namespace { |
+ |
+TEST_F(UpgradeCenterTest, NoUpgrade) { |
+ EXPECT_EQ(count_, 0u); |
+ base::scoped_nsobject<FakeUpgradeCenterClient> fake( |
+ [[FakeUpgradeCenterClient alloc] initWithTest:this]); |
+ [[UpgradeCenter sharedInstance] registerClient:fake]; |
+ EXPECT_EQ(count_, 0u); |
+ [[UpgradeCenter sharedInstance] unregisterClient:fake]; |
+}; |
+ |
+TEST_F(UpgradeCenterTest, GoodUpgradeAfterRegistration) { |
+ EXPECT_EQ(count_, 0u); |
+ base::scoped_nsobject<FakeUpgradeCenterClient> fake( |
+ [[FakeUpgradeCenterClient alloc] initWithTest:this]); |
+ [[UpgradeCenter sharedInstance] registerClient:fake]; |
+ EXPECT_EQ(count_, 0u); |
+ |
+ UpgradeRecommendedDetails details; |
+ details.next_version = "9999.9999.9999.9999"; |
+ details.upgrade_url = GURL("http://foobar.org"); |
+ [[UpgradeCenter sharedInstance] upgradeNotificationDidOccur:details]; |
+ EXPECT_EQ(count_, 1u); |
+ [[UpgradeCenter sharedInstance] unregisterClient:fake]; |
+}; |
+ |
+TEST_F(UpgradeCenterTest, GoodUpgradeBeforeRegistration) { |
+ UpgradeRecommendedDetails details; |
+ details.next_version = "9999.9999.9999.9999"; |
+ details.upgrade_url = GURL("http://foobar.org"); |
+ [[UpgradeCenter sharedInstance] upgradeNotificationDidOccur:details]; |
+ EXPECT_EQ(count_, 0u); |
+ base::scoped_nsobject<FakeUpgradeCenterClient> fake( |
+ [[FakeUpgradeCenterClient alloc] initWithTest:this]); |
+ [[UpgradeCenter sharedInstance] registerClient:fake]; |
+ EXPECT_EQ(count_, 1u); |
+ [[UpgradeCenter sharedInstance] unregisterClient:fake]; |
+}; |
+ |
+TEST_F(UpgradeCenterTest, NoRepeatedDisplay) { |
+ base::scoped_nsobject<FakeUpgradeCenterClient> fake( |
+ [[FakeUpgradeCenterClient alloc] initWithTest:this]); |
+ [[UpgradeCenter sharedInstance] registerClient:fake]; |
+ EXPECT_EQ(count_, 0u); |
+ |
+ // First notification should display |
+ UpgradeRecommendedDetails details; |
+ details.next_version = "9999.9999.9999.9999"; |
+ details.upgrade_url = GURL("http://foobar.org"); |
+ [[UpgradeCenter sharedInstance] upgradeNotificationDidOccur:details]; |
+ EXPECT_EQ(count_, 1u); |
+ |
+ // Second shouldn't, since it was just displayed. |
+ [[UpgradeCenter sharedInstance] upgradeNotificationDidOccur:details]; |
+ EXPECT_EQ(count_, 1u); |
+ |
+ // After enough time has elapsed, it should again. |
+ [[UpgradeCenter sharedInstance] setLastDisplayToPast]; |
+ [[UpgradeCenter sharedInstance] upgradeNotificationDidOccur:details]; |
+ EXPECT_EQ(count_, 2u); |
+ |
+ [[UpgradeCenter sharedInstance] unregisterClient:fake]; |
+}; |
+ |
+TEST_F(UpgradeCenterTest, NewVersionResetsInterval) { |
+ base::scoped_nsobject<FakeUpgradeCenterClient> fake( |
+ [[FakeUpgradeCenterClient alloc] initWithTest:this]); |
+ [[UpgradeCenter sharedInstance] registerClient:fake]; |
+ EXPECT_EQ(count_, 0u); |
+ |
+ // First notification should display |
+ UpgradeRecommendedDetails details; |
+ details.next_version = "9999.9999.9999.9998"; |
+ details.upgrade_url = GURL("http://foobar.org"); |
+ [[UpgradeCenter sharedInstance] upgradeNotificationDidOccur:details]; |
+ EXPECT_EQ(count_, 1u); |
+ |
+ // Second shouldn't, since it was just displayed. |
+ [[UpgradeCenter sharedInstance] upgradeNotificationDidOccur:details]; |
+ EXPECT_EQ(count_, 1u); |
+ |
+ // A new version should show right away though. |
+ details.next_version = "9999.9999.9999.9999"; |
+ [[UpgradeCenter sharedInstance] upgradeNotificationDidOccur:details]; |
+ EXPECT_EQ(count_, 2u); |
+ |
+ [[UpgradeCenter sharedInstance] unregisterClient:fake]; |
+}; |
+ |
+} // namespace |