arc: Add autotest method to return ARC provisioning state.

This allows not to use expensive and fragile set of adb commands and
returns ARC provisioning state as it appears in Chrome.

TEST=in context of crrev.com/i/717828
BUG=b:119349367

Change-Id: I3582fcca3e36a084dbe8e0ff3522930504ebeb85
Reviewed-on: https://chromium-review.googlesource.com/c/1340797Reviewed-by: Ken Rockot <rockot@google.com>
Reviewed-by: Steven Bennetts <stevenjb@chromium.org>
Commit-Queue: Yury Khmel <khmel@chromium.org>
Cr-Commit-Position: refs/heads/master@{#608980}

chrome/browser/chromeos/extensions/autotest_private/autotest_private_api.cc

@@ -683,6 +683,20 @@ ExtensionFunction::ResponseAction AutotestPrivateIsAppShownFunction::Run() {
       OneArgument(std::make_unique<base::Value>(window_attached)));
 }
 
+///////////////////////////////////////////////////////////////////////////////
+// AutotestPrivateIsAppShownFunction
+///////////////////////////////////////////////////////////////////////////////
+
+AutotestPrivateIsArcProvisionedFunction::
+    ~AutotestPrivateIsArcProvisionedFunction() = default;
+
+ExtensionFunction::ResponseAction
+AutotestPrivateIsArcProvisionedFunction::Run() {
+  DVLOG(1) << "AutotestPrivateIsArcProvisionedFunction";
+  return RespondNow(OneArgument(std::make_unique<base::Value>(
+      arc::IsArcProvisioned(Profile::FromBrowserContext(browser_context())))));
+}
+
 ///////////////////////////////////////////////////////////////////////////////
 // AutotestPrivateLaunchAppFunction
 ///////////////////////////////////////////////////////////////////////////////

chrome/browser/chromeos/extensions/autotest_private/autotest_private_api.h

@@ -256,6 +256,17 @@ class AutotestPrivateIsAppShownFunction : public UIThreadExtensionFunction {
   ResponseAction Run() override;
 };
 
