ui/web_dialogs/test/test_web_dialog_delegate.h

   1 // Copyright (c) 2012 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 UI_WEB_DIALOGS_TEST_TEST_WEB_DIALOG_DELEGATE_H_
   6 #define UI_WEB_DIALOGS_TEST_TEST_WEB_DIALOG_DELEGATE_H_
   7
   8 #include <string>
   9
  10 #include "base/compiler_specific.h"
  11 #include "ui/gfx/size.h"
  12 #include "ui/web_dialogs/web_dialog_delegate.h"
  13 #include "url/gurl.h"
  14
  15 namespace ui {
  16 namespace test {
  17
  18 class TestWebDialogDelegate : public WebDialogDelegate {
  19  public:
  20   explicit TestWebDialogDelegate(const GURL& url);
  21   virtual ~TestWebDialogDelegate();
  22
  23   void set_size(int width, int height) {
  24     size_.SetSize(width, height);
  25   }
  26
  27   // WebDialogDelegate implementation:
  28   virtual ModalType GetDialogModalType() const OVERRIDE;
  29   virtual base::string16 GetDialogTitle() const OVERRIDE;
  30   virtual GURL GetDialogContentURL() const OVERRIDE;
  31   virtual void GetWebUIMessageHandlers(
  32       std::vector<content::WebUIMessageHandler*>* handlers) const OVERRIDE;
  33   virtual void GetDialogSize(gfx::Size* size) const OVERRIDE;
  34   virtual std::string GetDialogArgs() const OVERRIDE;
  35   virtual void OnDialogClosed(const std::string& json_retval) OVERRIDE;
  36   virtual void OnCloseContents(content::WebContents* source,
  37                                bool* out_close_dialog) OVERRIDE;
  38   virtual bool ShouldShowDialogTitle() const OVERRIDE;
  39
  40  protected:
  41   const GURL url_;
  42   gfx::Size size_;
  43
  44   DISALLOW_COPY_AND_ASSIGN(TestWebDialogDelegate);
  45 };
  46
  47 }  // namespace test
  48 }  // namespace ui
  49
  50 #endif  // UI_WEB_DIALOGS_TEST_TEST_WEB_DIALOG_DELEGATE_H_