kami/test/unit-kami-reporter.cc

/*-
 * Copyright (c) 2022 The Johns Hopkins University Applied Physics
 * Laboratory LLC
 *
 * Permission is hereby granted, free of charge, to any person
 * obtaining a copy of this software and associated documentation files
 * (the "Software"), to deal in the Software without restriction,
 * including without limitation the rights to use, copy, modify, merge,
 * publish, distribute, sublicense, and/or sell copies of the Software,
 * and to permit persons to whom the Software is furnished to do so,
 * subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT.  IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <algorithm>
#include <memory>
#include <utility>
#include <vector>

#include <kami/agent.h>
#include <kami/population.h>
#include <kami/reporter.h>
#include <kami/staged.h>

#include <gmock/gmock.h>
#include <gtest/gtest.h>

using namespace kami;
using namespace std;

class TestAgent
        : public ReporterAgent {
public:
    AgentID step(shared_ptr<ReporterModel> model) override {
        return get_agent_id();
    }

    std::unique_ptr<nlohmann::json> collect() override {
        auto json_ret_val = std::make_unique<nlohmann::json>();

        (*json_ret_val)["fname"] = "Jesse";
        (*json_ret_val)["lname"] = "Pinkman";

        return json_ret_val;
    }
};

class TestModel
        : public ReporterModel {
public:
    shared_ptr<vector<AgentID>> retval;

    std::unique_ptr<nlohmann::json> collect() override {
        auto json_ret_val = std::make_unique<nlohmann::json>();

        (*json_ret_val)["fname"] = "Walter";
        (*json_ret_val)["lname"] = "White";

        return json_ret_val;
    }
};

class ReporterModelTest
        : public ::testing::Test {
protected:
    shared_ptr<TestModel> mod = nullptr;

    void SetUp() override {
        mod = make_shared<TestModel>();
        auto popul_foo = make_shared<Population>();
        auto sched_foo = make_shared<SequentialScheduler>();

        // Domain is not required for this test
        static_cast<void>(mod->set_population(popul_foo));
        static_cast<void>(mod->set_scheduler(sched_foo));

        for (auto i = 0; i < 3; i++) {
            auto agent_foo = make_shared<TestAgent>();
            static_cast<void>(popul_foo->add_agent(agent_foo));
        }
    }
};

TEST(ReporterModel, DefaultConstructor) {
    // There is really no way this can go wrong, but
    // we add this check anyway in case of future
    // changes.
    EXPECT_NO_THROW(
            const TestModel reporter_foo;
    );
}

TEST_F(ReporterModelTest, collect) {
    auto aval = mod->get_population()->get_agent_list();
    mod->step();

    auto rval = mod->collect();
    EXPECT_TRUE(rval);
    EXPECT_EQ(rval->dump(), "{\"fname\":\"Walter\",\"lname\":\"White\"}");
}

TEST_F(ReporterModelTest, report) {
    for (auto i = 0; i < 2; i++) {
        auto aval = mod->get_population()->get_agent_list();

        mod->step();
        auto rval = mod->collect();
    }

    auto rval = mod->report();
    EXPECT_EQ(rval->dump(),
            "[{\"agent_data\":[{\"agent_id\":\"13\",\"data\":{\"fname\":\"Jesse\",\"lname\":\"Pinkman\"}},{\"agent_id\":\"14\",\"data\":{\"fname\":\"Jesse\",\"lname\":\"Pinkman\"}},{\"agent_id\":\"15\",\"data\":{\"fname\":\"Jesse\",\"lname\":\"Pinkman\"}}],\"model_data\":{\"fname\":\"Walter\",\"lname\":\"White\"},\"step_id\":1},{\"agent_data\":[{\"agent_id\":\"13\",\"data\":{\"fname\":\"Jesse\",\"lname\":\"Pinkman\"}},{\"agent_id\":\"14\",\"data\":{\"fname\":\"Jesse\",\"lname\":\"Pinkman\"}},{\"agent_id\":\"15\",\"data\":{\"fname\":\"Jesse\",\"lname\":\"Pinkman\"}}],\"model_data\":{\"fname\":\"Walter\",\"lname\":\"White\"},\"step_id\":2}]");
}

int main(
        int argc,
        char** argv
) {
    ::testing::InitGoogleTest(&argc, argv);
    return RUN_ALL_TESTS();
}