1 files changed, 48 insertions, 0 deletions
diff --git a/chrome/browser/ui/webui/test_web_dialog_delegate.h b/chrome/browser/ui/webui/test_web_dialog_delegate.h
new file mode 100644
index 0000000..4e92b03
--- /dev/null
+++ b/chrome/browser/ui/webui/test_web_dialog_delegate.h
@@ -0,0 +1,48 @@
+// 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_WEBUI_TEST_WEB_DIALOG_DELEGATE_H_
+#define CHROME_BROWSER_UI_WEBUI_TEST_WEB_DIALOG_DELEGATE_H_
+#pragma once
+
+#include <string>
+
+#include "base/compiler_specific.h"
+#include "chrome/browser/ui/webui/web_dialog_ui.h"
+#include "ui/gfx/size.h"
+
+namespace test {
+
+class TestWebDialogDelegate : public WebDialogDelegate {
+ public:
+  explicit TestWebDialogDelegate(const GURL& url);
+  virtual ~TestWebDialogDelegate();
+
+  void set_size(int width, int height) {
+    size_.SetSize(width, height);
+  }
+
+  // WebDialogDelegate implementation:
+  virtual ui::ModalType GetDialogModalType() const OVERRIDE;
+  virtual string16 GetDialogTitle() const OVERRIDE;
+  virtual GURL GetDialogContentURL() const OVERRIDE;
+  virtual void GetWebUIMessageHandlers(
+      std::vector<content::WebUIMessageHandler*>* handlers) const OVERRIDE;
+  virtual void GetDialogSize(gfx::Size* size) const OVERRIDE;
+  virtual std::string GetDialogArgs() const OVERRIDE;
+  virtual void OnDialogClosed(const std::string& json_retval) OVERRIDE;
+  virtual void OnCloseContents(content::WebContents* source,
+                               bool* out_close_dialog) OVERRIDE;
+  virtual bool ShouldShowDialogTitle() const OVERRIDE;
+
+ protected:
+  const GURL url_;
+  gfx::Size size_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestWebDialogDelegate);
+};
+
+}  // namespace test
+
+#endif  // CHROME_BROWSER_UI_WEBUI_TEST_WEB_DIALOG_DELEGATE_H_