CWS Private API - isPendingCustodianApproval

chrome/browser/extensions/extension_service_sync_unittest.cc

Index: chrome/browser/extensions/extension_service_sync_unittest.cc
diff --git a/chrome/browser/extensions/extension_service_sync_unittest.cc b/chrome/browser/extensions/extension_service_sync_unittest.cc
index 2283501adf76ef1d6bc11b40d5eb644746e749ab..3af559e4a1c6e089ded8554c6c04da2f0c795f56 100644
--- a/chrome/browser/extensions/extension_service_sync_unittest.cc
+++ b/chrome/browser/extensions/extension_service_sync_unittest.cc
@@ -18,6 +18,7 @@
 #include "base/memory/weak_ptr.h"
 #include "base/metrics/field_trial.h"
 #include "base/test/mock_entropy_provider.h"
+#include "chrome/browser/extensions/api/webstore_private/webstore_private_api.h"
 #include "chrome/browser/extensions/component_loader.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_service_test_with_install.h"
@@ -33,6 +34,7 @@
 #include "components/browser_sync/browser/profile_sync_service.h"
 #include "components/crx_file/id_util.h"
 #include "components/variations/variations_associated_data.h"
+#include "extensions/browser/api_test_utils.h"
 #include "extensions/browser/app_sorting.h"
 #include "extensions/browser/extension_prefs.h"
 #include "extensions/browser/extension_registry.h"
@@ -58,6 +60,7 @@
 #include "chrome/browser/supervised_user/supervised_user_service_factory.h"
 #endif
 
+using extensions::api_test_utils::RunFunctionAndReturnSingleResult;
 using extensions::AppSorting;
 using extensions::Extension;
 using extensions::ExtensionPrefs;
@@ -1582,6 +1585,20 @@ class ExtensionServiceTestSupervised : public ExtensionServiceSyncTest,
     base::FeatureList::SetInstance(std::move(feature_list));
   }
 
