Simon Rozman
6aa63f3bd6
We need this for a rare use-case, so maintaining three flavors of diagstream is too expensive. Furthermore, the buffering was removed for simplicity. Signed-off-by: Simon Rozman <simon@rozman.si>
104 lines
2.6 KiB
C++
104 lines
2.6 KiB
C++
/*
|
|
SPDX-License-Identifier: MIT
|
|
Copyright © 2023 Amebis
|
|
*/
|
|
|
|
#include "pch.h"
|
|
|
|
using namespace std;
|
|
using namespace stdex;
|
|
using namespace Microsoft::VisualStudio::CppUnitTestFramework;
|
|
|
|
namespace UnitTests
|
|
{
|
|
TEST_CLASS(ios)
|
|
{
|
|
public:
|
|
TEST_METHOD(fstream)
|
|
{
|
|
WCHAR path[MAX_PATH];
|
|
ExpandEnvironmentStringsW(L"%windir%\\notepad.exe", path, _countof(path));
|
|
stdex::fstream f(path, ios_base::in | ios_base::binary);
|
|
Assert::IsTrue(f.good());
|
|
Assert::IsTrue(f.mtime() < chrono::system_clock::now());
|
|
}
|
|
|
|
TEST_METHOD(isharedstrstream)
|
|
{
|
|
static const char data[] = "\xde\xad\xca\xfe";
|
|
stdex::isharedstrstream f(data, _countof(data));
|
|
Assert::IsTrue(f.good());
|
|
|
|
char val;
|
|
f.read(&val, 1);
|
|
Assert::IsTrue(f.gcount() == 1);
|
|
Assert::IsTrue(f.good());
|
|
Assert::IsTrue(val == data[0]);
|
|
f.read(&val, 1);
|
|
Assert::IsTrue(f.gcount() == 1);
|
|
Assert::IsTrue(f.good());
|
|
Assert::IsTrue(val == data[1]);
|
|
|
|
f.seekg(_countof(data) - 1);
|
|
f.read(&val, 1);
|
|
Assert::IsTrue(f.gcount() == 1);
|
|
Assert::IsTrue(f.good());
|
|
Assert::IsTrue(val == data[_countof(data) - 1]);
|
|
|
|
f.read(&val, 1);
|
|
Assert::IsTrue(f.eof());
|
|
f.clear();
|
|
|
|
f.seekg(-2, ios_base::end);
|
|
f.read(&val, 1);
|
|
Assert::IsTrue(f.gcount() == 1);
|
|
Assert::IsTrue(f.good());
|
|
Assert::IsTrue(val == data[_countof(data) - 2]);
|
|
}
|
|
|
|
TEST_METHOD(diagstream)
|
|
{
|
|
constexpr size_t n = 3;
|
|
unique_ptr<std::fstream> f[n];
|
|
vector<std::fstream*> fv(n);
|
|
|
|
{
|
|
for (size_t i = 0; i < n; ++i) {
|
|
WCHAR path[MAX_PATH];
|
|
ExpandEnvironmentStringsW(stdex::sprintf(L"%%temp%%\\file%zu.dat", NULL, i).c_str(), path, _countof(path));
|
|
f[i].reset(new std::fstream(path, ios_base::out | ios_base::binary));
|
|
fv[i] = f[i].get();
|
|
}
|
|
stdex::diagstream d(fv.begin(), fv.end());
|
|
srand(0);
|
|
auto write_some_random = [](_Inout_ ostream& f, _In_ size_t amount)
|
|
{
|
|
for (size_t i = 0; i < amount; ++i) {
|
|
auto r = static_cast<uint32_t>(rand());
|
|
f.write(reinterpret_cast<const char*>(&r), sizeof(r) / sizeof(char));
|
|
}
|
|
};
|
|
write_some_random(d, 15);
|
|
d.seekp(3);
|
|
write_some_random(d, 2);
|
|
d.seekp(28);
|
|
write_some_random(d, 7);
|
|
}
|
|
|
|
{
|
|
for (size_t i = 0; i < n; ++i) {
|
|
WCHAR path[MAX_PATH];
|
|
ExpandEnvironmentStringsW(stdex::sprintf(L"%%temp%%\\file%zu.dat", NULL, i).c_str(), path, _countof(path));
|
|
f[i].reset(new std::fstream(path, ios_base::in | ios_base::binary));
|
|
fv[i] = f[i].get();
|
|
}
|
|
stdex::diagstream d(fv.begin(), fv.end());
|
|
do {
|
|
uint32_t r;
|
|
d.read(reinterpret_cast<char*>(&r), sizeof(r) / sizeof(char));
|
|
} while (!d.eof());
|
|
}
|
|
}
|
|
};
|
|
}
|