ce49d99c2f
Since we switched from autotools to Meson in commit
94592c1406, we don't need to include
`config.h` early to properly enable large file support. Meson passes
the required macros on the compiler command line instead of defining
them in `config.h`.
This means we can include `config.h` at any time, whenever we want to
check its macros, and there are no ordering constraints.
126 lines
3.0 KiB
C++
126 lines
3.0 KiB
C++
/*
|
|
* Unit tests for class RewindInputStream.
|
|
*/
|
|
|
|
#include "input/RewindInputStream.hxx"
|
|
#include "input/InputStream.hxx"
|
|
#include "thread/Mutex.hxx"
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <string.h>
|
|
|
|
class StringInputStream final : public InputStream {
|
|
const char *data;
|
|
size_t remaining;
|
|
|
|
public:
|
|
StringInputStream(const char *_uri,
|
|
Mutex &_mutex,
|
|
const char *_data)
|
|
:InputStream(_uri, _mutex),
|
|
data(_data), remaining(strlen(data)) {
|
|
SetReady();
|
|
}
|
|
|
|
/* virtual methods from InputStream */
|
|
bool IsEOF() noexcept override {
|
|
return remaining == 0;
|
|
}
|
|
|
|
size_t Read(void *ptr, size_t read_size) override {
|
|
size_t nbytes = std::min(remaining, read_size);
|
|
memcpy(ptr, data, nbytes);
|
|
data += nbytes;
|
|
remaining -= nbytes;
|
|
offset += nbytes;
|
|
return nbytes;
|
|
}
|
|
};
|
|
|
|
TEST(RewindInputStream, Basic)
|
|
{
|
|
Mutex mutex;
|
|
|
|
StringInputStream *sis =
|
|
new StringInputStream("foo://", mutex,
|
|
"foo bar");
|
|
EXPECT_TRUE(sis->IsReady());
|
|
|
|
auto ris = input_rewind_open(InputStreamPtr(sis));
|
|
EXPECT_TRUE(ris.get() != sis);
|
|
EXPECT_TRUE(ris != nullptr);
|
|
|
|
const std::lock_guard<Mutex> protect(mutex);
|
|
|
|
ris->Update();
|
|
EXPECT_TRUE(ris->IsReady());
|
|
EXPECT_FALSE(ris->KnownSize());
|
|
EXPECT_EQ(offset_type(0), ris->GetOffset());
|
|
|
|
char buffer[16];
|
|
size_t nbytes = ris->Read(buffer, 2);
|
|
EXPECT_EQ(size_t(2), nbytes);
|
|
EXPECT_EQ('f', buffer[0]);
|
|
EXPECT_EQ('o', buffer[1]);
|
|
EXPECT_EQ(offset_type(2), ris->GetOffset());
|
|
EXPECT_FALSE(ris->IsEOF());
|
|
|
|
nbytes = ris->Read(buffer, 2);
|
|
EXPECT_EQ(size_t(2), nbytes);
|
|
EXPECT_EQ('o', buffer[0]);
|
|
EXPECT_EQ(' ', buffer[1]);
|
|
EXPECT_EQ(offset_type(4), ris->GetOffset());
|
|
EXPECT_FALSE(ris->IsEOF());
|
|
|
|
ris->Seek(1);
|
|
EXPECT_EQ(offset_type(1), ris->GetOffset());
|
|
EXPECT_FALSE(ris->IsEOF());
|
|
|
|
nbytes = ris->Read(buffer, 2);
|
|
EXPECT_EQ(size_t(2), nbytes);
|
|
EXPECT_EQ('o', buffer[0]);
|
|
EXPECT_EQ('o', buffer[1]);
|
|
EXPECT_EQ(offset_type(3), ris->GetOffset());
|
|
EXPECT_FALSE(ris->IsEOF());
|
|
|
|
ris->Seek(0);
|
|
EXPECT_EQ(offset_type(0), ris->GetOffset());
|
|
EXPECT_FALSE(ris->IsEOF());
|
|
|
|
nbytes = ris->Read(buffer, 2);
|
|
EXPECT_EQ(size_t(2), nbytes);
|
|
EXPECT_EQ('f', buffer[0]);
|
|
EXPECT_EQ('o', buffer[1]);
|
|
EXPECT_EQ(offset_type(2), ris->GetOffset());
|
|
EXPECT_FALSE(ris->IsEOF());
|
|
|
|
nbytes = ris->Read(buffer, sizeof(buffer));
|
|
EXPECT_EQ(size_t(2), nbytes);
|
|
EXPECT_EQ('o', buffer[0]);
|
|
EXPECT_EQ(' ', buffer[1]);
|
|
EXPECT_EQ(offset_type(4), ris->GetOffset());
|
|
EXPECT_FALSE(ris->IsEOF());
|
|
|
|
nbytes = ris->Read(buffer, sizeof(buffer));
|
|
EXPECT_EQ(size_t(3), nbytes);
|
|
EXPECT_EQ('b', buffer[0]);
|
|
EXPECT_EQ('a', buffer[1]);
|
|
EXPECT_EQ('r', buffer[2]);
|
|
EXPECT_EQ(offset_type(7), ris->GetOffset());
|
|
EXPECT_TRUE(ris->IsEOF());
|
|
|
|
ris->Seek(3);
|
|
EXPECT_EQ(offset_type(3), ris->GetOffset());
|
|
EXPECT_FALSE(ris->IsEOF());
|
|
|
|
nbytes = ris->Read(buffer, sizeof(buffer));
|
|
EXPECT_EQ(size_t(4), nbytes);
|
|
EXPECT_EQ(' ', buffer[0]);
|
|
EXPECT_EQ('b', buffer[1]);
|
|
EXPECT_EQ('a', buffer[2]);
|
|
EXPECT_EQ('r', buffer[3]);
|
|
EXPECT_EQ(offset_type(7), ris->GetOffset());
|
|
EXPECT_TRUE(ris->IsEOF());
|
|
}
|