1 files changed, 48 insertions, 0 deletions
diff --git a/chrome/browser/extensions/api/notifications/notifications_api_unittest.cc b/chrome/browser/extensions/api/notifications/notifications_api_unittest.cc
new file mode 100644
index 0000000..b1c32acf
--- /dev/null
+++ b/chrome/browser/extensions/api/notifications/notifications_api_unittest.cc
@@ -0,0 +1,48 @@
+// Copyright (c) 2013 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/extensions/api/notifications/notifications_api.h"
+#include "chrome/browser/extensions/extension_function.h"
+#include "chrome/browser/extensions/extension_function_test_utils.h"
+#include "chrome/common/extensions/features/feature.h"
+#include "chrome/test/base/browser_with_test_window_test.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace utils = extension_function_test_utils;
+
+namespace extensions {
+
+class ExtensionNotificationsTest : public BrowserWithTestWindowTest {
+};
+
+TEST_F(ExtensionNotificationsTest, Channels) {
+  scoped_refptr<Extension> extension(utils::CreateEmptyExtensionWithLocation(
+        extensions::Manifest::UNPACKED));
+  scoped_refptr<NotificationsClearFunction> notification_function(
+      new extensions::NotificationsClearFunction());
+  notification_function->set_extension(extension.get());
+  {
+    Feature::ScopedCurrentChannel channel_scope(
+          chrome::VersionInfo::CHANNEL_CANARY);
+    ASSERT_TRUE(notification_function->IsNotificationsApiAvailable());
+  }
+  {
+    Feature::ScopedCurrentChannel channel_scope(
+          chrome::VersionInfo::CHANNEL_DEV);
+    ASSERT_TRUE(notification_function->IsNotificationsApiAvailable());
+  }
+  {
+    Feature::ScopedCurrentChannel channel_scope(
+          chrome::VersionInfo::CHANNEL_BETA);
+    ASSERT_FALSE(notification_function->IsNotificationsApiAvailable());
+  }
+  {
+    Feature::ScopedCurrentChannel channel_scope(
+          chrome::VersionInfo::CHANNEL_STABLE);
+    ASSERT_FALSE(notification_function->IsNotificationsApiAvailable());
+  }
+}
+
+}  // namespace extensions