germ/test_util.h

// Copyright 2015 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.

#ifndef GERM_TEST_UTIL_H_
#define GERM_TEST_UTIL_H_

#include <signal.h>

#include <string>

#include <gmock/gmock.h>

#include "germ/proto_bindings/soma_container_spec.pb.h"

namespace germ {

// Sends SIGTERM to all processes in the current process's process group if this
// object is alive for |seconds|.
class ScopedAlarm {
 public:
  explicit ScopedAlarm(unsigned int seconds);
  ~ScopedAlarm();

 private:
  struct sigaction oldact_;
};

// Matcher for ContainerSpecs. Only checks whether names match.
MATCHER_P(EqualsSpec, expected, "") {
  return expected.name() == arg.name();
}

soma::ContainerSpec MakeSpecForTest(const std::string& name);

}  // namespace germ

#endif  // GERM_TEST_UTIL_H_