src/chrome/browser/media_galleries/media_galleries_dialog_controller_test_util.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_MEDIA_GALLERIES_MEDIA_GALLERIES_DIALOG_CONTROLLER_TEST_UTIL_H_
   6 #define CHROME_BROWSER_MEDIA_GALLERIES_MEDIA_GALLERIES_DIALOG_CONTROLLER_TEST_UTIL_H_
   7
   8 #include "base/callback.h"
   9 #include "chrome/browser/media_galleries/media_galleries_dialog_controller.h"
  10
  11 class MockMediaGalleriesDialog : public MediaGalleriesDialog {
  12  public:
  13   typedef base::Callback<void(int update_count)> DialogDestroyedCallback;
  14
  15   explicit MockMediaGalleriesDialog(const DialogDestroyedCallback& callback);
  16   ~MockMediaGalleriesDialog() override;
  17
  18   // MediaGalleriesDialog implementation.
  19   void UpdateGalleries() override;
  20
  21   // Number up times UpdateResults has been called.
  22   int update_count() const;
  23
  24  private:
  25   // MediaGalleriesDialog implementation.
  26   void AcceptDialogForTesting() override;
  27
  28   int update_count_;
  29
  30   DialogDestroyedCallback dialog_destroyed_callback_;
  31
  32   DISALLOW_COPY_AND_ASSIGN(MockMediaGalleriesDialog);
  33 };
  34
  35 #endif  // CHROME_BROWSER_MEDIA_GALLERIES_MEDIA_GALLERIES_DIALOG_CONTROLLER_TEST_UTIL_H_