diff options
author | Haavard <havardpe@yahoo-inc.com> | 2017-02-03 13:48:49 +0000 |
---|---|---|
committer | Haavard <havardpe@yahoo-inc.com> | 2017-02-03 18:16:45 +0000 |
commit | 324c400925b1d8e4dfb700dd7253e467407e8db7 (patch) | |
tree | 9148cce12e27bd44c3b71d0140fe50c741add393 /vbench/src/tests/app_vbench/app_vbench_test.cpp | |
parent | 439a2d738d072598ec44e3f87f9436a10d6af1e5 (diff) |
use new input/output model in vbench
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; |