[Mac] Touch Bar Support for Dialogs

chrome/browser/ui/cocoa/bookmarks/bookmark_editor_base_controller_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 <Cocoa/Cocoa.h>
 
+#include "base/mac/mac_util.h"
 #include "base/strings/sys_string_conversions.h"
 #include "base/strings/utf_string_conversions.h"
+#include "base/test/scoped_feature_list.h"
 #include "chrome/browser/bookmarks/bookmark_model_factory.h"
 #import "chrome/browser/ui/cocoa/bookmarks/bookmark_editor_controller.h"
 #include "chrome/browser/ui/cocoa/test/cocoa_profile_test.h"
+#include "chrome/common/chrome_features.h"
+#include "chrome/grit/generated_resources.h"
 #include "chrome/test/base/test_browser_window.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/bookmarks/browser/bookmark_model.h"
 #include "components/strings/grit/components_strings.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #import "testing/gtest_mac.h"
 #include "testing/platform_test.h"
+#import "ui/base/cocoa/touch_bar_forward_declarations.h"
+#import "ui/base/cocoa/touch_bar_util.h"
 #include "ui/base/l10n/l10n_util_mac.h"
 
 using base::ASCIIToUTF16;
 using bookmarks::BookmarkExpandedStateTracker;
 using bookmarks::BookmarkModel;
 using bookmarks::BookmarkNode;
 
+namespace {
+
+// Touch bar identifier.
+NSString* const kBookmarkEditDialogTouchBarId = @"bookmark-edit-dialog";
+
+// Touch bar item identifiers.
+NSString* const kNewFolderTouchBarId = @"NEW-FOLDER";
+NSString* const kCancelTouchBarId = @"CANCEL";
+NSString* const kSaveTouchBarId = @"SAVE";
+
+}  // end namespace
+
 class BookmarkEditorBaseControllerTest : public CocoaProfileTest {
  public:
   BookmarkEditorBaseController* controller_;  // weak
   const BookmarkNode* folder_a_;
   const BookmarkNode* folder_b_;
   const BookmarkNode* folder_b_0_;
   const BookmarkNode* folder_b_3_;
   const BookmarkNode* folder_c_;
 
   void CreateModel() {
@@ skipping 219 matching lines @@
 
   [controller_ expandNodes:nodes];
 
   actual = [controller_ getExpandedNodes];
 
   EXPECT_EQ(nodes, actual);
 
   [controller_ cancel:nil];
 }
 
+// Verifies the dialog's touch bar.
+TEST_F(BookmarkEditorBaseControllerTest, TouchBar) {
+  if (!base::mac::IsAtLeastOS10_12()) {
+    [controller_ cancel:nil];
+    return;
+  }
+
+  base::test::ScopedFeatureList feature_list;
+  feature_list.InitAndEnableFeature(features::kBrowserTouchBar);
+
+  NSTouchBar* touch_bar = [controller_ makeTouchBar];
+  NSArray* touch_bar_items = [touch_bar itemIdentifiers];
+  EXPECT_TRUE([touch_bar_items
+      containsObject:ui::GetTouchBarItemId(kBookmarkEditDialogTouchBarId,
+                                           kNewFolderTouchBarId)]);
+  EXPECT_TRUE([touch_bar_items
+      containsObject:ui::GetTouchBarItemId(kBookmarkEditDialogTouchBarId,
+                                           kCancelTouchBarId)]);
+  EXPECT_TRUE([touch_bar_items
+      containsObject:ui::GetTouchBarItemId(kBookmarkEditDialogTouchBarId,
+                                           kSaveTouchBarId)]);
+
+  [controller_ cancel:nil];
+}
+
 class BookmarkFolderInfoTest : public CocoaTest { };
 
 TEST_F(BookmarkFolderInfoTest, Construction) {
   NSMutableArray* children = [NSMutableArray arrayWithObject:@"child"];
   // We just need a pointer, and any pointer will do.
   const BookmarkNode* fakeNode =
       reinterpret_cast<const BookmarkNode*>(&children);
   BookmarkFolderInfo* info =
     [BookmarkFolderInfo bookmarkFolderInfoWithFolderName:@"name"
                                               folderNode:fakeNode
                                                 children:children];
   EXPECT_TRUE(info);
   EXPECT_EQ([info folderName], @"name");
   EXPECT_EQ([info children], children);
   EXPECT_EQ([info folderNode], fakeNode);
 }