mirror of
https://github.com/XRPLF/clio.git
synced 2026-04-29 15:37:53 +00:00
103 lines
3.1 KiB
C++
103 lines
3.1 KiB
C++
#include "feed/FeedTestUtil.hpp"
|
|
#include "feed/impl/SingleFeedBase.hpp"
|
|
#include "util/MockPrometheus.hpp"
|
|
#include "util/MockWsBase.hpp"
|
|
#include "util/SyncExecutionCtxFixture.hpp"
|
|
#include "util/async/AnyExecutionContext.hpp"
|
|
#include "util/prometheus/Gauge.hpp"
|
|
#include "web/SubscriptionContextInterface.hpp"
|
|
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <memory>
|
|
|
|
namespace {
|
|
constexpr auto kFEED = R"JSON({"test": "test"})JSON";
|
|
} // namespace
|
|
|
|
using namespace feed::impl;
|
|
using namespace util::prometheus;
|
|
|
|
struct FeedBaseMockPrometheusTest : WithMockPrometheus, SyncExecutionCtxFixture {
|
|
protected:
|
|
web::SubscriptionContextPtr sessionPtr_ = std::make_shared<MockSession>();
|
|
std::shared_ptr<SingleFeedBase> testFeedPtr_ =
|
|
std::make_shared<SingleFeedBase>(ctx_, "testFeed");
|
|
MockSession* mockSessionPtr_ = dynamic_cast<MockSession*>(sessionPtr_.get());
|
|
};
|
|
|
|
TEST_F(FeedBaseMockPrometheusTest, subUnsub)
|
|
{
|
|
auto& counter = makeMock<GaugeInt>("subscriptions_current_number", "{stream=\"testFeed\"}");
|
|
EXPECT_CALL(counter, add(1));
|
|
EXPECT_CALL(counter, add(-1));
|
|
|
|
EXPECT_CALL(*mockSessionPtr_, onDisconnect);
|
|
testFeedPtr_->sub(sessionPtr_);
|
|
testFeedPtr_->unsub(sessionPtr_);
|
|
}
|
|
|
|
TEST_F(FeedBaseMockPrometheusTest, AutoUnsub)
|
|
{
|
|
auto& counter = makeMock<GaugeInt>("subscriptions_current_number", "{stream=\"testFeed\"}");
|
|
EXPECT_CALL(counter, add(1));
|
|
EXPECT_CALL(counter, add(-1));
|
|
|
|
web::SubscriptionContextInterface::OnDisconnectSlot slot;
|
|
EXPECT_CALL(*mockSessionPtr_, onDisconnect).WillOnce(testing::SaveArg<0>(&slot));
|
|
testFeedPtr_->sub(sessionPtr_);
|
|
slot(sessionPtr_.get());
|
|
sessionPtr_.reset();
|
|
}
|
|
|
|
class NamedSingleFeedTest : public SingleFeedBase {
|
|
public:
|
|
NamedSingleFeedTest(util::async::AnyExecutionContext& executionCtx)
|
|
: SingleFeedBase(executionCtx, "forTest")
|
|
{
|
|
}
|
|
};
|
|
|
|
using SingleFeedBaseTest = FeedBaseTest<NamedSingleFeedTest>;
|
|
|
|
TEST_F(SingleFeedBaseTest, Test)
|
|
{
|
|
EXPECT_CALL(*mockSessionPtr, onDisconnect);
|
|
EXPECT_CALL(*mockSessionPtr, send(sharedStringJsonEq(kFEED)));
|
|
testFeedPtr->sub(sessionPtr);
|
|
EXPECT_EQ(testFeedPtr->count(), 1);
|
|
testFeedPtr->pub(kFEED);
|
|
|
|
testFeedPtr->unsub(sessionPtr);
|
|
EXPECT_EQ(testFeedPtr->count(), 0);
|
|
testFeedPtr->pub(kFEED);
|
|
}
|
|
|
|
TEST_F(SingleFeedBaseTest, TestAutoDisconnect)
|
|
{
|
|
web::SubscriptionContextInterface::OnDisconnectSlot slot;
|
|
EXPECT_CALL(*mockSessionPtr, onDisconnect).WillOnce(testing::SaveArg<0>(&slot));
|
|
EXPECT_CALL(*mockSessionPtr, send(sharedStringJsonEq(kFEED)));
|
|
testFeedPtr->sub(sessionPtr);
|
|
EXPECT_EQ(testFeedPtr->count(), 1);
|
|
testFeedPtr->pub(kFEED);
|
|
|
|
slot(sessionPtr.get());
|
|
sessionPtr.reset();
|
|
EXPECT_EQ(testFeedPtr->count(), 0);
|
|
}
|
|
|
|
TEST_F(SingleFeedBaseTest, RepeatSub)
|
|
{
|
|
EXPECT_CALL(*mockSessionPtr, onDisconnect);
|
|
testFeedPtr->sub(sessionPtr);
|
|
EXPECT_EQ(testFeedPtr->count(), 1);
|
|
testFeedPtr->sub(sessionPtr);
|
|
EXPECT_EQ(testFeedPtr->count(), 1);
|
|
testFeedPtr->unsub(sessionPtr);
|
|
EXPECT_EQ(testFeedPtr->count(), 0);
|
|
testFeedPtr->unsub(sessionPtr);
|
|
EXPECT_EQ(testFeedPtr->count(), 0);
|
|
}
|