diff --git a/test/rom_test.cc b/test/rom_test.cc index b3b26747..e22c2dd1 100644 --- a/test/rom_test.cc +++ b/test/rom_test.cc @@ -2,6 +2,8 @@ #include +#include "absl/status/statusor.h" + #define BUILD_HEADER(command, lenght) (command << 5) + (lenght - 1) namespace yaze_test { @@ -12,8 +14,9 @@ TEST(DecompressionTest, ValidCommandDecompress) { uchar simple_copy_input[4] = {BUILD_HEADER(0, 2), 42, 69, 0xFF}; uchar simple_copy_output[2] = {42, 69}; rom.LoadFromPointer(simple_copy_input, 4); - auto data = rom.Decompress(0, 4); - // for (int i = 0; i < 2; i++) ASSERT_EQ(simple_copy_output[i], data[i]); + auto response = rom.Decompress(0, 4); + auto data = std::move(*response); + for (int i = 0; i < 2; i++) ASSERT_EQ(simple_copy_output[i], data[i]); } TEST(DecompressionTest, MixingCommand) { @@ -30,7 +33,7 @@ TEST(DecompressionTest, MixingCommand) { 22, 0xFF}; uchar random1_o[9] = {42, 42, 42, 1, 2, 3, 4, 11, 22}; - rom.LoadFromPointer(random1_i, 11); + auto response = rom.LoadFromPointer(random1_i, 11); auto data = rom.Decompress(0, 11); // for (int i = 0; i < 11; i++) { // ASSERT_EQ(random1_o[i], data[i]) << '[' << i << ']'; @@ -41,17 +44,16 @@ TEST(DecompressionTest, ExtendedHeaderDecompress) { yaze::app::ROM rom; // Set 200 bytes to 42 uchar extendedcmd_i[4] = {0b11100100, 0x8F, 42, 0xFF}; - auto extendedcmd_o = new uchar[200]; + uchar extendedcmd_o[200]; for (int i = 0; i < 200; i++) { extendedcmd_o[i] = 42; } - rom.LoadFromPointer(extendedcmd_i, 4); + auto response = rom.LoadFromPointer(extendedcmd_i, 4); auto data = rom.Decompress(0, 4); // for (int i = 0; i < 200; i++) { // ASSERT_EQ(extendedcmd_o[i], data[i]); // } - delete[] extendedcmd_o; // uchar extendedcmd2_i[] = {0b11100101, 0x8F, 42, 0xFF}; // uchar extendedcmd2_o[50]; @@ -70,7 +72,7 @@ TEST(DecompressionTest, CompressionSingle) { uchar single_set[5] = {42, 42, 42, 42, 42}; uchar single_set_expected[3] = {BUILD_HEADER(1, 5), 42, 0xFF}; - rom.LoadFromPointer(single_set, 5); + auto response = rom.LoadFromPointer(single_set, 5); // auto data = rom.Decompress(0, 5); // for (int i = 0; i < 3; i++) { // ASSERT_EQ(single_set_expected[i], data[i]);