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 23 using std::string; 24 25 namespace chromeos_update_engine { 26 27 using chromeos_update_engine::ActionPipe; 28 29 class ActionTestAction; 30 31 template<> 32 class ActionTraits<ActionTestAction> { 33 public: 34 typedef string OutputObjectType; 35 typedef string InputObjectType; 36 }; 37 38 // This is a simple Action class for testing. 39 class 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 54 class 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. 58 TEST(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
