icing/store/dynamic-trie-key-mapper_test.cc

// Copyright (C) 2019 Google LLC
//
// 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 "icing/store/dynamic-trie-key-mapper.h"

#include <string>

#include "icing/text_classifier/lib3/utils/base/status.h"
#include "gmock/gmock.h"
#include "gtest/gtest.h"
#include "icing/file/filesystem.h"
#include "icing/store/document-id.h"
#include "icing/testing/common-matchers.h"
#include "icing/testing/tmp-directory.h"

namespace icing {
namespace lib {

namespace {

constexpr int kMaxDynamicTrieKeyMapperSize = 3 * 1024 * 1024;  // 3 MiB

class DynamicTrieKeyMapperTest : public testing::Test {
 protected:
  void SetUp() override { base_dir_ = GetTestTempDir() + "/key_mapper"; }

  void TearDown() override {
    filesystem_.DeleteDirectoryRecursively(base_dir_.c_str());
  }

  std::string base_dir_;
  Filesystem filesystem_;
};

TEST_F(DynamicTrieKeyMapperTest, InvalidBaseDir) {
  EXPECT_THAT(DynamicTrieKeyMapper<DocumentId>::Create(
                  filesystem_, "/dev/null", kMaxDynamicTrieKeyMapperSize),
              StatusIs(libtextclassifier3::StatusCode::INTERNAL));
}

TEST_F(DynamicTrieKeyMapperTest, NegativeMaxKeyMapperSizeReturnsInternalError) {
  EXPECT_THAT(
      DynamicTrieKeyMapper<DocumentId>::Create(filesystem_, base_dir_, -1),
      StatusIs(libtextclassifier3::StatusCode::INTERNAL));
}

TEST_F(DynamicTrieKeyMapperTest, TooLargeMaxKeyMapperSizeReturnsInternalError) {
  EXPECT_THAT(DynamicTrieKeyMapper<DocumentId>::Create(
                  filesystem_, base_dir_, std::numeric_limits<int>::max()),
              StatusIs(libtextclassifier3::StatusCode::INTERNAL));
}

}  // namespace

}  // namespace lib
}  // namespace icing