rocksdb/utilities/blob_db/blob_db_test.cc

//  Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.
//  This source code is licensed under both the GPLv2 (found in the
//  COPYING file in the root directory) and Apache 2.0 License
//  (found in the LICENSE.Apache file in the root directory).

#ifndef ROCKSDB_LITE

#include "utilities/blob_db/blob_db.h"
#include "util/random.h"
#include "util/testharness.h"
#include "util/testutil.h"

namespace rocksdb {
class BlobDBTest : public testing::Test {
 public:
  BlobDBTest() {
    dbname_ = test::TmpDir() + "/blob_db_test";
    Options options;
    options.create_if_missing = true;
    EXPECT_TRUE(NewBlobDB(options, dbname_, &db_).ok());
  }

  ~BlobDBTest() { delete db_; }

  DB* db_;
  std::string dbname_;
};  // class BlobDBTest

TEST_F(BlobDBTest, Basic) {
  WriteOptions wo;
  ReadOptions ro;
  std::string value;

  ASSERT_OK(db_->Put(wo, "foo", "v1"));
  ASSERT_OK(db_->Put(wo, "bar", "v2"));

  ASSERT_OK(db_->Get(ro, "foo", &value));
  ASSERT_EQ("v1", value);
  ASSERT_OK(db_->Get(ro, "bar", &value));
  ASSERT_EQ("v2", value);
}

TEST_F(BlobDBTest, Large) {
  WriteOptions wo;
  ReadOptions ro;
  std::string value1, value2, value3;
  Random rnd(301);

  value1.assign(8999, '1');
  ASSERT_OK(db_->Put(wo, "foo", value1));
  value2.assign(9001, '2');
  ASSERT_OK(db_->Put(wo, "bar", value2));
  test::RandomString(&rnd, 13333, &value3);
  ASSERT_OK(db_->Put(wo, "barfoo", value3));

  std::string value;
  ASSERT_OK(db_->Get(ro, "foo", &value));
  ASSERT_EQ(value1, value);
  ASSERT_OK(db_->Get(ro, "bar", &value));
  ASSERT_EQ(value2, value);
  ASSERT_OK(db_->Get(ro, "barfoo", &value));
  ASSERT_EQ(value3, value);
}

}  //  namespace rocksdb

// A black-box test for the ttl wrapper around rocksdb
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 BlobDB is not supported in ROCKSDB_LITE\n");
  return 0;
}

#endif  // !ROCKSDB_LITE