86 lines
2.4 KiB
C++
86 lines
2.4 KiB
C++
|
/* Copyright (c) 2017, Google Inc.
|
||
|
|
*
|
||
|
|
* Permission to use, copy, modify, and/or distribute this software for any
|
||
|
|
* purpose with or without fee is hereby granted, provided that the above
|
||
|
|
* copyright notice and this permission notice appear in all copies.
|
||
|
|
*
|
||
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
|
||
|
|
* SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
||
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
|
||
|
|
* OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
|
||
|
|
* CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */
|
||
|
|
|
||
|
|
#include "file_test.h"
|
||
|
|
|
||
|
|
#include <assert.h>
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
#include <memory>
|
||
|
|
#include <string>
|
||
|
|
#include <utility>
|
||
|
|
|
||
|
|
#include <gtest/gtest.h>
|
||
|
|
|
||
|
|
|
||
|
|
std::string GetTestData(const char *path);
|
||
|
|
|
||
|
|
class StringLineReader : public FileTest::LineReader {
|
||
|
|
public:
|
||
|
|
explicit StringLineReader(const std::string &data)
|
||
|
|
: data_(data), offset_(0) {}
|
||
|
|
|
||
|
|
FileTest::ReadResult ReadLine(char *out, size_t len) override {
|
||
|
|
assert(len > 0);
|
||
|
|
if (offset_ == data_.size()) {
|
||
|
|
return FileTest::kReadEOF;
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t idx = data_.find('\n', offset_);
|
||
|
|
if (idx == std::string::npos) {
|
||
|
|
idx = data_.size();
|
||
|
|
} else {
|
||
|
|
idx++; // Include the newline.
|
||
|
|
}
|
||
|
|
|
||
|
|
if (idx - offset_ > len - 1) {
|
||
|
|
ADD_FAILURE() << "Line too long.";
|
||
|
|
return FileTest::kReadError;
|
||
|
|
}
|
||
|
|
|
||
|
|
memcpy(out, data_.data() + offset_, idx - offset_);
|
||
|
|
out[idx - offset_] = '\0';
|
||
|
|
offset_ = idx;
|
||
|
|
return FileTest::kReadSuccess;
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
std::string data_;
|
||
|
|
size_t offset_;
|
||
|
|
|
||
|
|
StringLineReader(const StringLineReader &) = delete;
|
||
|
|
StringLineReader &operator=(const StringLineReader &) = delete;
|
||
|
|
};
|
||
|
|
|
||
|
|
void FileTestGTest(const char *path, std::function<void(FileTest *)> run_test) {
|
||
|
|
std::unique_ptr<StringLineReader> reader(
|
||
|
|
new StringLineReader(GetTestData(path)));
|
||
|
|
FileTest t(std::move(reader));
|
||
|
|
|
||
|
|
while (true) {
|
||
|
|
switch (t.ReadNext()) {
|
||
|
|
case FileTest::kReadError:
|
||
|
|
ADD_FAILURE() << "Error reading test.";
|
||
|
|
return;
|
||
|
|
case FileTest::kReadEOF:
|
||
|
|
return;
|
||
|
|
case FileTest::kReadSuccess:
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
SCOPED_TRACE(testing::Message() << path << ", line " << t.start_line());
|
||
|
|
run_test(&t);
|
||
|
|
}
|
||
|
|
}
|