update_engine/common/action_unittest.cc

// Copyright 2010 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "update_engine/common/action.h"

#include <string>
#include <utility>

#include <gtest/gtest.h>

#include "update_engine/common/action_processor.h"

using std::string;

namespace chromeos_update_engine {

using chromeos_update_engine::ActionPipe;

class ActionTestAction;

template <>
class ActionTraits<ActionTestAction> {
 public:
  typedef string OutputObjectType;
  typedef string InputObjectType;
};

// This is a simple Action class for testing.
class ActionTestAction : public Action<ActionTestAction> {
 public:
  typedef string InputObjectType;
  typedef string OutputObjectType;
  ActionPipe<string>* in_pipe() { return in_pipe_.get(); }
  ActionPipe<string>* out_pipe() { return out_pipe_.get(); }
  ActionProcessor* processor() { return processor_; }
  void PerformAction() {}
  void CompleteAction() {
    ASSERT_TRUE(processor());
    processor()->ActionComplete(this, ErrorCode::kSuccess);
  }
  string Type() const { return "ActionTestAction"; }
};

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

// This test creates two simple Actions and sends a message via an ActionPipe
// from one to the other.
TEST(ActionTest, SimpleTest) {
  auto action = std::make_unique<ActionTestAction>();
  auto action_ptr = action.get();
  EXPECT_FALSE(action->in_pipe());
  EXPECT_FALSE(action->out_pipe());
  EXPECT_FALSE(action->processor());
  EXPECT_FALSE(action->IsRunning());

  ActionProcessor action_processor;
  action_processor.EnqueueAction(std::move(action));
  EXPECT_EQ(&action_processor, action_ptr->processor());
  action_processor.StartProcessing();
  EXPECT_TRUE(action_ptr->IsRunning());
  action_ptr->CompleteAction();
}

}  // namespace chromeos_update_engine