Show notifications on low disk space.

chrome/browser/chromeos/ui/low_disk_notification_unittest.cc

Index: chrome/browser/chromeos/ui/low_disk_notification_unittest.cc
diff --git a/chrome/browser/chromeos/ui/low_disk_notification_unittest.cc b/chrome/browser/chromeos/ui/low_disk_notification_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..abe580ce748cbd757ffb31215dcf59fff1d7ecf6
--- /dev/null
+++ b/chrome/browser/chromeos/ui/low_disk_notification_unittest.cc
@@ -0,0 +1,106 @@
+// Copyright 2016 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/chromeos/ui/low_disk_notification.h"
+
+#include <stdint.h>
+
+#include "base/threading/platform_thread.h"
+#include "base/time/time.h"
+#include "chrome/grit/generated_resources.h"
+#include "chromeos/dbus/dbus_thread_manager.h"
+#include "chromeos/dbus/fake_cryptohome_client.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "ui/base/l10n/l10n_util.h"
+#include "ui/base/resource/resource_bundle.h"
+#include "ui/chromeos/resources/grit/ui_chromeos_resources.h"
+#include "ui/message_center/fake_message_center.h"
+#include "ui/message_center/message_center.h"
+
+namespace {
+
+// Copied from low_disk_notification.cc
+const uint64_t kMediumNotification = (1 << 30) - 1;
+const uint64_t kHighNotification = (512 << 20) - 1;
+
+}  // namespace
+
+namespace chromeos {
+
+class LowDiskNotificationTest : public testing::Test,
+                                public message_center::FakeMessageCenter {
+ public:
+  LowDiskNotificationTest() {}
+
+  void SetUp() override {
+    DBusThreadManager::GetSetterForTesting()->SetCryptohomeClient(
+        std::unique_ptr<CryptohomeClient>(new FakeCryptohomeClient));
+    message_center::MessageCenter::Initialize();
+    low_disk_notification_.reset(new LowDiskNotification());
+    low_disk_notification_->SetMessageCenterForTest(this);
+    low_disk_notification_->SetNotificationIntervalForTest(
+        base::TimeDelta::FromMilliseconds(10));
+    notification_count_ = 0;
+  }
+
+  void TearDown() override {
+    low_disk_notification_.reset();
+    last_notification_.reset();
+    message_center::MessageCenter::Shutdown();
+    DBusThreadManager::Shutdown();
+  }
+
+  void AddNotification(
+      std::unique_ptr<message_center::Notification> notification) override {
+    last_notification_ = std::move(notification);
+    notification_count_++;
+  }
+
+ protected:
+  std::unique_ptr<LowDiskNotification> low_disk_notification_;
+  std::unique_ptr<message_center::Notification> last_notification_;
+  int notification_count_;
+};
+
+TEST_F(LowDiskNotificationTest, MediumLevelNotification) {
+  base::string16 expected_title =
+      l10n_util::GetStringUTF16(IDS_LOW_DISK_NOTIFICATION_TITLE);
+  low_disk_notification_->OnLowDiskSpace(kMediumNotification);
+  EXPECT_NE(nullptr, last_notification_);
+  EXPECT_EQ(expected_title, last_notification_->title());
+  EXPECT_EQ(1, notification_count_);
+}
+
+TEST_F(LowDiskNotificationTest, HighLevelReplacesMedium) {
+  base::string16 expected_title =
+      l10n_util::GetStringUTF16(IDS_CRITICALLY_LOW_DISK_NOTIFICATION_TITLE);
+  low_disk_notification_->OnLowDiskSpace(kMediumNotification);
+  low_disk_notification_->OnLowDiskSpace(kHighNotification);
+  EXPECT_NE(nullptr, last_notification_);
+  EXPECT_EQ(expected_title, last_notification_->title());
+  EXPECT_EQ(2, notification_count_);
+}
+
+TEST_F(LowDiskNotificationTest, NotificationsAreThrottled) {
+  low_disk_notification_->OnLowDiskSpace(kHighNotification);
+  base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(5));
+  low_disk_notification_->OnLowDiskSpace(kHighNotification);
+  EXPECT_EQ(1, notification_count_);
+}
+
+TEST_F(LowDiskNotificationTest, HighNotificationsAreShownAfterThrottling) {
+  low_disk_notification_->OnLowDiskSpace(kHighNotification);
+  base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(15));
+  low_disk_notification_->OnLowDiskSpace(kHighNotification);
+  EXPECT_EQ(2, notification_count_);
+}
+
+TEST_F(LowDiskNotificationTest, MediumNotificationsAreNotShownAfterThrottling) {
+  low_disk_notification_->OnLowDiskSpace(kMediumNotification);
+  base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(15));
+  low_disk_notification_->OnLowDiskSpace(kMediumNotification);
+  EXPECT_EQ(1, notification_count_);
+}
+
+}  // namespace chromeos