Convert utility process WiFi Credentials IPC to mojo

content/browser/utility_process_mojo_client_browsertest.cc

 // Copyright 2016 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 "content/public/browser/utility_process_mojo_client.h"
 
 #include <memory>
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/command_line.h"
 #include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/test/content_browser_test.h"
 #include "content/public/test/test_service.mojom.h"
 
 namespace content {
 
 // Test fixture used to make different Mojo calls to the utility process.
 class UtilityProcessMojoClientBrowserTest : public ContentBrowserTest {
  public:
-  void StartMojoService(bool disable_sandbox) {
+  void StartMojoService(bool disable_sandbox, bool run_elevated = false) {

[Sam McNally, 2017/01/16 04:16:20]

This would be nicer as an enum {SANDBOXED, UNSANDBOXED, ELEVATED}.

[Noel Gordon, 2017/01/16 11:09:58]

Acknowledged.

[Noel Gordon, 2017/01/18 06:02:03]

... but somewhat unrelated to this change: maybe best done on another patch as
we discussed.

     mojo_client_.reset(new UtilityProcessMojoClient<mojom::TestService>(
         base::ASCIIToUTF16("TestMojoProcess")));
 
     mojo_client_->set_error_callback(
         base::Bind(&UtilityProcessMojoClientBrowserTest::OnConnectionError,
                    base::Unretained(this)));
 
     // This test case needs to have the sandbox disabled.
     if (disable_sandbox)
       mojo_client_->set_disable_sandbox();
+#if defined(OS_WIN)
+    // This test case needs utility process privilege elevation.
+    if (run_elevated) {
+      CHECK(disable_sandbox);
+      mojo_client_->set_run_elevated();
+    }
+#endif  // defined(OS_WIN)
+
     mojo_client_->Start();
   }
 
   // Called when a response is received from a call to DoSomething() or
   // DoTerminateProcess().
   void OnResponseReceived() {
     response_received_ = true;
     done_closure_.Run();
   }
 
@@ skipping 85 matching lines @@
 
   mojo_client_->service()->CreateFolder(
       base::Bind(&UtilityProcessMojoClientBrowserTest::OnCreateFolderFinished,
                  base::Unretained(this)));
 
   run_loop.Run();
   EXPECT_TRUE(response_received_);
   EXPECT_TRUE(sandbox_succeeded_);
   EXPECT_FALSE(error_happened_);
 }
-#endif
+#endif  // !defined(OS_ANDROID)
+
+#if defined(OS_WIN)
+// Call a function that succeeds with process elevation. Elevation is only
+// available on WIN, and is used to permit UAC prompts for operations that
+// need user approval before proceeding.
+IN_PROC_BROWSER_TEST_F(UtilityProcessMojoClientBrowserTest, ElevatedSuccess) {
+  base::RunLoop run_loop;
+  done_closure_ = run_loop.QuitClosure();
+
+  bool elevated_utility_process = true;
+  StartMojoService(true, elevated_utility_process);
+
+  mojo_client_->service()->CreateFolder(
+      base::Bind(&UtilityProcessMojoClientBrowserTest::OnCreateFolderFinished,
+                 base::Unretained(this)));
+
+  run_loop.Run();
+  EXPECT_TRUE(response_received_);
+  EXPECT_TRUE(sandbox_succeeded_);
+  EXPECT_FALSE(error_happened_);
+}
+#endif  // defined(OS_WIN)
 
 }  // namespace content