oobe_config/oobe_config_test.cc

// Copyright 2018 The Chromium OS 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 "oobe_config/oobe_config.h"

#include <base/files/file_path.h>
#include <base/files/scoped_temp_dir.h>
#include <gtest/gtest.h>
#include <string>

#include "oobe_config/rollback_constants.h"

namespace oobe_config {

TEST(OobeConfigTest, SaveAndRestoreTest) {
  oobe_config::OobeConfig oobe_config;

  // Creating pre-rollback state.
  base::ScopedTempDir tempdir_before;
  ASSERT_TRUE(tempdir_before.CreateUniqueTempDir());
  oobe_config.set_prefix_path_for_testing(tempdir_before.GetPath());

  oobe_config.WriteFile(kSaveTempPath.Append(kInstallAttributesFileName),
                        "install_attributes");
  oobe_config.WriteFile(kSaveTempPath.Append(kOwnerKeyFileName), "owner");
  oobe_config.WriteFile(kSaveTempPath.Append(kPolicyFileName), "policy0");
  oobe_config.WriteFile(kSaveTempPath.Append(kPolicyDotOneFileNameForTesting),
                        "policy1");
  oobe_config.WriteFile(kSaveTempPath.Append(kShillDefaultProfileFileName),
                        "shill");

  // Saving rollback data.
  ASSERT_TRUE(oobe_config.UnencryptedRollbackSave());
  std::string rollback_data;
  ASSERT_TRUE(oobe_config.ReadFile(kUnencryptedStatefulRollbackDataPath,
                                   &rollback_data));
  EXPECT_FALSE(rollback_data.empty());

  // Simulate powerwash and only preserve rollback_data by creating new temp
  // dir.
  base::ScopedTempDir tempdir_after;
  ASSERT_TRUE(tempdir_after.CreateUniqueTempDir());
  oobe_config.set_prefix_path_for_testing(tempdir_after.GetPath());

  // Verify that we don't have any remaining files.
  std::string tmp_data = "x";
  ASSERT_FALSE(
      oobe_config.ReadFile(kUnencryptedStatefulRollbackDataPath, &tmp_data));
  EXPECT_TRUE(tmp_data.empty());
  // Preserving rollback data.
  ASSERT_TRUE(oobe_config.WriteFile(kUnencryptedStatefulRollbackDataPath,
                                    rollback_data));

  // Restore data.
  EXPECT_TRUE(oobe_config.UnencryptedRollbackRestore());

  // Verify that the config files are restored.
  std::string file_content;
  EXPECT_TRUE(oobe_config.ReadFile(
      kRestoreTempPath.Append(kInstallAttributesFileName), &file_content));
  EXPECT_EQ("install_attributes", file_content);
  EXPECT_TRUE(oobe_config.ReadFile(kRestoreTempPath.Append(kOwnerKeyFileName),
                                   &file_content));
  EXPECT_EQ("owner", file_content);
  EXPECT_TRUE(oobe_config.ReadFile(kRestoreTempPath.Append(kPolicyFileName),
                                   &file_content));
  EXPECT_EQ("policy0", file_content);
  EXPECT_TRUE(oobe_config.ReadFile(
      kRestoreTempPath.Append(kPolicyDotOneFileNameForTesting), &file_content));
  EXPECT_EQ("policy1", file_content);
  EXPECT_TRUE(oobe_config.ReadFile(
      kRestoreTempPath.Append(kShillDefaultProfileFileName), &file_content));
  EXPECT_EQ("shill", file_content);
}

}  // namespace oobe_config