Cocoa: Media gallery dialog unit test

chrome/browser/ui/cocoa/extensions/media_galleries_dialog_cocoa.h

 // 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.
 
 #ifndef CHROME_BROWSER_UI_COCOA_EXTENSIONS_MEDIA_GALLERIES_DIALOG_COCOA_H_
 #define CHROME_BROWSER_UI_COCOA_EXTENSIONS_MEDIA_GALLERIES_DIALOG_COCOA_H_
 
 #import <Cocoa/Cocoa.h>
 
+#include "base/gtest_prod_util.h"
 #include "chrome/browser/media_gallery/media_galleries_dialog_controller.h"
 #import "chrome/browser/ui/cocoa/constrained_window_mac.h"
 
 @class MediaGalleriesCocoaController;
 
 namespace chrome {
 
+class MediaGalleriesDialogTest;
+
 // This class displays an alert that can be used to grant permission for
 // extensions to access a gallery (media folders).
 class MediaGalleriesDialogCocoa :
     public ConstrainedWindowMacDelegateSystemSheet,
     public MediaGalleriesDialog {
  public:
   MediaGalleriesDialogCocoa(
       MediaGalleriesDialogController* controller,
       MediaGalleriesCocoaController* delegate);
   virtual ~MediaGalleriesDialogCocoa();
 
   // Called when the user clicks the Add Gallery button.
   void OnAddFolderClicked();
   // Called when the user toggles a gallery checkbox.
   void OnCheckboxToggled(NSButton* checkbox);
   // Called when the alert closes.
   void SheetDidEnd(NSInteger result);
 
   // MediaGalleriesDialog implementation:
   virtual void UpdateGallery(const MediaGalleryPrefInfo* gallery,
                              bool permitted) OVERRIDE;
 
   // ConstrainedWindowMacDelegateSystemSheet implementation.
   virtual void DeleteDelegate() OVERRIDE;
 
  private:
+  FRIEND_TEST_ALL_PREFIXES(MediaGalleriesDialogTest, InitializeCheckboxes);
+  FRIEND_TEST_ALL_PREFIXES(MediaGalleriesDialogTest, ToggleCheckboxes);
+  FRIEND_TEST_ALL_PREFIXES(MediaGalleriesDialogTest, UpdateAdds);
+
   void UpdateGalleryCheckbox(NSButton* checkbox,
                              const MediaGalleryPrefInfo* gallery,
                              bool permitted);
   void UpdateCheckboxContainerFrame();
 
   MediaGalleriesDialogController* controller_;  // weak
   ConstrainedWindow* window_;  // weak
 
   // True if the user has pressed accept.
   bool accepted_;
 
   // List of checkboxes ordered from bottom to top.
   scoped_nsobject<NSMutableArray> checkboxes_;
 
   // Container view for checkboxes.
   scoped_nsobject<NSView> checkbox_container_;
 
   // The alert that the dialog is being displayed as.
   scoped_nsobject<NSAlert> alert_;
 
   // An Objective-C class to route callbacks from Cocoa code.
   scoped_nsobject<MediaGalleriesCocoaController> cocoa_controller_;
 
   DISALLOW_COPY_AND_ASSIGN(MediaGalleriesDialogCocoa);
 };
 
 }  // namespace chrome
 
 #endif  // CHROME_BROWSER_UI_COCOA_EXTENSIONS_MEDIA_GALLERIES_DIALOG_COCOA_H_