lib/api/unittest/CMockSearcher.cc

/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0 and the following additional limitation. Functionality enabled by the
 * files subject to the Elastic License 2.0 may only be used in production when
 * invoked by an Elasticsearch process with a license key installed that permits
 * use of machine learning features. You may not use this file except in
 * compliance with the Elastic License 2.0 and the foregoing additional
 * limitation.
 */
#include "CMockSearcher.h"

#include <core/CLogger.h>

#include "CMockDataAdder.h"

CMockSearcher::CMockSearcher(const CMockDataAdder& mockDataAdder)
    : m_MockDataAdder{mockDataAdder} {
}

CMockSearcher::TIStreamP CMockSearcher::search(std::size_t currentDocNum, std::size_t /*limit*/) {
    if (currentDocNum == 0) {
        LOG_ERROR(<< "Current doc number cannot be 0 - data store requires 1-based numbers");
        return TIStreamP{};
    }

    const CMockDataAdder::TStrVec& events = m_MockDataAdder.events();
    if (currentDocNum > events.size()) {
        return TIStreamP{new std::istringstream("[ ]")};
    }
    return TIStreamP{new std::istringstream(events[currentDocNum - 1])};
}