[go: nahoru, domu]

blob: 70285716f57b84695f592fc1addf4dc169e961da [file] [log] [blame]
// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/unguessable_token.h"
#include <memory>
#include <sstream>
#include <type_traits>
#include "base/hash/hash.h"
#include "base/values.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace base {
void TestSmallerThanOperator(const UnguessableToken& a,
const UnguessableToken& b) {
EXPECT_TRUE(a < b);
EXPECT_FALSE(b < a);
}
TEST(UnguessableTokenTest, VerifyEveryBit) {
absl::optional<UnguessableToken> token = UnguessableToken::Deserialize(1, 2);
ASSERT_TRUE(token.has_value());
uint64_t high = 1;
uint64_t low = 2;
for (uint64_t bit = 1; bit != 0; bit <<= 1) {
uint64_t new_high = high ^ bit;
absl::optional<UnguessableToken> new_token =
UnguessableToken::Deserialize(new_high, low);
ASSERT_TRUE(new_token.has_value());
EXPECT_FALSE(*token == *new_token);
}
for (uint64_t bit = 1; bit != 0; bit <<= 1) {
uint64_t new_low = low ^ bit;
absl::optional<UnguessableToken> new_token =
UnguessableToken::Deserialize(high, new_low);
ASSERT_TRUE(new_token.has_value());
EXPECT_FALSE(*token == *new_token);
}
}
TEST(UnguessableTokenTest, VerifyEqualityOperators) {
// Deserialize is used for testing purposes.
// Use UnguessableToken::Create() in production code instead.
UnguessableToken token = UnguessableToken::Deserialize(1, 2).value();
UnguessableToken same_token = UnguessableToken::Deserialize(1, 2).value();
UnguessableToken diff_token = UnguessableToken::Deserialize(1, 3).value();
UnguessableToken empty_token;
EXPECT_TRUE(token == token);
EXPECT_FALSE(token != token);
EXPECT_TRUE(token == same_token);
EXPECT_FALSE(token != same_token);
EXPECT_FALSE(token == diff_token);
EXPECT_FALSE(diff_token == token);
EXPECT_TRUE(token != diff_token);
EXPECT_TRUE(diff_token != token);
EXPECT_TRUE(empty_token == empty_token);
EXPECT_FALSE(empty_token != empty_token);
for (const UnguessableToken& this_token : {token, same_token, diff_token}) {
EXPECT_FALSE(this_token == empty_token);
EXPECT_TRUE(this_token != empty_token);
}
}
TEST(UnguessableTokenTest, VerifyConstructors) {
UnguessableToken token = UnguessableToken::Create();
EXPECT_FALSE(token.is_empty());
EXPECT_TRUE(token);
UnguessableToken copied_token(token);
EXPECT_TRUE(copied_token);
EXPECT_EQ(token, copied_token);
UnguessableToken uninitialized;
EXPECT_TRUE(uninitialized.is_empty());
EXPECT_FALSE(uninitialized);
EXPECT_TRUE(UnguessableToken().is_empty());
EXPECT_FALSE(UnguessableToken());
}
TEST(UnguessableTokenTest, VerifySerialization) {
UnguessableToken token = UnguessableToken::Create();
uint64_t high = token.GetHighForSerialization();
uint64_t low = token.GetLowForSerialization();
EXPECT_TRUE(high);
EXPECT_TRUE(low);
absl::optional<UnguessableToken> Deserialized =
UnguessableToken::Deserialize(high, low);
ASSERT_TRUE(Deserialized.has_value());
EXPECT_EQ(token, *Deserialized);
}
// Common case (~88% of the time) - no leading zeroes in high_ nor low_.
TEST(UnguessableTokenTest, VerifyToString1) {
UnguessableToken token =
UnguessableToken::Deserialize(0x1234567890ABCDEF, 0xFEDCBA0987654321)
.value();
std::string expected = "1234567890ABCDEFFEDCBA0987654321";
EXPECT_EQ(expected, token.ToString());
std::string expected_stream = "(1234567890ABCDEFFEDCBA0987654321)";
std::stringstream stream;
stream << token;
EXPECT_EQ(expected_stream, stream.str());
}
// Less common case - leading zeroes in high_ or low_ (testing with both).
TEST(UnguessableTokenTest, VerifyToString2) {
UnguessableToken token = UnguessableToken::Deserialize(0x123, 0xABC).value();
std::string expected = "00000000000001230000000000000ABC";
EXPECT_EQ(expected, token.ToString());
std::string expected_stream = "(00000000000001230000000000000ABC)";
std::stringstream stream;
stream << token;
EXPECT_EQ(expected_stream, stream.str());
}
TEST(UnguessableTokenTest, VerifyToStringUniqueness) {
const UnguessableToken token1 =
UnguessableToken::Deserialize(0x0000000012345678, 0x0000000123456789)
.value();
const UnguessableToken token2 =
UnguessableToken::Deserialize(0x0000000123456781, 0x0000000023456789)
.value();
EXPECT_NE(token1.ToString(), token2.ToString());
}
TEST(UnguessableTokenTest, VerifyDeserializeZeroes) {
absl::optional<UnguessableToken> token = UnguessableToken::Deserialize(0, 0);
EXPECT_FALSE(token.has_value());
}
TEST(UnguessableTokenTest, VerifySmallerThanOperator) {
// Deserialize is used for testing purposes.
// Use UnguessableToken::Create() in production code instead.
{
SCOPED_TRACE("a.low < b.low and a.high == b.high.");
TestSmallerThanOperator(UnguessableToken::Deserialize(0, 1).value(),
UnguessableToken::Deserialize(0, 5).value());
}
{
SCOPED_TRACE("a.low == b.low and a.high < b.high.");
TestSmallerThanOperator(UnguessableToken::Deserialize(1, 0).value(),
UnguessableToken::Deserialize(5, 0).value());
}
{
SCOPED_TRACE("a.low < b.low and a.high < b.high.");
TestSmallerThanOperator(UnguessableToken::Deserialize(1, 1).value(),
UnguessableToken::Deserialize(5, 5).value());
}
{
SCOPED_TRACE("a.low > b.low and a.high < b.high.");
TestSmallerThanOperator(UnguessableToken::Deserialize(1, 10).value(),
UnguessableToken::Deserialize(10, 1).value());
}
}
TEST(UnguessableTokenTest, VerifyHash) {
UnguessableToken token = UnguessableToken::Create();
EXPECT_EQ(base::HashInts64(token.GetHighForSerialization(),
token.GetLowForSerialization()),
UnguessableTokenHash()(token));
}
TEST(UnguessableTokenTest, VerifyBasicUniqueness) {
EXPECT_NE(UnguessableToken::Create(), UnguessableToken::Create());
UnguessableToken token = UnguessableToken::Create();
EXPECT_NE(token.GetHighForSerialization(), token.GetLowForSerialization());
}
}