1// 2// Copyright (C) 2010 The Android Open Source Project 3// 4// Licensed under the Apache License, Version 2.0 (the "License"); 5// you may not use this file except in compliance with the License. 6// You may obtain a copy of the License at 7// 8// http://www.apache.org/licenses/LICENSE-2.0 9// 10// Unless required by applicable law or agreed to in writing, software 11// distributed under the License is distributed on an "AS IS" BASIS, 12// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13// See the License for the specific language governing permissions and 14// limitations under the License. 15// 16 17#include "update_engine/common/action.h" 18 19#include <gtest/gtest.h> 20#include <string> 21#include "update_engine/common/action_processor.h" 22 23using std::string; 24 25namespace chromeos_update_engine { 26 27using chromeos_update_engine::ActionPipe; 28 29class ActionTestAction; 30 31template<> 32class ActionTraits<ActionTestAction> { 33 public: 34 typedef string OutputObjectType; 35 typedef string InputObjectType; 36}; 37 38// This is a simple Action class for testing. 39class ActionTestAction : public Action<ActionTestAction> { 40 public: 41 typedef string InputObjectType; 42 typedef string OutputObjectType; 43 ActionPipe<string>* in_pipe() { return in_pipe_.get(); } 44 ActionPipe<string>* out_pipe() { return out_pipe_.get(); } 45 ActionProcessor* processor() { return processor_; } 46 void PerformAction() {} 47 void CompleteAction() { 48 ASSERT_TRUE(processor()); 49 processor()->ActionComplete(this, ErrorCode::kSuccess); 50 } 51 string Type() const { return "ActionTestAction"; } 52}; 53 54class ActionTest : public ::testing::Test { }; 55 56// This test creates two simple Actions and sends a message via an ActionPipe 57// from one to the other. 58TEST(ActionTest, SimpleTest) { 59 ActionTestAction action; 60 61 EXPECT_FALSE(action.in_pipe()); 62 EXPECT_FALSE(action.out_pipe()); 63 EXPECT_FALSE(action.processor()); 64 EXPECT_FALSE(action.IsRunning()); 65 66 ActionProcessor action_processor; 67 action_processor.EnqueueAction(&action); 68 EXPECT_EQ(&action_processor, action.processor()); 69 70 action_processor.StartProcessing(); 71 EXPECT_TRUE(action.IsRunning()); 72 action.CompleteAction(); 73 EXPECT_FALSE(action.IsRunning()); 74} 75 76} // namespace chromeos_update_engine 77