mirror of
https://github.com/XRPLF/clio.git
synced 2026-04-29 15:37:53 +00:00
64 lines
1.6 KiB
C++
64 lines
1.6 KiB
C++
#include "feed/FeedTestUtil.hpp"
|
|
#include "feed/impl/ForwardFeed.hpp"
|
|
#include "util/async/AnyExecutionContext.hpp"
|
|
#include "web/SubscriptionContextInterface.hpp"
|
|
|
|
#include <boost/json/parse.hpp>
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <memory>
|
|
|
|
using namespace feed::impl;
|
|
namespace json = boost::json;
|
|
using namespace util::prometheus;
|
|
|
|
namespace {
|
|
|
|
constexpr auto kFEED = R"JSON({"test": "test"})JSON";
|
|
|
|
} // namespace
|
|
|
|
class NamedForwardFeedTest : public ForwardFeed {
|
|
public:
|
|
NamedForwardFeedTest(util::async::AnyExecutionContext& executionCtx)
|
|
: ForwardFeed(executionCtx, "test")
|
|
{
|
|
}
|
|
};
|
|
|
|
using FeedForwardTest = FeedBaseTest<NamedForwardFeedTest>;
|
|
|
|
TEST_F(FeedForwardTest, Pub)
|
|
{
|
|
EXPECT_CALL(*mockSessionPtr, onDisconnect);
|
|
testFeedPtr->sub(sessionPtr);
|
|
EXPECT_EQ(testFeedPtr->count(), 1);
|
|
|
|
auto const json = json::parse(kFEED).as_object();
|
|
EXPECT_CALL(*mockSessionPtr, send(sharedStringJsonEq(kFEED))).Times(1);
|
|
testFeedPtr->pub(json);
|
|
|
|
testFeedPtr->unsub(sessionPtr);
|
|
EXPECT_EQ(testFeedPtr->count(), 0);
|
|
testFeedPtr->pub(json);
|
|
}
|
|
|
|
TEST_F(FeedForwardTest, AutoDisconnect)
|
|
{
|
|
web::SubscriptionContextInterface::OnDisconnectSlot slot;
|
|
EXPECT_CALL(*mockSessionPtr, onDisconnect).WillOnce(testing::SaveArg<0>(&slot));
|
|
testFeedPtr->sub(sessionPtr);
|
|
EXPECT_EQ(testFeedPtr->count(), 1);
|
|
|
|
auto const json = json::parse(kFEED).as_object();
|
|
EXPECT_CALL(*mockSessionPtr, send(sharedStringJsonEq(kFEED)));
|
|
testFeedPtr->pub(json);
|
|
|
|
slot(sessionPtr.get());
|
|
sessionPtr.reset();
|
|
EXPECT_EQ(testFeedPtr->count(), 0);
|
|
|
|
testFeedPtr->pub(json);
|
|
}
|