libminifi/test/unit/DiskSpaceWatchdogTests.cpp

/**
 *
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <chrono>

#include "../TestBase.h"

#include "properties/Configure.h"
#include "DiskSpaceWatchdog.h"

namespace minifi = org::apache::nifi::minifi;
namespace dsg = minifi::disk_space_watchdog;
namespace chr = std::chrono;

TEST_CASE("disk_space_watchdog::read_config", "[dsg::read_config]") {
  const auto mebibytes = 1024 * 1024;

  SECTION("defaults are present") {
    const minifi::Configure configure;
    const auto conf = dsg::read_config(configure);
    REQUIRE(conf.interval >= chr::nanoseconds{0});
    REQUIRE(conf.restart_threshold_bytes > conf.stop_threshold_bytes);
  }

  SECTION("basic") {
    minifi::Configure configure;
    configure.set(minifi::Configure::minifi_disk_space_watchdog_stop_threshold, std::to_string(10 * mebibytes));
    configure.set(minifi::Configure::minifi_disk_space_watchdog_restart_threshold, std::to_string(25 * mebibytes));
    configure.set(minifi::Configure::minifi_disk_space_watchdog_interval, "2000 millis");
    const auto conf = dsg::read_config(configure);
    REQUIRE(conf.stop_threshold_bytes == 10 * mebibytes);
    REQUIRE(conf.restart_threshold_bytes == 25 * mebibytes);
    REQUIRE(conf.interval == chr::seconds{2});
  }

  SECTION("units") {
    minifi::Configure configure;
    configure.set(minifi::Configure::minifi_disk_space_watchdog_stop_threshold, "100 MB");
    configure.set(minifi::Configure::minifi_disk_space_watchdog_restart_threshold, "250 MB");
    configure.set(minifi::Configure::minifi_disk_space_watchdog_interval, "7 sec");
    const auto conf = dsg::read_config(configure);
    REQUIRE(conf.stop_threshold_bytes == 100 * mebibytes);
    REQUIRE(conf.restart_threshold_bytes == 250 * mebibytes);
    REQUIRE(conf.interval == chr::seconds{7});
  }
}