YugabyteDB (2.13.0.0-b42, bfc6a6643e7399ac8a0e81d06a3ee6d6571b33ab)

//  Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.

//  This source code is licensed under the BSD-style license found in the

//  LICENSE file in the root directory of this source tree. An additional grant

//  of patent rights can be found in the PATENTS file in the same directory.

//

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

//

#ifndef ROCKSDB_LITE

#include "yb/rocksdb/tools/ldb_cmd.h"

#include "yb/rocksdb/util/testharness.h"

#include "yb/rocksdb/env.h"

#include "yb/rocksdb/db/db_test_util.h"

#include "yb/util/test_macros.h"

DECLARE_bool(TEST_exit_on_finish);

const string kDbName = "/lbd_cmd_test";

const string kKeyFileOption = "--key_file=" + rocksdb::DBTestBase::kKeyId + ":" +

                              rocksdb::DBTestBase::kKeyFile;

class LdbCmdTest : public rocksdb::DBTestBase, public testing::WithParamInterface<bool> {

 public:

  LdbCmdTest() : rocksdb::DBTestBase(kDbName, GetParam()) {}

  yb::Status WriteRecordsAndFlush() {

    RETURN_NOT_OK(Put("k1", "v1"));

    RETURN_NOT_OK(Put("k2", "v2"));

    return Flush();

  }

  std::vector<std::string> CreateArgs(std::string command_name) {

    std::vector<std::string> args;

    args.push_back("./ldb");

    args.push_back("--db=" + dbname_);

    args.push_back(command_name);

    if (GetParam()) {

      args.push_back(kKeyFileOption);

    }

    return args;

  }

  void RunLdb(const std::vector<std::string>& args) {

    std::vector<char *> args_ptr;

    args_ptr.resize(args.size());

    std::transform(args.begin(), args.end(), args_ptr.begin(), [](const std::string& str) {

      return const_cast<char*>(str.c_str());

    });

    rocksdb::LDBTool tool;

    ASSERT_NO_FATALS(tool.Run(static_cast<int>(args_ptr.size()), args_ptr.data()));

  }

};

INSTANTIATE_TEST_CASE_P(EncryptionEnabled, LdbCmdTest, ::testing::Bool());

TEST_P(LdbCmdTest, HexToString) {

  // map input to expected outputs.

  map<string, vector<int>> inputMap = {

      {"0x7", {7}},         {"0x5050", {80, 80}}, {"0xFF", {-1}},

      {"0x1234", {18, 52}}, {"0xaa", {-86}}, {"0x123", {18, 3}},

  };

  for (const auto& inPair : inputMap) {

    auto actual = rocksdb::LDBCommand::HexToString(inPair.first);

    auto expected = inPair.second;

    for (unsigned int i = 0; i < actual.length(); i++) {

      ASSERT_EQ(expected[i], static_cast<int>(actual[i]));

    }

  }

}

TEST_P(LdbCmdTest, HexToStringBadInputs) {

  const vector<string> badInputs = {

      "0xZZ", "123", "0xx5", "0x11G", "Ox12", "0xT", "0x1Q1",

  };

  for (const auto& badInput : badInputs) {

    try {

      rocksdb::LDBCommand::HexToString(badInput);

      std::cerr << "Should fail on bad hex value: " << badInput << "\n";

      FAIL();

    } catch (...) {

    }

  }

}

TEST_P(LdbCmdTest, Scan) {

  FLAGS_TEST_exit_on_finish = false;

  ASSERT_OK(WriteRecordsAndFlush());

  auto args = CreateArgs("scan");

  args.push_back("--only_verify_checksums");

  ASSERT_NO_FATALS(RunLdb(args));

}

TEST_P(LdbCmdTest, CheckConsistency) {

  FLAGS_TEST_exit_on_finish = false;

  ASSERT_OK(WriteRecordsAndFlush());

  auto args = CreateArgs("checkconsistency");

  ASSERT_NO_FATALS(RunLdb(args));

}

int main(int argc, char** argv) {

  ::testing::InitGoogleTest(&argc, argv);

  return RUN_ALL_TESTS();

}

#else

#include <stdio.h>

int main(int argc, char** argv) {

  fprintf(stderr, "SKIPPED as LDBCommand is not supported in ROCKSDB_LITE\n");

  return 0;

}

#endif  // ROCKSDB_LITE