chromeos/startup/startup_unittest.cc

// Copyright 2020 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 "chromeos/startup/startup.h"

#include <sys/mman.h>
#include <sys/types.h>
#include <unistd.h>

#include <string>

#include "base/command_line.h"
#include "base/files/file_util.h"
#include "base/files/scoped_file.h"
#include "base/logging.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/string_piece.h"
#include "base/test/scoped_command_line.h"
#include "chromeos/startup/startup_switches.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/abseil-cpp/absl/types/optional.h"

namespace chromeos {
namespace {

base::ScopedFD CreateMemoryFile(const base::StringPiece content) {
  base::ScopedFD file(memfd_create("test", 0));
  if (!file.is_valid()) {
    PLOG(ERROR) << "Failed to create a memory file";
    return base::ScopedFD();
  }

  if (!base::WriteFileDescriptor(file.get(), content)) {
    LOG(ERROR) << "Failed to write the data";
    return base::ScopedFD();
  }

  // Reset the cursor.
  if (lseek(file.get(), 0, SEEK_SET) < 0) {
    PLOG(ERROR) << "Failed to reset the file position";
    return base::ScopedFD();
  }

  return file;
}

}  // namespace

TEST(ChromeOSStartup, Startup) {
  constexpr char kTestData[] = "test test test test";
  base::ScopedFD file = CreateMemoryFile(kTestData);

  base::test::ScopedCommandLine scoped_command_line;
  base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();

  // Release the FD. The FD is consumed in ReadStartupData().
  command_line->AppendSwitchASCII(switches::kCrosStartupDataFD,
                                  base::NumberToString(file.release()));

  absl::optional<std::string> data = ReadStartupData();
  EXPECT_EQ(data, kTestData);
}

TEST(ChromeOSStartup, NoFlag) {
  EXPECT_FALSE(ReadStartupData());
}

}  // namespace chromeos