| Index: tests/StreamTest.cpp
|
| diff --git a/tests/StreamTest.cpp b/tests/StreamTest.cpp
|
| index 926cfc25796b11a84716aae5ed4c77004fa42400..2a4499716da6e834987215d02d8d526b54ca8f6a 100644
|
| --- a/tests/StreamTest.cpp
|
| +++ b/tests/StreamTest.cpp
|
| @@ -11,6 +11,7 @@
|
| #include "SkOSFile.h"
|
| #include "SkRandom.h"
|
| #include "SkStream.h"
|
| +#include "SkStreamPriv.h"
|
| #include "Test.h"
|
|
|
| #ifndef SK_BUILD_FOR_WIN
|
| @@ -335,3 +336,49 @@ DEF_TEST(StreamPeek_BlockMemoryStream, rep) {
|
| }
|
| stream_peek_test(rep, asset, expected);
|
| }
|
| +
|
| +namespace {
|
| +class DumbStream : public SkStream {
|
| +public:
|
| + DumbStream(const uint8_t* data, size_t n)
|
| + : fData(data), fCount(n), fIdx(0) {}
|
| + size_t read(void* buffer, size_t size) override {
|
| + size_t c = SkTMin(fCount - fIdx, size);
|
| + if (c) {
|
| + memcpy(buffer, &fData[fIdx], size);
|
| + fIdx += c;
|
| + }
|
| + return c;
|
| + }
|
| + bool isAtEnd() const override {
|
| + return fCount > fIdx;
|
| + }
|
| + private:
|
| + const uint8_t* fData;
|
| + size_t fCount, fIdx;
|
| +};
|
| +} // namespace
|
| +
|
| +DEF_TEST(StreamCopy, reporter) {
|
| + SkRandom random(123456);
|
| + static const size_t N = 10000;
|
| + uint8_t src[N];
|
| + for (size_t j = 0; j < N; ++j) {
|
| + src[j] = random.nextU() & 0xff;
|
| + }
|
| + DumbStream dumbStream(src, N);
|
| + SkDynamicMemoryWStream tgt;
|
| + if (!SkStreamCopy(&tgt, &dumbStream)) {
|
| + ERRORF(reporter, "SkStreamCopy failed");
|
| + return;
|
| + }
|
| + SkAutoTUnref<SkData> data(tgt.copyToData());
|
| + tgt.reset();
|
| + if (data->size() != N) {
|
| + ERRORF(reporter, "SkStreamCopy incorrect size");
|
| + return;
|
| + }
|
| + if (0 != memcmp(data->data(), src, N)) {
|
| + ERRORF(reporter, "SkStreamCopy bad copy");
|
| + }
|
| +}
|
|
|