// Copyright (c) 2014 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_exit_handler.h"

#include <signal.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <unistd.h>

#include <vector>

#include <base/message_loop/message_loop.h>
#include <base/run_loop.h>
#include <gtest/gtest.h>

#include "login_manager/job_manager.h"
#include "login_manager/system_utils_impl.h"

namespace login_manager {

// A fake job manager implementation for testing.
class FakeJobManager : public JobManagerInterface {
 public:
  explicit FakeJobManager(base::RunLoop* run_loop) : run_loop_(run_loop) {}
  virtual ~FakeJobManager() {}

  pid_t managed_pid() { return 0; }
  const siginfo_t& last_status() { return last_status_; }

  // Implementation of JobManagerInterface.
  virtual bool IsManagedJob(pid_t pid) OVERRIDE { return true; }
  virtual void HandleExit(const siginfo_t& s) OVERRIDE {
    last_status_ = s;
    run_loop_->Quit();
  }
  virtual void RequestJobExit() OVERRIDE {}
  virtual void EnsureJobExit(base::TimeDelta timeout) OVERRIDE {}

 private:
  base::RunLoop* run_loop_;
  siginfo_t last_status_;

  DISALLOW_COPY_AND_ASSIGN(FakeJobManager);
};

class ChildExitHandlerTest : public ::testing::Test {
 public:
  ChildExitHandlerTest()
      : handler_(&system_utils_), fake_manager_(&run_loop_) {}
  virtual ~ChildExitHandlerTest() {}

  virtual void SetUp() {
    std::vector<JobManagerInterface*> managers;
    managers.push_back(&fake_manager_);
    handler_.Init(managers);
  }

 protected:
  base::MessageLoopForIO loop_;
  base::RunLoop run_loop_;
  SystemUtilsImpl system_utils_;
  ChildExitHandler handler_;
  FakeJobManager fake_manager_;

 private:
  DISALLOW_COPY_AND_ASSIGN(ChildExitHandlerTest);
};

TEST_F(ChildExitHandlerTest, ChildExit) {
  // Fork off a child process that exits immediately.
  pid_t child_pid = system_utils_.fork();
  if (child_pid == 0) {
    _Exit(EXIT_SUCCESS);
  }

  // Spin the message loop.
  run_loop_.Run();

  // Verify child termination has been reported to |fake_manager|.
  EXPECT_EQ(child_pid, fake_manager_.last_status().si_pid);
  EXPECT_EQ(SIGCHLD, fake_manager_.last_status().si_signo);
  EXPECT_EQ(static_cast<int>(CLD_EXITED), fake_manager_.last_status().si_code);
  EXPECT_EQ(EXIT_SUCCESS, fake_manager_.last_status().si_status);
}

}  // namespace login_manager