// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include <vespa/messagebus/emptyreply.h>
#include <vespa/messagebus/imessagehandler.h>
#include <vespa/messagebus/replygate.h>
#include <vespa/messagebus/routablequeue.h>
#include <vespa/messagebus/testlib/simplemessage.h>
#include <vespa/vespalib/testkit/testapp.h>

using namespace mbus;

namespace {

struct MyGate : public ReplyGate {
    static int ctorCnt;
    static int dtorCnt;

    MyGate(IMessageHandler &sender) : ReplyGate(sender) {
        ++ctorCnt;
    }

    ~MyGate() override {
        ++dtorCnt;
    }
};

int MyGate::ctorCnt = 0;
int MyGate::dtorCnt = 0;

struct MyReply : public EmptyReply {
    static int ctorCnt;
    static int dtorCnt;

    MyReply() : EmptyReply() {
        ++ctorCnt;
    }

    ~MyReply() override {
        ++dtorCnt;
    }
};

int MyReply::ctorCnt = 0;
int MyReply::dtorCnt = 0;

struct MySender : public IMessageHandler {
    // giving a sync reply here is against the API contract, but it is
    // ok for testing.
    void handleMessage(Message::UP msg) override {
        auto reply = std::make_unique<MyReply>();
        msg->swapState(*reply);
        IReplyHandler &handler = reply->getCallStack().pop(*reply);
        handler.handleReply(std::move(reply));
    }
};
}

TEST("replygate_test") {
    {
        RoutableQueue q;
        MySender      sender;
        auto gate = vespalib::make_ref_counted<MyGate>(sender);
        {
            auto msg = std::make_unique<SimpleMessage>("test");
            msg->pushHandler(q);
            gate->handleMessage(std::move(msg));
        }
        EXPECT_TRUE(q.size() == 1);
        EXPECT_TRUE(MyReply::ctorCnt == 1);
        EXPECT_TRUE(MyReply::dtorCnt == 0);
        gate->close();
        {
            auto msg = std::make_unique<SimpleMessage>("test");
            msg->pushHandler(q);
            gate->handleMessage(std::move(msg));
        }
        EXPECT_TRUE(q.size() == 1);
        EXPECT_TRUE(MyReply::ctorCnt == 2);
        EXPECT_TRUE(MyReply::dtorCnt == 1);
        EXPECT_TRUE(MyGate::ctorCnt == 1);
        EXPECT_TRUE(MyGate::dtorCnt == 0);
        gate.reset();
        EXPECT_TRUE(MyGate::ctorCnt == 1);
        EXPECT_TRUE(MyGate::dtorCnt == 1);
    }
    EXPECT_TRUE(MyReply::ctorCnt == 2);
    EXPECT_TRUE(MyReply::dtorCnt == 2);
}

TEST_MAIN() { TEST_RUN_ALL(); }