// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/vespalib/testkit/testapp.h>
#include <vbench/test/all.h>
#include <vespa/vespalib/util/slaveproc.h>

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) {
    InputReader in(input);
    while (!endsWith(buffer.get(), end)) {
        char c = in.read();
        if (in.failed()) {
            return;
        }
        buffer.reserve(1).data[0] = c;
        buffer.commit(1);
    }
}

TEST("dumpurl usage") {
    std::string out;
    EXPECT_FALSE(SlaveProc::run("../../apps/dumpurl/vbench_dumpurl_app", out));
    fprintf(stderr, "%s\n", out.c_str());
}

TEST_MT_F("run dumpurl", 2, ServerSocket()) {
    if (thread_id == 0) {
        Stream::UP stream = f1.accept();
        SimpleBuffer ignore;
        readUntil(*stream, ignore, "\r\n\r\n");
        OutputWriter out(*stream, 256);
        out.write("HTTP/1.1 200\r\n");
        out.write("content-length: 4\r\n");
        out.write("\r\n");
        out.write("data");
    } else {
        std::string out;
        EXPECT_TRUE(SlaveProc::run(strfmt("../../apps/dumpurl/vbench_dumpurl_app localhost %d /foo",
                                          f1.port()).c_str(), out));
        fprintf(stderr, "%s\n", out.c_str());
    }
}

TEST_MAIN_WITH_PROCESS_PROXY() { TEST_RUN_ALL(); }