Cocoa: Media gallery dialog unit test

chrome/browser/ui/cocoa/extensions/media_galleries_dialog_cocoa.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.
 
 #include "chrome/browser/ui/cocoa/extensions/media_galleries_dialog_cocoa.h"
 
 #include "base/sys_string_conversions.h"
 #include "grit/generated_resources.h"
 #include "ui/base/l10n/l10n_util.h"
 
@@ skipping 75 matching lines @@
   UpdateCheckboxContainerFrame();
   [alert_ setAccessoryView:checkbox_container_];
 
   // As a safeguard against the user skipping reading over the dialog and just
   // confirming, the button will be unavailable for dialogs without any checks
   // until the user toggles something.
   [[[alert_ buttons] objectAtIndex:0] setEnabled:
       controller_->HasPermittedGalleries()];
 
   set_sheet(alert_);
-  window_ = new ConstrainedWindowMac(controller->tab_contents(), this);
+  // May be NULL during tests.
+  if (controller->tab_contents())
+    window_ = new ConstrainedWindowMac(controller->tab_contents(), this);
 }
 
 MediaGalleriesDialogCocoa::~MediaGalleriesDialogCocoa() {
 }
 
 void MediaGalleriesDialogCocoa::OnAddFolderClicked() {
   controller_->OnAddFolderClicked();
 }
 
 void MediaGalleriesDialogCocoa::OnCheckboxToggled(NSButton* checkbox) {
@@ skipping 10 matching lines @@
       controller_->DidToggleGallery(gallery, [checkbox state] == NSOnState);
       break;
     }
   }
 }
 
 void MediaGalleriesDialogCocoa::SheetDidEnd(NSInteger result) {
   switch (result) {
     case NSAlertFirstButtonReturn:
       accepted_ = true;
-      window_->CloseConstrainedWindow();
+      if (window_)
+        window_->CloseConstrainedWindow();
       break;
     case NSAlertSecondButtonReturn:
-      window_->CloseConstrainedWindow();
+      if (window_)
+        window_->CloseConstrainedWindow();
       break;
     default:
       NOTREACHED();
       break;
   }
 }
 
 void MediaGalleriesDialogCocoa::UpdateGalleryCheckbox(
     NSButton* checkbox,
     const MediaGalleryPrefInfo* gallery,
@@ skipping 66 matching lines @@
 
 // static
 MediaGalleriesDialog* MediaGalleriesDialog::Create(
     MediaGalleriesDialogController* controller) {
   scoped_nsobject<MediaGalleriesCocoaController> cocoa_controller(
       [[MediaGalleriesCocoaController alloc] init]);
   return new MediaGalleriesDialogCocoa(controller, cocoa_controller);
 }
 
 }  // namespace chrome