simpleperf/workload_test.cpp

   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 }