ceph/src/seastar/tests/unit/defer_test.cc

   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 2016 ScyllaDB
  20  */
  21
  22 #define BOOST_TEST_MODULE core
  23
  24 #include <boost/test/included/unit_test.hpp>
  25 #include <seastar/util/defer.hh>
  26
  27 using namespace seastar;
  28
  29 BOOST_AUTO_TEST_CASE(test_defer_does_not_run_when_canceled) {
  30     bool ran = false;
  31     {
  32         auto d = defer([&] () noexcept {
  33             ran = true;
  34         });
  35         d.cancel();
  36     }
  37     BOOST_REQUIRE(!ran);
  38 }
  39
  40 BOOST_AUTO_TEST_CASE(test_defer_runs) {
  41     bool ran = false;
  42     {
  43         auto d = defer([&] () noexcept {
  44             ran = true;
  45         });
  46     }
  47     BOOST_REQUIRE(ran);
  48 }
  49
  50 BOOST_AUTO_TEST_CASE(test_defer_runs_once_when_moved) {
  51     int ran = 0;
  52     {
  53         auto d = defer([&] () noexcept {
  54             ++ran;
  55         });
  56         {
  57             auto d2 = std::move(d);
  58         }
  59         BOOST_REQUIRE_EQUAL(1, ran);
  60     }
  61     BOOST_REQUIRE_EQUAL(1, ran);
  62 }
  63
  64 BOOST_AUTO_TEST_CASE(test_defer_does_not_run_when_moved_after_cancelled) {
  65     int ran = 0;
  66     {
  67         auto d = defer([&] () noexcept {
  68             ++ran;
  69         });
  70         d.cancel();
  71         {
  72             auto d2 = std::move(d);
  73         }
  74     }
  75     BOOST_REQUIRE_EQUAL(0, ran);
  76 }