//===----------------------------------------------------------------------===//
//
//                         BusTub
//
// trie_debug_test.cpp
//
// Identification: test/primer/trie_debug_test.cpp
//
// Copyright (c) 2015-2025, Carnegie Mellon University Database Group
//
//===----------------------------------------------------------------------===//

#include <fmt/format.h>
#include <zipfian_int_distribution.h>
#include <bitset>
#include <functional>
#include <numeric>
#include <optional>
#include <random>
#include <thread>  // NOLINT

#include "common/exception.h"
#include "gtest/gtest.h"
#include "primer/trie.h"
#include "primer/trie_answer.h"
#include "trie_debug_answer.h"  // NOLINT

namespace bustub {

TEST(TrieDebugger, TestCase) {
  std::mt19937_64 gen(23333);
  zipfian_int_distribution<uint32_t> dis(0, 1000);

  auto trie = Trie();
  for (uint32_t i = 0; i < 100; i++) {
    std::string key = fmt::format("{}", dis(gen));
    auto value = dis(gen);
    switch (i) {
      // Test the first three values from the random generator.
      case 0:
        ASSERT_EQ(value, 128) << "Random generator not portable, please post on Piazza for help.";
        break;
      case 1:
        ASSERT_EQ(value, 16) << "Random generator not portable, please post on Piazza for help.";
        break;
      case 2:
        ASSERT_EQ(value, 41) << "Random generator not portable, please post on Piazza for help.";
        break;
      default:
        break;
    }
    trie = trie.Put<uint32_t>(key, value);
  }

  // Put a breakpoint here.

  // (1) How many children nodes are there on the root?
  // Replace `CASE_1_YOUR_ANSWER` in `trie_answer.h` with the correct answer.
  if (CASE_1_YOUR_ANSWER != Case1CorrectAnswer()) {
    ASSERT_TRUE(false) << "case 1 not correct";
  }

  // (2) How many children nodes are there on the node of prefix `9`?
  // Replace `CASE_2_YOUR_ANSWER` in `trie_answer.h` with the correct answer.
  if (CASE_2_YOUR_ANSWER != Case2CorrectAnswer()) {
    ASSERT_TRUE(false) << "case 2 not correct";
  }

  // (3) What's the value for `969`?
  // Replace `CASE_3_YOUR_ANSWER` in `trie_answer.h` with the correct answer.
  if (CASE_3_YOUR_ANSWER != Case3CorrectAnswer()) {
    ASSERT_TRUE(false) << "case 3 not correct";
  }
}

}  // namespace bustub