cloud/blockstore/libs/notify/impl/notify_ut.cpp - main - githubmirror/nbs

cloud/blockstore/libs/notify/impl/notify_ut.cpp
191 строка5 KB
Georg-0001
[Blockstore] Notify service refactoring to further support new notification format (#4007)
08 авг 2025, 12:30
08 авг 2025, 12:30311fc3f
#include "notify.h"

#include "json_generator.h"

#include <cloud/blockstore/libs/notify/iface/config.h>

#include <cloud/storage/core/libs/diagnostics/logging.h>
#include <cloud/storage/core/libs/iam/iface/client.h>

#include <library/cpp/testing/unittest/env.h>
#include <library/cpp/testing/unittest/registar.h>
#include <library/cpp/testing/unittest/tests_data.h>

namespace NCloud::NBlockStore::NNotify {

using namespace NThreading;

namespace {

////////////////////////////////////////////////////////////////////////////////

const constexpr int V1 = 1;
const constexpr int V2 = 2;

static constexpr TDuration WaitTimeout = TDuration::Seconds(30);

////////////////////////////////////////////////////////////////////////////////

// Same class as TIamTokenClientForStub, but returns nonempty token.
class TIamTokenClientForTests final: public NIamClient::IIamTokenClient
{
private:
    NIamClient::TTokenInfo Token =
        NIamClient::TTokenInfo{"XXXXXXXXXXXXXXXXXXXXXXXXXX", TInstant::Zero()};

public:
    TResultOrError<NIamClient::TTokenInfo> GetToken() override
    {
        return Token;
    }

    TFuture<TResultOrError<NIamClient::TTokenInfo>> GetTokenAsync() override
    {
        return MakeFuture(TResultOrError(Token));
    }

    void Start() override
    {}

    void Stop() override
    {}
};

using TIamTokenClientForTestsPtr = std::shared_ptr<TIamTokenClientForTests>;

////////////////////////////////////////////////////////////////////////////////

auto MakeConfig()
{
    const TString port = getenv("NOTIFY_SERVICE_MOCK_PORT");

    NProto::TNotifyConfig proto;
    proto.SetEndpoint("https://localhost:" + port + "/notify/v1/send");
    proto.SetCaCertFilename(
        JoinFsPaths(getenv("TEST_CERT_FILES_DIR"), "server.crt"));

    return std::make_shared<TNotifyConfig>(std::move(proto));
}

auto MakeConfigV2()
{
    const TString port = getenv("NOTIFY_SERVICE_MOCK_PORT");

    NProto::TNotifyConfig proto;
    proto.SetEndpoint("https://localhost:" + port + "/notify/v2/send");
    proto.SetCaCertFilename(
        JoinFsPaths(getenv("TEST_CERT_FILES_DIR"), "server.crt"));
    proto.SetVersion(2);

    return std::make_shared<TNotifyConfig>(std::move(proto));
}

auto CreateNotifyService(int version)
{
    return CreateService(
        version == V2 ? MakeConfigV2() : MakeConfig(),
        std::make_shared<TIamTokenClientForTests>(),
        std::make_unique<TJsonGenerator>());
}

void ShouldNotifyDiskErrorImpl(int version)
{
    auto service = CreateNotifyService(version);

    service->Start();

    auto r = service->Notify({
            .CloudId = "yc-nbs",
            .FolderId = "yc-nbs.folder",
            .Timestamp = TInstant::ParseIso8601("2024-04-01T00:00:01Z"),
            .Event =
                TDiskError{
                    .DiskId = "ShouldNotifyDiskErrorV" + ToString(version)},
        }).GetValue(WaitTimeout);

    UNIT_ASSERT_C(!HasError(r), r);

    service->Stop();
}

void ShouldNotifyDiskBackOnlineImpl(int version)
{
    auto service = CreateNotifyService(version);
    service->Start();

    auto r = service->Notify({
            .CloudId = "yc-nbs",
            .FolderId = "yc-nbs.folder",
            .Timestamp = TInstant::ParseIso8601("2023-01-01T00:00:01Z"),
            .Event = TDiskBackOnline{.DiskId = "nrd0"},
        }).GetValue(WaitTimeout);

    UNIT_ASSERT_C(!HasError(r), r);

    service->Stop();
}

void ShouldNotifyAboutLotsOfDiskErrorsImpl(int version)
{
    auto service = CreateNotifyService(version);
    service->Start();

    TVector<TFuture<NProto::TError>> futures;
    for (ui32 i = 0; i < 20; ++i) {
        futures.push_back(service->Notify({
            .CloudId = "yc-nbs",
            .FolderId = "yc-nbs.folder",
            .Timestamp = TInstant::ParseIso8601("2023-01-01T00:00:01Z"),
            .Event = TDiskError{.DiskId = "nrd0"},
        }));
    }

    for (auto& f: futures) {
        auto r = f.GetValue(WaitTimeout);
        UNIT_ASSERT_C(!HasError(r), r);
    }

    service->Stop();
}

}   // namespace

////////////////////////////////////////////////////////////////////////////////

Y_UNIT_TEST_SUITE(TNotifyTest)
{
    Y_UNIT_TEST(ShouldNotifyDiskError)
    {
        ShouldNotifyDiskErrorImpl(V1);
    }

    Y_UNIT_TEST(ShouldNotifyDiskBackOnline)
    {
        ShouldNotifyDiskBackOnlineImpl(V1);
    }

    Y_UNIT_TEST(ShouldNotifyAboutLotsOfDiskErrors)
    {
        ShouldNotifyAboutLotsOfDiskErrorsImpl(V1);
    }
}

Y_UNIT_TEST_SUITE(TNotifyTestV2)
{
    Y_UNIT_TEST(ShouldNotifyDiskErrorV2)
    {
        ShouldNotifyDiskErrorImpl(V2);
    }

    Y_UNIT_TEST(ShouldNotifyDiskBackOnlineV2)
    {
        ShouldNotifyDiskBackOnlineImpl(V2);
    }

    Y_UNIT_TEST(ShouldNotifyAboutLotsOfDiskErrorsV2)
    {
        ShouldNotifyAboutLotsOfDiskErrorsImpl(V2);
    }
}

}   // namespace NCloud::NBlockStore::NNotify