aboutsummaryrefslogtreecommitdiff
path: root/absl/random/mock_distributions_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'absl/random/mock_distributions_test.cc')
-rw-r--r--absl/random/mock_distributions_test.cc9
1 files changed, 9 insertions, 0 deletions
diff --git a/absl/random/mock_distributions_test.cc b/absl/random/mock_distributions_test.cc
index 917799f0..80df8714 100644
--- a/absl/random/mock_distributions_test.cc
+++ b/absl/random/mock_distributions_test.cc
@@ -19,6 +19,7 @@
#include "gmock/gmock.h"
#include "gtest/gtest.h"
+#include "absl/numeric/int128.h"
#include "absl/random/distributions.h"
#include "absl/random/mocking_bit_gen.h"
#include "absl/random/random.h"
@@ -81,6 +82,14 @@ TEST(MockUniform, OutOfBoundsIsAllowed) {
EXPECT_EQ(absl::Uniform<int>(gen, 1, 100), 0);
}
+TEST(ValidatedMockDistributions, UniformUInt128Works) {
+ absl::random_internal::MockingBitGenImpl<true> gen;
+
+ EXPECT_CALL(absl::MockUniform<absl::uint128>(), Call(gen))
+ .WillOnce(Return(absl::Uint128Max()));
+ EXPECT_EQ(absl::Uniform<absl::uint128>(gen), absl::Uint128Max());
+}
+
TEST(ValidatedMockDistributions, UniformDoubleBoundaryCases) {
absl::random_internal::MockingBitGenImpl<true> gen;