+  bool IsExtensionPendingCustodianApproval(const std::string& extension_id) {
+    scoped_refptr<
+        extensions::WebstorePrivateIsExtensionPendingCustodianApprovalFunction>
+        function(

[Marc Treib, 2016/05/09 14:50:57]

You could do
auto function = make_scoped_refptr(new ...)
to save some characters. Also maybe 
using extensions::WebstorePrivateIsExtensionPendingCustodianApprovalFunction;
above if that saves a line break.

[mamir, 2016/05/09 15:30:15]

Done.

+            new extensions::
+                WebstorePrivateIsExtensionPendingCustodianApprovalFunction());
+    std::unique_ptr<base::Value> result(RunFunctionAndReturnSingleResult(
+        function.get(), "[\"" + extension_id + "\"]", browser_context()));
+
+    bool copy_bool_result = false;
+    EXPECT_TRUE(result->GetAsBoolean(&copy_bool_result));
+    return copy_bool_result;
+  }
+
   void InitServices(bool profile_is_supervised) {
     ExtensionServiceInitParams params = CreateDefaultInitParams();
     params.profile_is_supervised = profile_is_supervised;
@@ -1685,6 +1702,7 @@ TEST_F(ExtensionServiceTestSupervised, InstallOnlyAllowedByCustodian) {
   EXPECT_FALSE(extensions[0]);
   ASSERT_TRUE(extensions[1]);
   EXPECT_TRUE(registry()->enabled_extensions().Contains(extensions[1]->id()));
+  EXPECT_FALSE(IsExtensionPendingCustodianApproval(extensions[1]->id()));
 }
 
 TEST_F(ExtensionServiceTestSupervised,
@@ -1705,7 +1723,9 @@ TEST_F(ExtensionServiceTestSupervised,
   ASSERT_TRUE(extensions[0]);
   ASSERT_TRUE(extensions[1]);
   EXPECT_TRUE(registry()->disabled_extensions().Contains(extensions[0]->id()));
+  EXPECT_TRUE(IsExtensionPendingCustodianApproval(extensions[0]->id()));
   EXPECT_TRUE(registry()->enabled_extensions().Contains(extensions[1]->id()));
+  EXPECT_FALSE(IsExtensionPendingCustodianApproval(extensions[1]->id()));
 }
 
 TEST_F(ExtensionServiceTestSupervised,
@@ -1734,6 +1754,7 @@ TEST_F(ExtensionServiceTestSupervised,
 
   // The extension should not be enabled anymore.
   EXPECT_FALSE(registry()->enabled_extensions().Contains(id));
+  EXPECT_TRUE(IsExtensionPendingCustodianApproval(id));
 }
 
 TEST_F(ExtensionServiceTestSupervised,
@@ -1793,6 +1814,7 @@ TEST_F(ExtensionServiceTestSupervised, UpdateWithoutPermissionIncrease) {
   ASSERT_TRUE(extension);
   // The version should have changed.
   EXPECT_NE(extension->VersionString(), old_version);
+  EXPECT_FALSE(IsExtensionPendingCustodianApproval(id));
 }
 
 TEST_F(ExtensionServiceTestSupervised, UpdateWithPermissionIncreaseNoApproval) {
@@ -1814,6 +1836,7 @@ TEST_F(ExtensionServiceTestSupervised, UpdateWithPermissionIncreaseNoApproval) {
               CreateExtensionUpdateRequest(RequestId(id, version2), testing::_))
       .Times(0);
   UpdatePermissionsTestExtension(id, version2, DISABLED);
+  EXPECT_FALSE(IsExtensionPendingCustodianApproval(id));
 }
 
 TEST_F(ExtensionServiceTestSupervised,
@@ -1835,6 +1858,7 @@ TEST_F(ExtensionServiceTestSupervised,
   EXPECT_CALL(*creator, CreateExtensionUpdateRequest(RequestId(id, version2),
                                                      testing::_));
   UpdatePermissionsTestExtension(id, version2, DISABLED);
+  EXPECT_TRUE(IsExtensionPendingCustodianApproval(id));
 
   // Simulate a custodian approval for re-enabling the extension coming in
   // through Sync, but set the old version. This can happen when there already
@@ -1862,6 +1886,7 @@ TEST_F(ExtensionServiceTestSupervised,
       id, base::Version(version1)));
   EXPECT_FALSE(extension_sync_service()->HasPendingReenable(
       id, base::Version(version2)));
+  EXPECT_TRUE(IsExtensionPendingCustodianApproval(id));
 }
 
 TEST_F(ExtensionServiceTestSupervised,
@@ -1881,6 +1906,7 @@ TEST_F(ExtensionServiceTestSupervised,
   EXPECT_CALL(*creator, CreateExtensionUpdateRequest(RequestId(id, version2),
                                                      testing::_));
   UpdatePermissionsTestExtension(id, version2, DISABLED);
+  EXPECT_TRUE(IsExtensionPendingCustodianApproval(id));
 
   // Simulate a custodian approval for re-enabling the extension coming in
   // through Sync.
@@ -1898,6 +1924,7 @@ TEST_F(ExtensionServiceTestSupervised,
   extension_sync_service()->ProcessSyncChanges(FROM_HERE, list);
   // The extension should have gotten re-enabled.
   EXPECT_TRUE(registry()->enabled_extensions().Contains(id));
+  EXPECT_FALSE(IsExtensionPendingCustodianApproval(id));
 }
 
 TEST_F(ExtensionServiceTestSupervised,
@@ -1971,6 +1998,8 @@ TEST_F(ExtensionServiceTestSupervised, SupervisedUserInitiatedInstalls) {
   EXPECT_TRUE(registry()->disabled_extensions().Contains(extension->id()));
   EXPECT_FALSE(registry()->enabled_extensions().Contains(extension->id()));
 
+  EXPECT_TRUE(IsExtensionPendingCustodianApproval(extension->id()));
+
   // Simulate a custodian approval for enabling the extension coming in
   // through Sync.
   sync_pb::EntitySpecifics specifics;
@@ -1987,6 +2016,7 @@ TEST_F(ExtensionServiceTestSupervised, SupervisedUserInitiatedInstalls) {
   extension_sync_service()->ProcessSyncChanges(FROM_HERE, list);
   // The extension should be enabled now.
   EXPECT_TRUE(registry()->enabled_extensions().Contains(extension->id()));
+  EXPECT_FALSE(IsExtensionPendingCustodianApproval(extension->id()));
 }
 
 TEST_F(ExtensionServiceSyncTest, SyncUninstallByCustodianSkipsPolicy) {