// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/common/pseudonymization_salt.h"

#include <tuple>

#include "base/test/gtest_util.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace content {

class PseudonymizationSaltTest : public ::testing::Test {
  void SetUp() override { ResetSaltForTesting(); }
};

using PseudonymizationSaltDeathTest = PseudonymizationSaltTest;

TEST_F(PseudonymizationSaltTest, SetAndGetSalt) {
  SetPseudonymizationSalt(0xDEADBEEF);
  uint32_t salt = GetPseudonymizationSalt();
  EXPECT_EQ(salt, 0xDEADBEEF);
}

// Early during process startup (before Browser process reaches salt
// initialization;  or before a Child process receives the salt over IPC), the
// pseudonymization salt is not initialized and is invalid - this test verifies
// presence of a DCHECK in GetPseudonymizationSalt that verifies that
// GetPseudonymizationSalt is only called after salt initialization.
//
// See also https://crbug.com/1339992 which should move salt initialization
// much earlier in Renderer processes (and other Child and Utility processes).
TEST_F(PseudonymizationSaltDeathTest, GetEmptySalt) {
  EXPECT_DCHECK_DEATH({ std::ignore = GetPseudonymizationSalt(); });
}

TEST_F(PseudonymizationSaltDeathTest, SetTwoDifferentSalts) {
  SetPseudonymizationSalt(0xDEADBEEF);
  EXPECT_DCHECK_DEATH({ SetPseudonymizationSalt(0xBEEFDEAD); });
}

TEST_F(PseudonymizationSaltDeathTest, SetZeroSalt) {
  EXPECT_DCHECK_DEATH({ SetPseudonymizationSalt(0x0u); });
}

TEST_F(PseudonymizationSaltTest, SetTwoSaltsAllowed) {
  SetPseudonymizationSalt(0xDEADBEEF);
  SetPseudonymizationSalt(0xDEADBEEF);
  uint32_t salt = GetPseudonymizationSalt();
  EXPECT_EQ(salt, 0xDEADBEEF);
}

}  // namespace content