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
11class MockMediaGalleriesDialog : public MediaGalleriesDialog {
12 public:
13  typedef base::Callback<void(int update_count)> DialogDestroyedCallback;
14
15  explicit MockMediaGalleriesDialog(const DialogDestroyedCallback& callback);
16  virtual ~MockMediaGalleriesDialog();
17
18  // MediaGalleriesDialog implementation.
19  virtual void UpdateGalleries() OVERRIDE;
20
21  // Number up times UpdateResults has been called.
22  int update_count() const;
23
24 private:
25  // MediaGalleriesDialog implementation.
26  virtual 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_
36