ceph/src/seastar/include/seastar/testing/test_runner.hh

   1 /*
   2  * This file is open source software, licensed to you under the terms
   3  * of the Apache License, Version 2.0 (the "License").  See the NOTICE file
   4  * distributed with this work for additional information regarding copyright
   5  * ownership.  You may not use this file except in compliance with the License.
   6  *
   7  * You may obtain a copy of the License at
   8  *
   9  *   http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing,
  12  * software distributed under the License is distributed on an
  13  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  14  * KIND, either express or implied.  See the License for the
  15  * specific language governing permissions and limitations
  16  * under the License.
  17  */
  18 /*
  19  * Copyright (C) 2015 Cloudius Systems, Ltd.
  20  */
  21
  22 #pragma once
  23
  24 #include <memory>
  25 #include <functional>
  26 #include <atomic>
  27 #include <random>
  28 #include <seastar/core/future.hh>
  29 #include <seastar/core/posix.hh>
  30 #include <seastar/testing/exchanger.hh>
  31 #include <seastar/testing/random.hh>
  32
  33 namespace seastar {
  34
  35 namespace testing {
  36
  37 class test_runner {
  38 private:
  39     std::unique_ptr<posix_thread> _thread;
  40     std::atomic<bool> _started{false};
  41     exchanger<std::function<future<>()>> _task;
  42     bool _done = false;
  43     int _exit_code{0};
  44
  45     struct start_thread_args {
  46         int ac;
  47         char** av;
  48         start_thread_args(int ac_, char** av_) noexcept : ac(ac_), av(av_) {}
  49     };
  50     std::unique_ptr<start_thread_args> _st_args;
  51     void start_thread(int ac, char** av);
  52 public:
  53     // Returns whether initialization was successful.
  54     // Will return as soon as the seastar::app was started.
  55     bool start(int argc, char** argv);
  56     ~test_runner();
  57     void run_sync(std::function<future<>()> task);
  58     // Returns the return value of the underlying `seastar::app::run()`.
  59     int finalize();
  60 };
  61
  62 test_runner& global_test_runner();
  63
  64 }
  65
  66 }