child_job_unittest.cc

// Copyright (c) 2009-2010 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 "login_manager/child_job.h"

#include <gtest/gtest.h>
#include <unistd.h>

#include <base/command_line.h>
#include <base/logging.h>

namespace login_manager {

class SetUidExecJobTest : public ::testing::Test { };

TEST(SetUidExecJobTest, PopulateArgvTest) {
  const char *my_argv[] = { "zero",
                            "one",
                            "two" };
  const int my_argc = 3;
  CommandLine cl(my_argc, my_argv);

  SetUidExecJob job(&cl, NULL, false);

  std::vector<std::string> data = job.ExtractArgvForTest();
  std::vector<std::string>::const_iterator it  = data.begin();
  for (int i = 1; i < my_argc; ++i) {
    EXPECT_EQ(my_argv[i], *it);
    ++it;
  }
  EXPECT_TRUE(data.end() == it);
}

// I would love to re-use the code in the following functions, but I'd
// have to make several methods of SetUidExecJob public to do so.
TEST(SetUidExecJobTest, FlagAppendTest) {
  const char *my_argv[] = { "zero",
                            "one",
                            "two" };
  const int my_argc = 3;
  CommandLine cl(my_argc, my_argv);

  SetUidExecJob job(&cl, NULL, true);
  job.AppendNeededFlags();

  std::vector<std::string> data = job.ExtractArgvForTest();
  std::vector<std::string>::const_iterator it  = data.begin();
  for (int i = 1; i < my_argc; ++i) {
    EXPECT_EQ(my_argv[i], *it);
    ++it;
  }
  EXPECT_EQ(SetUidExecJob::kLoginManagerFlag, *it++);
  EXPECT_TRUE(data.end() == it) << *it;
}

TEST(SetUidExecJobTest, AppendUserFlagTest) {
  const char *my_argv[] = { "zero",
                            "one",
                            "two" };
  const int my_argc = 3;
  CommandLine cl(my_argc, my_argv);

  std::string user("me");
  SetUidExecJob job(&cl, NULL, true);
  job.SetState(user);
  job.AppendNeededFlags();

  std::vector<std::string> data = job.ExtractArgvForTest();
  std::vector<std::string>::const_iterator it  = data.begin();
  for (int i = 1; i < my_argc; ++i) {
    EXPECT_EQ(my_argv[i], *it);
    ++it;
  }
  std::string userflag(SetUidExecJob::kLoginUserFlag);
  userflag.append(user);
  EXPECT_EQ(userflag, *it++);
  EXPECT_TRUE(data.end() == it) << *it;
}

TEST(SetUidExecJobTest, ShouldStopTest) {
  const char *my_argv[] = { "zero",
                            "one",
                            "two" };
  const int my_argc = 3;
  CommandLine cl(my_argc, my_argv);

  SetUidExecJob job(&cl, NULL, false);
  job.RecordTime();
  EXPECT_TRUE(job.ShouldStop());
}

TEST(SetUidExecJobTest, ShouldNotStopTest) {
  const char *my_argv[] = { "zero",
                            "one",
                            "two" };
  const int my_argc = 3;
  CommandLine cl(my_argc, my_argv);

  SetUidExecJob job(&cl, NULL, false);
  job.RecordTime();
  sleep(1);
  EXPECT_FALSE(job.ShouldStop());
}

}  // namespace login_manager