arangodb/tests/Basics/ApplicationServerTest.cpp

////////////////////////////////////////////////////////////////////////////////
/// @brief test suite for application server
///
/// @file
///
/// DISCLAIMER
///
/// Copyright 2004-2012 triagens GmbH, Cologne, Germany
///
/// Licensed 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.
///
/// Copyright holder is triAGENS GmbH, Cologne, Germany
///
/// @author Jan Steemann
/// @author Copyright 2017, ArangoDB GmbH, Cologne, Germany
////////////////////////////////////////////////////////////////////////////////

#include "Basics/Common.h"

#include "gtest/gtest.h"

#include "ApplicationFeatures/ApplicationFeature.h"
#include "ApplicationFeatures/ApplicationServer.h"
#include "Basics/Exceptions.h"
#include "ProgramOptions/ProgramOptions.h"

using namespace arangodb;

// -----------------------------------------------------------------------------
// --SECTION--                                                        test suite
// -----------------------------------------------------------------------------

////////////////////////////////////////////////////////////////////////////////
/// @brief setup
////////////////////////////////////////////////////////////////////////////////

class TestFeature : public application_features::ApplicationFeature {
 public:
  TestFeature(application_features::ApplicationServer& server,
              std::string const& name, std::vector<std::string> const& startsAfter,
              std::vector<std::string> const& startsBefore)
      : ApplicationFeature(server, name) {
    for (auto const& it : startsAfter) {
      this->startsAfter(it);
    }
    for (auto const& it : startsBefore) {
      this->startsBefore(it);
    }
  }
};

TEST(ApplicationServerTest, test_startsAfterValid) {
  bool failed = false;
  std::function<void(std::string const&)> callback = [&failed](std::string const&) {
    failed = true;
  };

  auto options =
      std::make_shared<options::ProgramOptions>("arangod", "something", "",
                                                "path");
  application_features::ApplicationServer server(options, "path");

  auto feature1 = std::make_unique<TestFeature>(server, "feature1",
                                                std::vector<std::string>{},
                                                std::vector<std::string>{});
  auto feature2 =
      std::make_unique<TestFeature>(server, "feature2", std::vector<std::string>{"feature1"},
                                    std::vector<std::string>{});

  server.registerFailCallback(callback);
  server.addFeature(feature1.get());
  server.addFeature(feature2.get());
  server.setupDependencies(true);

  EXPECT_TRUE(!failed);
  EXPECT_TRUE(feature1->doesStartBefore("feature2"));
  EXPECT_TRUE(!feature1->doesStartAfter("feature2"));
  EXPECT_TRUE(!feature1->doesStartBefore("feature1"));
  EXPECT_TRUE(feature1->doesStartAfter("feature1"));
  EXPECT_TRUE(!feature2->doesStartBefore("feature1"));
  EXPECT_TRUE(feature2->doesStartAfter("feature1"));
  EXPECT_TRUE(!feature2->doesStartBefore("feature2"));
  EXPECT_TRUE(feature2->doesStartAfter("feature2"));

  feature1.release();
  feature2.release();
}

TEST(ApplicationServerTest, test_startsAfterCyclic) {
  bool failed = false;
  std::function<void(std::string const&)> callback = [&failed](std::string const&) {
    failed = true;
  };

  auto options =
      std::make_shared<options::ProgramOptions>("arangod", "something", "",
                                                "path");
  application_features::ApplicationServer server(options, "path");

  auto feature1 =
      std::make_unique<TestFeature>(server, "feature1", std::vector<std::string>{"feature2"},
                                    std::vector<std::string>{});
  auto feature2 =
      std::make_unique<TestFeature>(server, "feature2", std::vector<std::string>{"feature1"},
                                    std::vector<std::string>{});

  server.registerFailCallback(callback);
  server.addFeature(feature1.get());
  server.addFeature(feature2.get());

  try {
    server.setupDependencies(true);
  } catch (basics::Exception const& ex) {
    EXPECT_TRUE(ex.code() == TRI_ERROR_INTERNAL);
    failed = true;
  }
  EXPECT_TRUE(failed);

  feature1.release();
  feature2.release();
}

TEST(ApplicationServerTest, test_startsBeforeCyclic) {
  bool failed = false;
  std::function<void(std::string const&)> callback = [&failed](std::string const&) {
    failed = true;
  };

  auto options =
      std::make_shared<options::ProgramOptions>("arangod", "something", "",
                                                "path");
  application_features::ApplicationServer server(options, "path");

  auto feature1 =
      std::make_unique<TestFeature>(server, "feature1", std::vector<std::string>{},
                                    std::vector<std::string>{"feature2"});
  auto feature2 =
      std::make_unique<TestFeature>(server, "feature2", std::vector<std::string>{},
                                    std::vector<std::string>{"feature1"});

  server.registerFailCallback(callback);
  server.addFeature(feature1.get());
  server.addFeature(feature2.get());

  try {
    server.setupDependencies(true);
  } catch (basics::Exception const& ex) {
    EXPECT_TRUE(ex.code() == TRI_ERROR_INTERNAL);
    failed = true;
  }
  EXPECT_TRUE(failed);

  feature1.release();
  feature2.release();
}