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