Refactor stream.

3 files changed, 101 insertions, 0 deletions

diff --git a/test/base/io/AutoReadStreamTest.cpp b/test/base/io/AutoReadStreamTest.cpp
new file mode 100644
index 00000000..e95b7c24
--- /dev/null
+++ b/test/base/io/AutoReadStreamTest.cpp
@@ -0,0 +1,32 @@
+#include "cru/base/io/AutoReadStream.h"
+#include "cru/base/io/MemoryStream.h"
+
+#include <catch2/catch_test_macros.hpp>
+
+#include <algorithm>
+#include <cstddef>
+#include <ranges>
+
+TEST_CASE("AutoReadStream should work.", "[io][stream]") {
+  using namespace cru::io;
+
+  const int size = 100;
+  std::vector<std::byte> buffer(size);
+  buffer[1] = std::byte(0xf0);
+  buffer[2] = std::byte(0x0f);
+
+  MemoryStream underlying_stream(buffer.data(), buffer.size(), true);
+  AutoReadStream stream(&underlying_stream, true, false);
+
+  REQUIRE(stream.CanRead());
+  REQUIRE(!stream.CanWrite());
+
+  std::vector<std::byte> buffer2(size * 2);
+
+  auto read = stream.Read(buffer2.data(), buffer2.size());
+  auto read2 = stream.Read(buffer2.data(), size, 1);
+
+  REQUIRE(read == size);
+  REQUIRE(std::ranges::equal(buffer, buffer2 | std::views::take(size)));
+  REQUIRE(read2 == Stream::kEOF);
+}
diff --git a/test/base/io/BufferStreamTest.cpp b/test/base/io/BufferStreamTest.cpp
new file mode 100644
index 00000000..70780b48
--- /dev/null
+++ b/test/base/io/BufferStreamTest.cpp
@@ -0,0 +1,39 @@
+#include "cru/base/io/BufferStream.h"
+
+#include <catch2/catch_test_macros.hpp>
+
+#include <algorithm>
+#include <cstddef>
+#include <ranges>
+#include <thread>
+
+TEST_CASE("BufferStream should work.", "[io][stream]") {
+  using namespace cru::io;
+
+  const int size = 100;
+  std::vector<std::byte> buffer(size);
+  buffer[1] = std::byte(0xf0);
+  buffer[2] = std::byte(0x0f);
+
+  BufferStream stream;
+
+  std::vector<std::byte> buffer2(size * 2);
+  cru::Index read, read2;
+
+  std::thread read_thread([&] {
+    read = stream.Read(buffer2.data(), buffer2.size());
+    read2 = stream.Read(buffer2.data(), size, 1);
+  });
+
+  std::thread write_thread([&] {
+    stream.Write(buffer.data(), buffer.size());
+    stream.WriteEof();
+  });
+
+  read_thread.join();
+  write_thread.join();
+
+  REQUIRE(read == size);
+  REQUIRE(std::ranges::equal(buffer, buffer2 | std::views::take(size)));
+  REQUIRE(read2 == Stream::kEOF);
+}
diff --git a/test/base/io/MemoryStreamTest.cpp b/test/base/io/MemoryStreamTest.cpp
new file mode 100644
index 00000000..684971b1
--- /dev/null
+++ b/test/base/io/MemoryStreamTest.cpp
@@ -0,0 +1,30 @@
+#include "cru/base/io/MemoryStream.h"
+#include "cru/base/io/Stream.h"
+
+#include <catch2/catch_test_macros.hpp>
+
+#include <algorithm>
+#include <cstddef>
+#include <ranges>
+
+TEST_CASE("MemoryStream should work.", "[io][stream]") {
+  using namespace cru::io;
+  const int size = 100;
+  std::vector<std::byte> buffer(size);
+
+  buffer[1] = std::byte(0xf0);
+  buffer[2] = std::byte(0x0f);
+
+  MemoryStream stream(buffer.data(), size, true);
+
+  REQUIRE(stream.CanRead());
+  REQUIRE(!stream.CanWrite());
+
+  std::vector<std::byte> buffer2(size * 2);
+
+  auto read = stream.Read(buffer2.data(), buffer2.size());
+  auto read2 = stream.Read(buffer2.data(), 1);
+  REQUIRE(read == size);
+  REQUIRE(std::ranges::equal(buffer, buffer2 | std::views::take(size)));
+  REQUIRE(read2 == Stream::kEOF);
+}