diff options
Diffstat (limited to 'vbench/src/tests/app_vbench/app_vbench_test.cpp')
-rw-r--r-- | vbench/src/tests/app_vbench/app_vbench_test.cpp | 23 |
1 files changed, 13 insertions, 10 deletions
diff --git a/vbench/src/tests/app_vbench/app_vbench_test.cpp b/vbench/src/tests/app_vbench/app_vbench_test.cpp index d1285255959..e65dd9badc0 100644 --- a/vbench/src/tests/app_vbench/app_vbench_test.cpp +++ b/vbench/src/tests/app_vbench/app_vbench_test.cpp @@ -6,16 +6,19 @@ using namespace vbench; using vespalib::SlaveProc; +using InputReader = vespalib::InputReader; +using OutputWriter = vespalib::OutputWriter; + bool endsWith(const Memory &mem, const string &str) { return (mem.size < str.size()) ? false : (strncmp(mem.data + mem.size - str.size(), str.data(), str.size()) == 0); } void readUntil(Input &input, SimpleBuffer &buffer, const string &end) { - ByteInput in(input); + InputReader in(input); while (!endsWith(buffer.get(), end)) { - int c = in.get(); - if (c < 0) { + char c = in.read(); + if (in.failed()) { return; } buffer.reserve(1).data[0] = c; @@ -38,13 +41,13 @@ TEST_MT_F("run vbench", 2, ServerSocket()) { } SimpleBuffer ignore; readUntil(*stream, ignore, "\r\n\r\n"); - BufferedOutput out(*stream, 256); - out.append("HTTP/1.1 200\r\n"); - out.append("content-length: 4\r\n"); - out.append("X-Yahoo-Vespa-NumHits 10\r\n"); - out.append("X-Yahoo-Vespa-TotalHitCount 100\r\n"); - out.append("\r\n"); - out.append("data"); + OutputWriter out(*stream, 256); + out.write("HTTP/1.1 200\r\n"); + out.write("content-length: 4\r\n"); + out.write("X-Yahoo-Vespa-NumHits 10\r\n"); + out.write("X-Yahoo-Vespa-TotalHitCount 100\r\n"); + out.write("\r\n"); + out.write("data"); } } else { std::string out; |