Add native file picker impl using SelectFileDialog.

chrome/browser/intents/native_services_browsertest.cc

+// 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 <algorithm>
+#include <iterator>
+#include <string>
+
+#include "base/command_line.h"
+#include "base/file_path.h"
+#include "base/string_util.h"
+#include "base/utf_string_conversions.h"
+#include "base/memory/scoped_ptr.h"
+#include "chrome/browser/intents/intent_service_host.h"
+#include "chrome/browser/intents/native_services.h"
+#include "chrome/browser/intents/web_intents_util.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/browser_tabstrip.h"
+#include "chrome/common/chrome_switches.h"
+#include "chrome/test/base/in_process_browser_test.h"
+#include "chrome/test/base/ui_test_utils.h"
+#include "content/public/browser/web_contents.h"
+#include "content/public/browser/web_intents_dispatcher.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "ui/base/dialogs/select_file_dialog_factory.h"
+#include "webkit/glue/web_intent_data.h"
+#include "webkit/glue/web_intent_service_data.h"
+
+namespace {
+
+const std::string kPoodlePath = "/home/poodles/skippy.png";
+const int kUnsetReplyType = -1;
+const string16 kUnsetReplyValue = ASCIIToUTF16("-1");
+bool picker_success_mode = true;
+
+class TestIntentsDispatcher : public content::WebIntentsDispatcher {
+ public:
+  explicit TestIntentsDispatcher(const webkit_glue::WebIntentData& intent)
+      : intent_(intent),
+        reply_value_(kUnsetReplyType),
+        reply_string_(kUnsetReplyValue) {}
+
+  virtual const webkit_glue::WebIntentData& GetIntent() OVERRIDE {
+    return intent_;
+  }
+
+  virtual void DispatchIntent(content::WebContents* web_contents) OVERRIDE {}
+  virtual void ResetDispatch() OVERRIDE {}
+
+  virtual void SendReplyMessage(
+      webkit_glue::WebIntentReplyType reply_type,
+      const string16& data) OVERRIDE {
+    reply_value_ = reply_type;
+    reply_string_ = data;
+  }
+
+  virtual void RegisterReplyNotification(
+      const base::Callback<void(webkit_glue::WebIntentReplyType)>&) OVERRIDE {
+  }
+
+  webkit_glue::WebIntentData intent_;
+
+  int reply_value_;
+  string16 reply_string_;
+};
+
+// Stub SelectFileDialog designed for testing.
+class TestSelectFileDialog : public ui::SelectFileDialog {
+ public:
+  // Main factory method which returns correct type.
+  static ui::SelectFileDialog* Create(
+      Listener* listener,
+      ui::SelectFilePolicy* policy) {
+    return new TestSelectFileDialog(listener, policy);
+  }
+
+  // BaseShellDialog implementation.
+  virtual bool IsRunning(gfx::NativeWindow parent_window) const OVERRIDE {
+    return false;
+  }
+  virtual void ListenerDestroyed() OVERRIDE {}
+  virtual bool HasMultipleFileTypeChoicesImpl() OVERRIDE {
+    return false;
+  }
+
+ protected:
+  TestSelectFileDialog(
+      Listener* listener, ui::SelectFilePolicy* policy)
+      : ui::SelectFileDialog(listener, policy),
+        listener_(listener),
+        policy_(policy) {
+  }
+  virtual ~TestSelectFileDialog() {}
+
+  // SelectFileDialog implementation.
+  // |params| is user data we pass back via the Listener interface.
+  virtual void SelectFileImpl(
+      Type type,
+      const string16& title,
+      const FilePath& default_path,
+      const FileTypeInfo* file_types,
+      int file_type_index,
+      const FilePath::StringType& default_extension,
+      gfx::NativeWindow owning_window,
+      void* params) OVERRIDE {
+    if (picker_success_mode)
+      listener_->FileSelected(
+          FilePath::FromUTF8Unsafe(kPoodlePath), 0, NULL);
+    else
+      listener_->FileSelectionCanceled(NULL);
+  }
+
+ private:
+  Listener* listener_;
+  ui::SelectFilePolicy* policy_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestSelectFileDialog);
+};
+
+class TestSelectFileDialogFactory : public ui::SelectFileDialogFactory {
+ public:
+  virtual ui::SelectFileDialog* Create(
+      ui::SelectFileDialog::Listener* listener,
+      ui::SelectFilePolicy* policy) {
+    return TestSelectFileDialog::Create(listener, policy);
+  }
+};
+
+}  // namespace
+
+class NativeServicesBrowserTest : public InProcessBrowserTest {
+ protected:
+  NativeServicesBrowserTest() {}
+
+  virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
+    // We start the test server now instead of in
+    // SetUpInProcessBrowserTestFixture so that we can get its port number.
+    ASSERT_TRUE(test_server()->Start());
+
+    command_line->AppendSwitch(switches::kWebIntentsNativeServicesEnabled);
+  }
+
+  virtual void SetUpOnMainThread() OVERRIDE {
+    factory_.reset(new TestSelectFileDialogFactory());
+    ui::SelectFileDialog::SetFactory(factory_.get());
+  }
+
+  virtual Browser* GetBrowser() { return browser(); }
+  scoped_ptr<TestSelectFileDialogFactory> factory_;
+};
+
+IN_PROC_BROWSER_TEST_F(NativeServicesBrowserTest, PickFileSelected) {
+  content::WebContents* tab = chrome::GetActiveWebContents(GetBrowser());
+
+  webkit_glue::WebIntentData intent;
+  intent.action = ASCIIToUTF16(web_intents::kActionPick);
+  intent.type = ASCIIToUTF16("image/*");
+
+  TestIntentsDispatcher dispatcher(intent);
+  web_intents::NativeServiceFactory factory;
+  GURL url(web_intents::kNativeFilePickerUrl);
+  scoped_ptr<web_intents::IntentServiceHost> service(
+      factory.CreateServiceInstance(url, intent, tab));
+  service->HandleIntent(&dispatcher);
+  EXPECT_EQ(
+      webkit_glue::WEB_INTENT_REPLY_SUCCESS, dispatcher.reply_value_);
+  // data should be set to something...
+  EXPECT_TRUE(EqualsASCII(dispatcher.reply_string_, kPoodlePath));
+}
+
+IN_PROC_BROWSER_TEST_F(NativeServicesBrowserTest, PickFileCancelled) {
+  picker_success_mode = false;
+  content::WebContents* tab = chrome::GetActiveWebContents(GetBrowser());
+
+  webkit_glue::WebIntentData intent;
+  intent.action = ASCIIToUTF16(web_intents::kActionPick);
+  intent.type = ASCIIToUTF16("image/*");
+
+  TestIntentsDispatcher dispatcher(intent);
+  web_intents::NativeServiceFactory factory;
+  GURL url(web_intents::kNativeFilePickerUrl);
+  scoped_ptr<web_intents::IntentServiceHost> service(
+      factory.CreateServiceInstance(url, intent, tab));
+  service->HandleIntent(&dispatcher);
+  EXPECT_EQ(
+      webkit_glue::WEB_INTENT_REPLY_FAILURE, dispatcher.reply_value_);
+}