+class AutotestPrivateIsArcProvisionedFunction
+    : public UIThreadExtensionFunction {
+ public:
+  DECLARE_EXTENSION_FUNCTION("autotestPrivate.isArcProvisioned",
+                             AUTOTESTPRIVATE_ISARCPROVISIONED)
+
+ private:
+  ~AutotestPrivateIsArcProvisionedFunction() override;
+  ResponseAction Run() override;
+};
+
 class AutotestPrivateLaunchAppFunction : public UIThreadExtensionFunction {
  public:
   DECLARE_EXTENSION_FUNCTION("autotestPrivate.launchApp",

chrome/browser/chromeos/extensions/autotest_private/autotest_private_apitest.cc

@@ -8,6 +8,7 @@
 #include "chrome/browser/chromeos/arc/arc_session_manager.h"
 #include "chrome/browser/chromeos/extensions/autotest_private/autotest_private_api.h"
 #include "chrome/browser/extensions/extension_apitest.h"
+#include "components/arc/arc_prefs.h"
 #include "components/arc/arc_util.h"
 
 namespace extensions {
@@ -37,7 +38,20 @@ IN_PROC_BROWSER_TEST_F(AutotestPrivateApiTest, AutotestPrivate) {
   AutotestPrivateAPI::GetFactoryInstance()
       ->Get(browser()->profile())
       ->set_test_mode(true);
-  ASSERT_TRUE(RunComponentExtensionTest("autotest_private")) << message_;
+  ASSERT_TRUE(RunComponentExtensionTestWithArg("autotest_private", "default"))
+      << message_;
+}
+
+IN_PROC_BROWSER_TEST_F(AutotestPrivateApiTest, AutotestPrivateArcProvisioned) {
+  // Turn on testing mode so we don't kill the browser.
+  AutotestPrivateAPI::GetFactoryInstance()
+      ->Get(browser()->profile())
+      ->set_test_mode(true);
+  // Provisioning is completed.
+  browser()->profile()->GetPrefs()->SetBoolean(arc::prefs::kArcSignedIn, true);
+  ASSERT_TRUE(
+      RunComponentExtensionTestWithArg("autotest_private", "arcProvisioned"))
+      << message_;
 }
 
 }  // namespace extensions

chrome/browser/extensions/extension_apitest.cc

@@ -231,6 +231,13 @@ bool ExtensionApiTest::RunComponentExtensionTest(
                               kFlagEnableFileAccess | kFlagLoadAsComponent);
 }
 
+bool ExtensionApiTest::RunComponentExtensionTestWithArg(
+    const std::string& extension_name,
+    const char* custom_arg) {
+  return RunExtensionTestImpl(extension_name, std::string(), custom_arg,
+                              kFlagEnableFileAccess | kFlagLoadAsComponent);
+}
+
 bool ExtensionApiTest::RunExtensionTestNoFileAccess(
     const std::string& extension_name) {
   return RunExtensionTestImpl(extension_name, std::string(), NULL, kFlagNone);

chrome/browser/extensions/extension_apitest.h

@@ -98,6 +98,10 @@ class ExtensionApiTest : public ExtensionBrowserTest {
   // Same as RunExtensionTest, but loads extension as component.
   bool RunComponentExtensionTest(const std::string& extension_name);
 
+  // Same as RunComponentExtensionTest, but provides extra arg.
+  bool RunComponentExtensionTestWithArg(const std::string& extension_name,
+                                        const char* custom_arg);
+
   // Same as RunExtensionTest, but disables file access.
   bool RunExtensionTestNoFileAccess(const std::string& extension_name);
 

chrome/common/extensions/api/autotest_private.idl

@@ -104,6 +104,8 @@ namespace autotestPrivate {
 
   callback IsAppShownCallback = void (boolean appShown);
 
+  callback IsArcProvisionedCallback = void (boolean arcProvisioned);
+
   callback TakeScreenshotCallback = void (DOMString base64Png);
 
   callback VoidCallback = void ();
@@ -175,6 +177,9 @@ namespace autotestPrivate {
     // Returns true if requested app is shown in Chrome.
     static void isAppShown(DOMString appId, IsAppShownCallback callback);
 
+    // Returns true if ARC is provisioned.
+    static void isArcProvisioned(IsArcProvisionedCallback callback);
+
     // Launches an application from the launcher with the given appId.
     static void launchApp(DOMString appId, VoidCallback callback);
 

chrome/test/data/extensions/api_test/autotest_private/test.js

@@ -2,7 +2,7 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-chrome.test.runTests([
+var defaultTests = [
   // logout/restart/shutdown don't do anything as we don't want to kill the
   // browser with these tests.
   function logout() {
@@ -212,4 +212,41 @@ chrome.test.runTests([
         chrome.test.callbackFail(
             'Assistant is not available for the current user'));
   },
-]);
+  // This test verifies that ARC is not provisioned by default.
+  function isArcProvisioned() {
+    chrome.autotestPrivate.isArcProvisioned(
+        function(arcProvisioned) {
+          chrome.test.assertFalse(arcProvisioned);
+          chrome.test.assertNoLastError();
+          chrome.test.succeed();
+        });
+  },
+];
+
+var arcProvisionedTests = [
+  // This test verifies that isArcProvisioned returns True in case ARC
+  // provisiong is done.
+  function isArcProvisioned() {
+    chrome.autotestPrivate.isArcProvisioned(
+        function(arcProvisioned) {
+          chrome.test.assertTrue(arcProvisioned);
+          chrome.test.assertNoLastError();
+          chrome.test.succeed();
+        });
+  },
+];
+
+var test_suites = {
+  'default': defaultTests,
+  'arcProvisioned': arcProvisionedTests
+};
+
+chrome.test.getConfig(function(config) {
+  var suite = test_suites[config.customArg];
+  if (config.customArg in test_suites) {
+    chrome.test.runTests(test_suites[config.customArg]);
+  } else {
+    chrome.test.fail('Invalid test suite');
+  }
+});
+

extensions/browser/extension_function_histogram_value.h

@@ -1353,6 +1353,7 @@ enum HistogramValue {
   TABS_GOBACK = 1290,
   BRAILLEDISPLAYPRIVATE_UPDATEBLUETOOTHBRAILLEDISPLAYADDRESS = 1291,
   AUTOTESTPRIVATE_SETASSISTANTENABLED = 1292,
+  AUTOTESTPRIVATE_ISARCPROVISIONED = 1293,
   // Last entry: Add new entries above, then run:
   // python tools/metrics/histograms/update_extension_histograms.py
   ENUM_BOUNDARY

tools/metrics/histograms/enums.xml

@@ -17276,6 +17276,7 @@ Called by update_net_error_codes.py.-->
   <int value="1291"
       label="BRAILLEDISPLAYPRIVATE_UPDATEBLUETOOTHBRAILLEDISPLAYADDRESS"/>
   <int value="1292" label="AUTOTESTPRIVATE_SETASSISTANTENABLED"/>
+  <int value="1293" label="AUTOTESTPRIVATE_ISARCPROVISIONED"/>
 </enum>
 
 <enum name="ExtensionIconState">