YugabyteDB (2.13.1.0-b60, 21121d69985fbf76aa6958d8f04a9bfa936293b5)

// Copyright (c) YugaByte, Inc.

//

// 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.

//

#ifndef YB_INTEGRATION_TESTS_CQL_TEST_BASE_H

#define YB_INTEGRATION_TESTS_CQL_TEST_BASE_H

#include "yb/integration-tests/cql_test_util.h"

#include "yb/integration-tests/mini_cluster.h"

#include "yb/integration-tests/yb_mini_cluster_test_base.h"

#include "yb/yql/cql/cqlserver/cql_server.h"

namespace yb {

template <class MiniClusterType>

class CqlTestBase : public MiniClusterTestWithClient<MiniClusterType> {

 public:

  static constexpr auto kDefaultNumMasters = 1;

  static constexpr auto kDefaultNumTabletServers = 3;

  virtual int num_masters() {

    return kDefaultNumMasters;

  }

  virtual int num_tablet_servers() {

    return kDefaultNumTabletServers;

  }

  void SetUp() override;

  CHECKED_STATUS RestartCluster();

  void ShutdownCluster();

  CHECKED_STATUS StartCluster();

 protected:

  void DoTearDown() override;

  virtual void SetUpFlags() {}

  std::unique_ptr<CppCassandraDriver> driver_;

  std::unique_ptr<cqlserver::CQLServer> cql_server_;

  typename MiniClusterType::Options mini_cluster_opt_;

 private:

  void SetupClusterOpt();

  CHECKED_STATUS StartCQLServer();

  std::string cql_host_;

  uint16_t cql_port_ = 0;

};

} // namespace yb

#endif // YB_INTEGRATION_TESTS_CQL_TEST_BASE_H