diff --git a/Tests/AK/TestBitStream.cpp b/Tests/AK/TestBitStream.cpp index e67a00659d..cc4cadb74a 100644 --- a/Tests/AK/TestBitStream.cpp +++ b/Tests/AK/TestBitStream.cpp @@ -8,6 +8,8 @@ #include #include +using namespace Test::Randomized; + // Note: This does not do any checks on the internal representation, it just ensures that the behavior of the input and output streams match. TEST_CASE(little_endian_bit_stream_input_output_match) { @@ -129,3 +131,90 @@ TEST_CASE(big_endian_bit_stream_input_output_match) EXPECT_EQ(0b1101001000100001u, result); } } + +RANDOMIZED_TEST_CASE(roundtrip_u8_little_endian) +{ + GEN(n, Gen::unsigned_int(NumericLimits::max())); + + auto memory_stream = make(); + LittleEndianOutputBitStream sut_write { MaybeOwned(*memory_stream) }; + LittleEndianInputBitStream sut_read { MaybeOwned(*memory_stream) }; + + MUST(sut_write.write_bits(n, 8)); + MUST(sut_write.flush_buffer_to_stream()); + auto result = MUST(sut_read.read_bits(8)); + + EXPECT_EQ(result, n); +} + +RANDOMIZED_TEST_CASE(roundtrip_u16_little_endian) +{ + GEN(n, Gen::unsigned_int(NumericLimits::max())); + + auto memory_stream = make(); + LittleEndianOutputBitStream sut_write { MaybeOwned(*memory_stream) }; + LittleEndianInputBitStream sut_read { MaybeOwned(*memory_stream) }; + + MUST(sut_write.write_bits(n, 16)); + MUST(sut_write.flush_buffer_to_stream()); + auto result = MUST(sut_read.read_bits(16)); + + EXPECT_EQ(result, n); +} + +RANDOMIZED_TEST_CASE(roundtrip_u32_little_endian) +{ + GEN(n, Gen::unsigned_int(NumericLimits::max())); + + auto memory_stream = make(); + LittleEndianOutputBitStream sut_write { MaybeOwned(*memory_stream) }; + LittleEndianInputBitStream sut_read { MaybeOwned(*memory_stream) }; + + MUST(sut_write.write_bits(n, 32)); + MUST(sut_write.flush_buffer_to_stream()); + auto result = MUST(sut_read.read_bits(32)); + + EXPECT_EQ(result, n); +} + +RANDOMIZED_TEST_CASE(roundtrip_u8_big_endian) +{ + GEN(n, Gen::unsigned_int(NumericLimits::max())); + + auto memory_stream = make(); + BigEndianOutputBitStream sut_write { MaybeOwned(*memory_stream) }; + BigEndianInputBitStream sut_read { MaybeOwned(*memory_stream) }; + + MUST(sut_write.write_bits(n, 8)); + auto result = MUST(sut_read.read_bits(8)); + + EXPECT_EQ(result, n); +} + +RANDOMIZED_TEST_CASE(roundtrip_u16_big_endian) +{ + GEN(n, Gen::unsigned_int(NumericLimits::max())); + + auto memory_stream = make(); + BigEndianOutputBitStream sut_write { MaybeOwned(*memory_stream) }; + BigEndianInputBitStream sut_read { MaybeOwned(*memory_stream) }; + + MUST(sut_write.write_bits(n, 16)); + auto result = MUST(sut_read.read_bits(16)); + + EXPECT_EQ(result, n); +} + +RANDOMIZED_TEST_CASE(roundtrip_u32_big_endian) +{ + GEN(n, Gen::unsigned_int(NumericLimits::max())); + + auto memory_stream = make(); + BigEndianOutputBitStream sut_write { MaybeOwned(*memory_stream) }; + BigEndianInputBitStream sut_read { MaybeOwned(*memory_stream) }; + + MUST(sut_write.write_bits(n, 32)); + auto result = MUST(sut_read.read_bits(32)); + + EXPECT_EQ(result, n); +}