Allow PolicyLoadStatusSample to override reporting method

components/policy/core/common/preg_parser_unittest.cc

 // Copyright (c) 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/preg_parser.h"
 
 #include <utility>
 
 #include "base/base_paths.h"
+#include "base/bind.h"
 #include "base/files/file_path.h"
 #include "base/json/json_writer.h"
 #include "base/logging.h"
 #include "base/memory/ptr_util.h"
 #include "base/path_service.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/values.h"
 #include "components/policy/core/common/policy_load_status.h"
 #include "components/policy/core/common/registry_dict.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace policy {
 namespace preg_parser {
 namespace {
 
 // Preg files are relative to |kRegistryPolBaseDir|.
 const char kRegistryPolBaseDir[] = "chrome/test/data/policy/gpo";
 const char kRegistryPolFile[] = "parser_test/registry.pol";
 const char kInvalidEncodingRegistryPolFile[] = "invalid_encoding/registry.pol";
+const char kNonExistingRegistryPolFile[] = "does_not_exist.pol";
 
 const char kRegistryKey[] = "SOFTWARE\\Policies\\Chromium";
 
 // Check whether two RegistryDicts equal each other.
 testing::AssertionResult RegistryDictEquals(const RegistryDict& a,
                                             const RegistryDict& b) {
   auto iter_key_a = a.keys().begin();
   auto iter_key_b = b.keys().begin();
   for (; iter_key_a != a.keys().end() && iter_key_b != b.keys().end();
        ++iter_key_a, ++iter_key_b) {
@@ skipping 127 matching lines @@
       test_data_dir_.AppendASCII(kInvalidEncodingRegistryPolFile));
   PolicyLoadStatusSample status;
   RegistryDict dict;
   ASSERT_TRUE(preg_parser::ReadFile(test_file, base::ASCIIToUTF16(kRegistryKey),
                                     &dict, &status));
 
   RegistryDict empty;
   EXPECT_TRUE(RegistryDictEquals(dict, empty));
 }
 
+TEST_F(PRegParserTest, OverrideReportCallback) {
+  // Tests overriding the PolicyLoadStatusSample's ReportCallback.
+  std::bitset<POLICY_LOAD_STATUS_SIZE> status_bits;
+  PolicyLoadStatusSample::ReportCallback report_callback =
+      base::Bind([](std::bitset<POLICY_LOAD_STATUS_SIZE>* status_bits,
+                    PolicyLoadStatus status) { (*status_bits)[status] = true; },
+                 &status_bits);
+
+  {
+    // Needs to be scoped because reporting happens in the destructor of
+    // PolicyLoadStatusSample.
+    PolicyLoadStatusSample status(report_callback);
+    RegistryDict dict;
+    base::FilePath test_file(
+        test_data_dir_.AppendASCII(kNonExistingRegistryPolFile));
+    ASSERT_FALSE(preg_parser::ReadFile(
+        test_file, base::ASCIIToUTF16(kRegistryKey), &dict, &status));
+  }
+
+  std::bitset<POLICY_LOAD_STATUS_SIZE> expected_status_bits;
+  expected_status_bits[POLICY_LOAD_STATUS_STARTED] = true;
+  expected_status_bits[POLICY_LOAD_STATUS_READ_ERROR] = true;
+  EXPECT_EQ(expected_status_bits, status_bits);
+}
+
 }  // namespace
 }  // namespace preg_parser
 }  // namespace policy