Final part on compatibility layer: LOGBROKER-7215

ref:777c67aadbf705d19034a09a792b2df61ba53697

1 files changed, 47 insertions, 0 deletions

diff --git a/library/cpp/tvmauth/client/ut/src_checker_ut.cpp b/library/cpp/tvmauth/client/ut/src_checker_ut.cpp
new file mode 100644
index 0000000000..bd1646d6b9
--- /dev/null
+++ b/library/cpp/tvmauth/client/ut/src_checker_ut.cpp
@@ -0,0 +1,47 @@
+#include "common.h"
+
+#include <library/cpp/tvmauth/client/mocked_updater.h>
+#include <library/cpp/tvmauth/client/misc/src_checker.h>
+#include <library/cpp/tvmauth/client/misc/api/threaded_updater.h>
+
+#include <library/cpp/tvmauth/type.h>
+#include <library/cpp/tvmauth/unittest.h>
+
+#include <library/cpp/testing/unittest/registar.h>
+
+using namespace NTvmAuth;
+
+Y_UNIT_TEST_SUITE(SrcChecker) {
+    Y_UNIT_TEST(Ctor) {
+        UNIT_ASSERT_EXCEPTION_CONTAINS(
+            TSrcChecker(new TMockedUpdater),
+            TBrokenTvmClientSettings,
+            "Need to use TClientSettings::EnableRolesFetching");
+    }
+
+    Y_UNIT_TEST(Check) {
+        NRoles::TRolesPtr roles = std::make_shared<NRoles::TRoles>(
+            NRoles::TRoles::TMeta{},
+            NRoles::TRoles::TTvmConsumers{
+                {12345, std::make_shared<NRoles::TConsumerRoles>(
+                            THashMap<TString, NRoles::TEntitiesPtr>())},
+            },
+            NRoles::TRoles::TUserConsumers{},
+            std::make_shared<TString>());
+        const TSrcChecker checker(new TMockedUpdater({.Roles = roles}));
+
+        UNIT_ASSERT_EXCEPTION_CONTAINS(
+            checker.Check(NUnittest::CreateServiceTicket(ETicketStatus::Expired, 12345)),
+            TIllegalUsage,
+            "Service ticket must be valid");
+
+        TCheckedServiceTicket ticket;
+        UNIT_ASSERT_NO_EXCEPTION(
+            ticket = checker.Check(NUnittest::CreateServiceTicket(ETicketStatus::Ok, 12345)));
+        UNIT_ASSERT_VALUES_EQUAL(ETicketStatus::Ok, ticket.GetStatus());
+
+        UNIT_ASSERT_NO_EXCEPTION(
+            ticket = checker.Check(NUnittest::CreateServiceTicket(ETicketStatus::Ok, 9999)));
+        UNIT_ASSERT_VALUES_EQUAL(ETicketStatus::NoRoles, ticket.GetStatus());
+    }
+}