1 /* 2 * Copyright (C) 2015 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 <gtest/gtest.h> 18 19 #include <signal.h> 20 21 #include "scoped_signal_handler.h" 22 #include "utils.h" 23 #include "workload.h" 24 25 static volatile bool signaled; 26 static void signal_handler(int) { 27 signaled = true; 28 } 29 30 TEST(workload, success) { 31 signaled = false; 32 ScopedSignalHandler scoped_signal_handler({SIGCHLD}, signal_handler); 33 auto workload = Workload::CreateWorkload({"sleep", "1"}); 34 ASSERT_TRUE(workload != nullptr); 35 ASSERT_TRUE(workload->GetPid() != 0); 36 ASSERT_TRUE(workload->Start()); 37 while (!signaled) { 38 } 39 } 40 41 TEST(workload, execvp_failure) { 42 auto workload = Workload::CreateWorkload({"/dev/null"}); 43 ASSERT_TRUE(workload != nullptr); 44 ASSERT_FALSE(workload->Start()); 45 } 46 47 static void run_signaled_workload() { 48 { 49 signaled = false; 50 ScopedSignalHandler scoped_signal_handler({SIGCHLD}, signal_handler); 51 auto workload = Workload::CreateWorkload({"sleep", "10"}); 52 ASSERT_TRUE(workload != nullptr); 53 ASSERT_TRUE(workload->Start()); 54 ASSERT_EQ(0, kill(workload->GetPid(), SIGKILL)); 55 while (!signaled) { 56 } 57 } 58 // Make sure all destructors are called before exit(). 59 exit(0); 60 } 61 62 TEST(workload, signaled_warning) { 63 ASSERT_EXIT(run_signaled_workload(), testing::ExitedWithCode(0), 64 "child process was terminated by signal"); 65 } 66 67 static void run_exit_nonzero_workload() { 68 { 69 signaled = false; 70 ScopedSignalHandler scoped_signal_handler({SIGCHLD}, signal_handler); 71 auto workload = Workload::CreateWorkload({"ls", "nonexistdir"}); 72 ASSERT_TRUE(workload != nullptr); 73 ASSERT_TRUE(workload->Start()); 74 while (!signaled) { 75 } 76 } 77 // Make sure all destructors are called before exit(). 78 exit(0); 79 } 80 81 TEST(workload, exit_nonzero_warning) { 82 ASSERT_EXIT(run_exit_nonzero_workload(), testing::ExitedWithCode(0), 83 "child process exited with exit code"); 84 } 85
