Enable fetching of admin policies for login screen apps

components/policy/core/common/schema_registry_unittest.cc

 // Copyright 2013 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 "components/policy/core/common/schema_registry.h"
 
 #include <memory>
 
 #include "components/policy/core/common/policy_namespace.h"
 #include "components/policy/core/common/schema.h"
@@ skipping 231 matching lines @@
 
   EXPECT_CALL(observer, OnSchemaRegistryUpdated(false));
   registry2.reset();
   Mock::VerifyAndClearExpectations(&observer);
 
   combined.RemoveObserver(&observer);
 }
 
 TEST(SchemaRegistryTest, ForwardingSchemaRegistry) {
   std::unique_ptr<SchemaRegistry> registry(new SchemaRegistry);
-  ForwardingSchemaRegistry forwarding(registry.get());
+  ForwardingSchemaRegistry forwarding(registry.get(),
+                                      false /* forward_readiness */);
   MockSchemaRegistryObserver observer;
   forwarding.AddObserver(&observer);
 
   EXPECT_FALSE(registry->IsReady());
   EXPECT_FALSE(forwarding.IsReady());
   // They always have the same SchemaMap.
   EXPECT_TRUE(SchemaMapEquals(registry->schema_map(), forwarding.schema_map()));
 
   EXPECT_CALL(observer, OnSchemaRegistryUpdated(true));
   registry->RegisterComponent(PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, "abc"),
@@ skipping 11 matching lines @@
   EXPECT_FALSE(registry->IsReady());
   EXPECT_FALSE(forwarding.IsReady());
 
   registry->SetReady(POLICY_DOMAIN_EXTENSIONS);
   registry->SetReady(POLICY_DOMAIN_SIGNIN_EXTENSIONS);
   EXPECT_FALSE(registry->IsReady());
   EXPECT_FALSE(forwarding.IsReady());
 
   registry->SetReady(POLICY_DOMAIN_CHROME);
   EXPECT_TRUE(registry->IsReady());
-  // The ForwardingSchemaRegistry becomes ready independently of the wrapped
+  // The ForwardingSchemaRegistry, if constructed with forward_readiness
+  // parameter equal to false, becomes ready independently of the wrapped
   // registry.
   EXPECT_FALSE(forwarding.IsReady());
 
   EXPECT_TRUE(SchemaMapEquals(registry->schema_map(), forwarding.schema_map()));
   Mock::VerifyAndClearExpectations(&observer);
 
   forwarding.SetReady(POLICY_DOMAIN_EXTENSIONS);
   forwarding.SetReady(POLICY_DOMAIN_SIGNIN_EXTENSIONS);
   EXPECT_FALSE(forwarding.IsReady());
   Mock::VerifyAndClearExpectations(&observer);
 
   EXPECT_CALL(observer, OnSchemaRegistryReady());
   forwarding.SetReady(POLICY_DOMAIN_CHROME);
   EXPECT_TRUE(forwarding.IsReady());
   Mock::VerifyAndClearExpectations(&observer);
 
   // Keep the same SchemaMap when the original registry is gone.
   // No notifications are expected in this case either.
   scoped_refptr<SchemaMap> schema_map = registry->schema_map();
   registry.reset();
   EXPECT_TRUE(SchemaMapEquals(schema_map, forwarding.schema_map()));
   Mock::VerifyAndClearExpectations(&observer);
 
   forwarding.RemoveObserver(&observer);
 }
 
+TEST(SchemaRegistryTest, ReadinessForwardingSchemaRegistry) {
+  // Test ForwardingSchemaRegistry when constructed with forward_readiness equal
+  // to true.
+
+  std::unique_ptr<SchemaRegistry> registry(new SchemaRegistry);
+
+  ForwardingSchemaRegistry forwarding_1(registry.get(),
+                                        true /* forward_readiness */);
+  EXPECT_FALSE(registry->IsReady());
+  EXPECT_FALSE(forwarding_1.IsReady());
+
+  // Once the wrapped registry gets ready, the forwarding schema registry
+  // becomes ready too.
+  registry->SetReady(POLICY_DOMAIN_CHROME);
+  registry->SetReady(POLICY_DOMAIN_EXTENSIONS);
+  registry->SetReady(POLICY_DOMAIN_SIGNIN_EXTENSIONS);
+  EXPECT_TRUE(registry->IsReady());
+  EXPECT_TRUE(forwarding_1.IsReady());
+
+  // The wrapped registry was ready at the time when the forwarding registry was
+  // constructed, so the forwarding registry is immediately ready too.
+  ForwardingSchemaRegistry forwarding_2(registry.get(),
+                                        true /* forward_readiness */);
+  EXPECT_TRUE(forwarding_2.IsReady());
+
+  // Destruction of the wrapped registry doesn't change the readiness of
+  // forwarding registry.
+  registry.reset();
+  EXPECT_TRUE(forwarding_1.IsReady());
+  EXPECT_TRUE(forwarding_2.IsReady());
+}
+
 }  // namespace policy