YugabyteDB (2.13.0.0-b42, bfc6a6643e7399ac8a0e81d06a3ee6d6571b33ab)

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

//

// The following only applies to changes made to this file as part of YugaByte development.

//

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

//

#include <string>

#include <gtest/gtest.h>

#include "yb/common/placement_info.h"

#include "yb/util/test_macros.h"

namespace yb {

// Test the tablespace info parsing.

TEST(PlacementInfoTest, TestTablespaceJsonProcessing) {

  // Variables to be used throughout the test.

  const string& valid_json =

      "{\"num_replicas\":3,\"placement_blocks\":"

      "[{\"cloud\":\"c1\",\"region\":\"r1\",\"zone\":\"z1\",\"min_num_replicas\":2},"

      "{\"cloud\":\"c2\",\"region\":\"r2\",\"zone\":\"z2\",\"min_num_replicas\":1}]}";

  QLValuePB option, invalid_option;

  option.set_string_value("replica_placement=" + valid_json);

  invalid_option.set_string_value("read_replica_placement=" + valid_json);

  vector<QLValuePB> options;

  // Negative tests.

  // 1. Empty input.

  ASSERT_NOK(PlacementInfoConverter::FromQLValue(options));

  // 2. Invalid number of options.

  options.push_back(option);

  options.push_back(invalid_option);

  ASSERT_NOK(PlacementInfoConverter::FromQLValue(options));

  // 3. Invalid option name.

  options.clear();

  options.push_back(invalid_option);

  ASSERT_NOK(PlacementInfoConverter::FromQLValue(options));

  // 4. Empty json.

  options.clear();

  QLValuePB opt_empty_value;

  opt_empty_value.set_string_value("replica_placement=[{}]");

  options.emplace_back(opt_empty_value);

  ASSERT_NOK(PlacementInfoConverter::FromQLValue(options));

  // 5. Missing num_replicas field.

  options.clear();

  QLValuePB invalid_json_option;

  invalid_json_option.set_string_value("replica_placement={\"placement_blocks\":"

      "[{\"cloud\":\"c1\",\"region\":\"r1\",\"zone\":\"z1\",\"min_num_replicas\":3}]}");

  options.emplace_back(invalid_json_option);

  ASSERT_NOK(PlacementInfoConverter::FromQLValue(options));

  // 6. Invalid value for num_replicas field.

  options.clear();

  invalid_json_option.set_string_value(

      "replica_placement={\"num_replicas\":\"abc\",\"placement_blocks\":"

      "[{\"cloud\":\"c1\",\"region\":\"r1\",\"zone\":\"z1\",\"min_num_replicas\":3}]}");

  options.emplace_back(invalid_json_option);

  ASSERT_NOK(PlacementInfoConverter::FromQLValue(options));

  // 7. Missing placement blocks field.

  options.clear();

  invalid_json_option.set_string_value("replica_placement={\"num_replicas\":3}");

  options.emplace_back(invalid_json_option);

  ASSERT_NOK(PlacementInfoConverter::FromQLValue(options));

  // 8. Missing keys in placement blocks.

  options.clear();

  invalid_json_option.set_string_value(

      "replica_placement={\"num_replicas\":\"abc\",\"placement_blocks\":"

      "[{\"cloud\":\"c1\",\"region\":\"r1\",\"zone\":\"z1\"}]}");

  options.emplace_back(invalid_json_option);

  ASSERT_NOK(PlacementInfoConverter::FromQLValue(options));

  // 9. Invalid format for "min_num_replicas".

  options.clear();

  invalid_json_option.set_string_value(

        "replica_placement={\"num_replicas\":3,\"placement_blocks\":"

        "[{\"cloud\":\"c1\",\"region\":\"r1\",\"zone\":\"z1\",\"min_num_replicas\":\"abc\"}]}");

  options.emplace_back(invalid_json_option);

  ASSERT_NOK(PlacementInfoConverter::FromQLValue(options));

  // 10. Invalid json.

  options.clear();

  invalid_json_option.set_string_value("replica_placement=["

      "{\"cloud\":\"c1\",\"region\":\"r1\",\"zone\":\"z1\",\"min_number_of_replica");

  options.emplace_back(invalid_json_option);

  ASSERT_NOK(PlacementInfoConverter::FromQLValue(options));

  // 11. Test whether total replication factor is populated correctly.

  options.clear();

  options.push_back(option);

  PlacementInfoConverter::Placement result = EXPECT_RESULT(

      PlacementInfoConverter::FromQLValue(options));

  const auto placement_infos = result.placement_infos;

  ASSERT_EQ(result.num_replicas, 3);

  // 12. Test whether the cloud/region/zone information has been populated correctly.

  ASSERT_EQ(placement_infos.size(), 2);

  for (auto& placement_block : placement_infos) {

    if (placement_block.cloud == "c1") {

      ASSERT_EQ(placement_block.region, "r1");

      ASSERT_EQ(placement_block.zone, "z1");

      ASSERT_EQ(placement_block.min_num_replicas, 2);

      continue;

    }

    ASSERT_EQ(placement_block.cloud, "c2");

    ASSERT_EQ(placement_block.region, "r2");

    ASSERT_EQ(placement_block.zone, "z2");

    ASSERT_EQ(placement_block.min_num_replicas, 1);

  }

}

} // namespace yb