File association for app shims. (Mac)

chrome/browser/web_applications/web_app_mac_unittest.mm

 // 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 "chrome/browser/web_applications/web_app_mac.h"
 
 #import <Cocoa/Cocoa.h>
 #include <errno.h>
 #include <sys/xattr.h>
 
+#include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/mac/foundation_util.h"
 #include "base/mac/scoped_nsobject.h"
 #include "base/path_service.h"
 #include "base/strings/sys_string_conversions.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/common/chrome_paths.h"
+#include "chrome/common/chrome_switches.h"
 #import "chrome/common/mac/app_mode_common.h"
 #include "grit/theme_resources.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #import "testing/gtest_mac.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "ui/gfx/image/image.h"
 
 using ::testing::_;
 using ::testing::Return;
 using ::testing::NiceMock;
 
 namespace {
 
 const char kFakeChromeBundleId[] = "fake.cfbundleidentifier";
 
 class WebAppShortcutCreatorMock : public web_app::WebAppShortcutCreator {
  public:
-  explicit WebAppShortcutCreatorMock(
+  WebAppShortcutCreatorMock(
       const base::FilePath& app_data_dir,
       const ShellIntegration::ShortcutInfo& shortcut_info)
       : WebAppShortcutCreator(app_data_dir,
-                              shortcut_info) {
+                              shortcut_info,
+                              extensions::FileHandlersInfo(NULL)) {
+  }
+
+  WebAppShortcutCreatorMock(
+      const base::FilePath& app_data_dir,
+      const ShellIntegration::ShortcutInfo& shortcut_info,
+      const extensions::FileHandlersInfo& file_handlers_info)
+      : WebAppShortcutCreator(app_data_dir,
+                              shortcut_info,
+                              file_handlers_info) {
   }
 
   MOCK_CONST_METHOD0(GetApplicationsDirname, base::FilePath());
   MOCK_CONST_METHOD1(GetAppBundleById,
                      base::FilePath(const std::string& bundle_id));
   MOCK_CONST_METHOD0(RevealAppShimInFinder, void());
 
  private:
   DISALLOW_COPY_AND_ASSIGN(WebAppShortcutCreatorMock);
 };
@@ skipping 230 matching lines @@
   EXPECT_CALL(shortcut_creator, RevealAppShimInFinder())
       .Times(0);
   EXPECT_TRUE(shortcut_creator.CreateShortcuts(
       SHORTCUT_CREATION_AUTOMATED, ShellIntegration::ShortcutLocations()));
 
   EXPECT_CALL(shortcut_creator, RevealAppShimInFinder());
   EXPECT_TRUE(shortcut_creator.CreateShortcuts(
       SHORTCUT_CREATION_BY_USER, ShellIntegration::ShortcutLocations()));
 }
 
+TEST_F(WebAppShortcutCreatorTest, FileHandlers) {
+  CommandLine::ForCurrentProcess()->AppendSwitch(
+      switches::kEnableAppsFileAssociations);
+  std::vector<extensions::FileHandlerInfo> handlers;
+  extensions::FileHandlerInfo handler_0;
+  handler_0.extensions.insert("ext0");
+  handler_0.extensions.insert("ext1");
+  handler_0.types.insert("type0");
+  handler_0.types.insert("type1");
+  handlers.push_back(handler_0);
+  extensions::FileHandlerInfo handler_1;
+  handler_1.extensions.insert("ext2");
+  handler_1.types.insert("type2");
+  handlers.push_back(handler_1);
+
+  NiceMock<WebAppShortcutCreatorMock> shortcut_creator(
+      app_data_dir_, info_, extensions::FileHandlersInfo(&handlers));
+  EXPECT_CALL(shortcut_creator, GetApplicationsDirname())
+      .WillRepeatedly(Return(destination_dir_));
+  EXPECT_TRUE(shortcut_creator.CreateShortcuts(
+      SHORTCUT_CREATION_AUTOMATED, ShellIntegration::ShortcutLocations()));
+
+  base::FilePath plist_path =
+      shim_path_.Append("Contents").Append("Info.plist");
+  NSDictionary* plist = [NSDictionary dictionaryWithContentsOfFile:
+      base::mac::FilePathToNSString(plist_path)];
+  NSArray* file_handlers =
+      [plist objectForKey:app_mode::kCFBundleDocumentTypesKey];
+
+  NSDictionary* file_handler_0 = [file_handlers objectAtIndex:0];
+  EXPECT_NSEQ(app_mode::kBundleTypeRoleViewer,
+              [file_handler_0 objectForKey:app_mode::kCFBundleTypeRoleKey]);
+  NSArray* file_handler_0_extensions =
+      [file_handler_0 objectForKey:app_mode::kCFBundleTypeExtensionsKey];
+  EXPECT_TRUE([file_handler_0_extensions containsObject:@"ext0"]);
+  EXPECT_TRUE([file_handler_0_extensions containsObject:@"ext1"]);
+  NSArray* file_handler_0_types =
+      [file_handler_0 objectForKey:app_mode::kCFBundleTypeMIMETypesKey];
+  EXPECT_TRUE([file_handler_0_types containsObject:@"type0"]);
+  EXPECT_TRUE([file_handler_0_types containsObject:@"type1"]);
+
+  NSDictionary* file_handler_1 = [file_handlers objectAtIndex:1];
+  EXPECT_NSEQ(app_mode::kBundleTypeRoleViewer,
+              [file_handler_1 objectForKey:app_mode::kCFBundleTypeRoleKey]);
+  NSArray* file_handler_1_extensions =
+      [file_handler_1 objectForKey:app_mode::kCFBundleTypeExtensionsKey];
+  EXPECT_TRUE([file_handler_1_extensions containsObject:@"ext2"]);
+  NSArray* file_handler_1_types =
+      [file_handler_1 objectForKey:app_mode::kCFBundleTypeMIMETypesKey];
+  EXPECT_TRUE([file_handler_1_types containsObject:@"type2"]);
+}
+
 }  // namespace web_app