From 38756822825e20eca3b9e01b735946175223d692 Mon Sep 17 00:00:00 2001
From: Yuqian Yang <crupest@crupest.life>
Date: Sat, 7 Mar 2026 20:42:37 +0800
Subject: Refactor stream.

---
 test/base/io/BufferStreamTest.cpp | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 test/base/io/BufferStreamTest.cpp

(limited to 'test/base/io/BufferStreamTest.cpp')

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);
+}
-- 
cgit v1.2.3