106 lines
2.2 KiB
C++
106 lines
2.2 KiB
C++
/*
|
|
SPDX-License-Identifier: MIT
|
|
Copyright © 2023 Amebis
|
|
*/
|
|
|
|
#include "pch.h"
|
|
|
|
using namespace std;
|
|
using namespace stdex;
|
|
using namespace stdex::stream;
|
|
using namespace Microsoft::VisualStudio::CppUnitTestFramework;
|
|
|
|
namespace UnitTests
|
|
{
|
|
TEST_CLASS(stream)
|
|
{
|
|
public:
|
|
TEST_METHOD(async)
|
|
{
|
|
constexpr size_t total = 1000;
|
|
memory_file source(mul(total, sizeof(size_t)));
|
|
{
|
|
async_writer<70> writer(source);
|
|
for (size_t i = 0; i < total; ++i) {
|
|
Assert::IsTrue(writer.ok());
|
|
writer << i;
|
|
}
|
|
}
|
|
Assert::AreEqual<stdex::stream::fpos_t>(0, source.seekbeg(0));
|
|
{
|
|
async_reader<50> reader(source);
|
|
size_t x;
|
|
for (size_t i = 0; i < total; ++i) {
|
|
reader >> x;
|
|
Assert::IsTrue(reader.ok());
|
|
Assert::AreEqual(i, x);
|
|
}
|
|
reader >> x;
|
|
Assert::IsFalse(reader.ok());
|
|
}
|
|
}
|
|
|
|
TEST_METHOD(replicator)
|
|
{
|
|
constexpr size_t total = 1000;
|
|
|
|
memory_file f1(mul(total, sizeof(size_t)));
|
|
|
|
sys_string filename2, filename3;
|
|
filename2 = filename3 = temp_path();
|
|
filename2 += _T("stdex-stream-replicator-2.tmp");
|
|
file f2(
|
|
filename2.c_str(),
|
|
mode_for_reading | mode_for_writing | mode_create | mode_binary);
|
|
|
|
filename3 += _T("stdex-stream-replicator-3.tmp");
|
|
cached_file f3(
|
|
filename3.c_str(),
|
|
mode_for_reading | mode_for_writing | mode_create | mode_binary,
|
|
128);
|
|
|
|
{
|
|
stdex::stream::replicator writer;
|
|
buffer f2_buf(f2, 0, 32);
|
|
writer.push_back(&f1);
|
|
writer.push_back(&f2_buf);
|
|
writer.push_back(&f3);
|
|
for (size_t i = 0; i < total; ++i) {
|
|
Assert::IsTrue(writer.ok());
|
|
writer << i;
|
|
}
|
|
}
|
|
|
|
f1.seekbeg(0);
|
|
f2.seekbeg(0);
|
|
f3.seekbeg(0);
|
|
{
|
|
buffer f2_buf(f2, 64, 0);
|
|
size_t x;
|
|
for (size_t i = 0; i < total; ++i) {
|
|
f1 >> x;
|
|
Assert::IsTrue(f1.ok());
|
|
Assert::AreEqual(i, x);
|
|
f2_buf >> x;
|
|
Assert::IsTrue(f2_buf.ok());
|
|
Assert::AreEqual(i, x);
|
|
f3 >> x;
|
|
Assert::IsTrue(f3.ok());
|
|
Assert::AreEqual(i, x);
|
|
}
|
|
f1 >> x;
|
|
Assert::IsFalse(f1.ok());
|
|
f2_buf >> x;
|
|
Assert::IsFalse(f2_buf.ok());
|
|
f3 >> x;
|
|
Assert::IsFalse(f3.ok());
|
|
}
|
|
|
|
f2.close();
|
|
std::filesystem::remove(filename2);
|
|
f3.close();
|
|
std::filesystem::remove(filename3);
|
|
}
|
|
};
|
|
}
|