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 259 matching lines @@
   EXPECT_TRUE(SchemaMapEquals(registry->schema_map(), forwarding.schema_map()));
 
   // No notifications expected for these calls.
   EXPECT_FALSE(registry->IsReady());
   EXPECT_FALSE(forwarding.IsReady());
 
   registry->SetExtensionsDomainsReady();
   EXPECT_FALSE(registry->IsReady());
   EXPECT_FALSE(forwarding.IsReady());
 
+  EXPECT_CALL(observer, OnSchemaRegistryReady());
   registry->SetDomainReady(POLICY_DOMAIN_CHROME);
   EXPECT_TRUE(registry->IsReady());
-  // The ForwardingSchemaRegistry becomes ready independently of the wrapped
-  // registry.
-  EXPECT_FALSE(forwarding.IsReady());
+  EXPECT_TRUE(forwarding.IsReady());
+  Mock::VerifyAndClearExpectations(&observer);
 
   EXPECT_TRUE(SchemaMapEquals(registry->schema_map(), forwarding.schema_map()));
   Mock::VerifyAndClearExpectations(&observer);
 
   forwarding.SetExtensionsDomainsReady();
-  EXPECT_FALSE(forwarding.IsReady());
-  Mock::VerifyAndClearExpectations(&observer);
-
-  EXPECT_CALL(observer, OnSchemaRegistryReady());
   forwarding.SetDomainReady(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, ForwardingSchemaRegistryReadiness) {
+  std::unique_ptr<SchemaRegistry> registry(new SchemaRegistry);
+
+  ForwardingSchemaRegistry forwarding_1(registry.get());
+  EXPECT_FALSE(registry->IsReady());
+  EXPECT_FALSE(forwarding_1.IsReady());
+
+  // Once the wrapped registry gets ready, the forwarding schema registry
+  // becomes ready too.
+  registry->SetAllDomainsReady();
+  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());
+  EXPECT_TRUE(forwarding_2.IsReady());
+
+  // Destruction of the wrapped registry doesn't change the readiness of the
+  // forwarding registry.
+  registry.reset();
+  EXPECT_TRUE(forwarding_1.IsReady());
+  EXPECT_TRUE(forwarding_2.IsReady());
+}
+
 }  // namespace policy