PrebuiltStringMap.cpp

//===--- PrebuiltStringMap.cpp - Unit tests for PrebuiltStringMap ---------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

#include "swift/Runtime/PrebuiltStringMap.h"
#include "gtest/gtest.h"

static bool stringIsNull(const char *str) { return str == nullptr; }

TEST(PrebuiltStringMapTest, basic) {
  auto testOnce = [&](unsigned testEntryCount) {
    std::vector<std::pair<std::string, unsigned>> testVector;
    testVector.reserve(testEntryCount);
    for (unsigned i = 0; i < testEntryCount; i++) {
      std::string key;
      for (unsigned j = 0; j < i; j++)
        key += std::to_string(j);
      testVector.push_back({key, i});
    }

    using Map = swift::PrebuiltStringMap<const char *, unsigned, stringIsNull>;

    unsigned mapSize = testEntryCount * 4 / 3;
    void *mapAllocation = calloc(1, Map::byteSize(mapSize));
    Map *map = new (mapAllocation) Map(mapSize);

    for (auto &[key, value] : testVector) {
      const char *keyCStr = key.c_str();
      auto *element = map->insert(keyCStr);
      EXPECT_NE(element, nullptr);

      element->key = keyCStr;
      element->value = value;
    }

    for (auto &[key, value] : testVector) {
      const char *keyCStr = key.c_str();
      auto *element = map->find(keyCStr);
      EXPECT_NE(element, nullptr);

      EXPECT_EQ(element->key, key);
      EXPECT_EQ(element->value, value);
    }

    // Test the interface that takes a string and length without NUL
    // termination.
    for (auto &[key, value] : testVector) {
      auto keyCopy = key;
      keyCopy += "xyz";
      const char *keyCStr = keyCopy.c_str();
      auto *element = map->find(keyCStr, key.size());
      EXPECT_NE(element, nullptr);

      EXPECT_EQ(element->key, key);
      EXPECT_EQ(element->value, value);
    }

    free(mapAllocation);
  };

  testOnce(10);
  testOnce(100);
  testOnce(1000);
}