#include "test_helpers.hpp"
#include "duckdb/common/types/bignum.hpp"
#include "catch.hpp"

using namespace duckdb;
using namespace std;

TEST_CASE("Test Bignum::FromByteArray", "[api]") {
	{ // 0
		uint8_t data[] {0};
		idx_t size = 1;
		bool is_negative = false;
		auto str = Bignum::FromByteArray(data, size, is_negative);
		REQUIRE(str.size() == size + 3);
		REQUIRE(uint8_t(str[0]) == 0x80);
		REQUIRE(uint8_t(str[1]) == 0x00);
		REQUIRE(uint8_t(str[2]) == uint8_t(size));
		for (idx_t i = 0; i < size; i++) {
			REQUIRE(uint8_t(str[3 + i]) == data[i]);
		}
	}
	{ // -1
		uint8_t data[] {1};
		idx_t size = 1;
		bool is_negative = true;
		auto str = Bignum::FromByteArray(data, size, is_negative);
		REQUIRE(str.size() == size + 3);
		REQUIRE(uint8_t(str[0]) == 0x7f);
		REQUIRE(uint8_t(str[1]) == 0xff);
		REQUIRE(uint8_t(str[2]) == uint8_t(~size));
		for (idx_t i = 0; i < size; i++) {
			REQUIRE(uint8_t(str[3 + i]) == uint8_t(~data[i]));
		}
	}
	{ // max bignum == max double == 2^1023 * (1 + (1 − 2^−52)) == 2^1024 - 2^971 ==
		// 179769313486231570814527423731704356798070567525844996598917476803157260780028538760589558632766878171540458953514382464234321326889464182768467546703537516986049910576551282076245490090389328944075868508455133942304583236903222948165808559332123348274797826204144723168738177180919299881250404026184124858368
		uint8_t data[] {
		    // little endian
		    0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x07, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		};
		idx_t size = 128;
		bool is_negative = false;
		auto str = Bignum::FromByteArray(data, size, is_negative);
		REQUIRE(str.size() == size + 3);
		REQUIRE(uint8_t(str[0]) == 0x80);
		REQUIRE(uint8_t(str[1]) == 0x00);
		REQUIRE(uint8_t(str[2]) == uint8_t(size));
		for (idx_t i = 0; i < size; i++) {
			REQUIRE(uint8_t(str[3 + i]) == data[i]);
		}
	}
	{ // min bignum == min double == -(2^1023 * (1 + (1 − 2^−52))) == -(2^1024 - 2^971) ==
		// -179769313486231570814527423731704356798070567525844996598917476803157260780028538760589558632766878171540458953514382464234321326889464182768467546703537516986049910576551282076245490090389328944075868508455133942304583236903222948165808559332123348274797826204144723168738177180919299881250404026184124858368
		uint8_t data[] {
		    // little endian (absolute value)
		    0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x07, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
		};
		idx_t size = 128;
		bool is_negative = true;
		auto str = Bignum::FromByteArray(data, size, is_negative);
		REQUIRE(str.size() == size + 3);
		REQUIRE(uint8_t(str[0]) == 0x7f);
		REQUIRE(uint8_t(str[1]) == 0xff);
		REQUIRE(uint8_t(str[2]) == uint8_t(~size));
		for (idx_t i = 0; i < size; i++) {
			REQUIRE(uint8_t(str[3 + i]) == uint8_t(~data[i]));
		}
	}
}