From dcdfb9fc7e008e2473651d3c3150caeb2dec5dbc Mon Sep 17 00:00:00 2001
From: Alex Beregszaszi <alex@rtfs.hu>
Date: Tue, 4 May 2021 23:00:16 +0100
Subject: [PATCH] cpp: Add support for _uin256be literals

Only extending the literal tests, because the other tests are covered
via bytes32 tests given uint256be is an alias.
---
 include/evmc/evmc.hpp       |  7 +++++++
 test/unittests/cpp_test.cpp | 22 ++++++++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/include/evmc/evmc.hpp b/include/evmc/evmc.hpp
index 6d0bac9f2..a880018c0 100644
--- a/include/evmc/evmc.hpp
+++ b/include/evmc/evmc.hpp
@@ -332,6 +332,13 @@ constexpr bytes32 operator""_bytes32() noexcept
 {
     return internal::from_literal<bytes32, c...>();
 }
+
+/// Literal for evmc::uint256be.
+template <char... c>
+constexpr uint256be operator""_uint256be() noexcept
+{
+    return internal::from_literal<uint256be, c...>();
+}
 }  // namespace literals
 
 using namespace literals;
diff --git a/test/unittests/cpp_test.cpp b/test/unittests/cpp_test.cpp
index 94b1d398c..aba868223 100644
--- a/test/unittests/cpp_test.cpp
+++ b/test/unittests/cpp_test.cpp
@@ -387,6 +387,28 @@ TEST(cpp, bytes32_from_uint)
               0x000000000000000000000000000000000000000000000000c1c2c3c4c5c6c7c8_bytes32);
 }
 
+TEST(cpp, uint256be_from_uint)
+{
+    using evmc::uint256be;
+    using evmc::operator""_uint256be;
+
+    static_assert(uint256be{0} == uint256be{}, "");
+    static_assert(uint256be{3}.bytes[31] == 3, "");
+    static_assert(uint256be{0xfe00000000000000}.bytes[24] == 0xfe, "");
+
+    EXPECT_EQ(uint256be{0}, uint256be{});
+    EXPECT_EQ(uint256be{0x01},
+              0x0000000000000000000000000000000000000000000000000000000000000001_uint256be);
+    EXPECT_EQ(uint256be{0xff},
+              0x00000000000000000000000000000000000000000000000000000000000000ff_uint256be);
+    EXPECT_EQ(uint256be{0x500},
+              0x0000000000000000000000000000000000000000000000000000000000000500_uint256be);
+    EXPECT_EQ(uint256be{0x8000000000000000},
+              0x0000000000000000000000000000000000000000000000008000000000000000_uint256be);
+    EXPECT_EQ(uint256be{0xc1c2c3c4c5c6c7c8},
+              0x000000000000000000000000000000000000000000000000c1c2c3c4c5c6c7c8_uint256be);
+}
+
 TEST(cpp, address_from_uint)
 {
     using evmc::address;