intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 176 insertions, 0 deletions

diff --git a/library/cpp/tvmauth/client/ut/checker_ut.cpp b/library/cpp/tvmauth/client/ut/checker_ut.cpp
new file mode 100644
index 0000000000..54a25974c1
--- /dev/null
+++ b/library/cpp/tvmauth/client/ut/checker_ut.cpp
@@ -0,0 +1,176 @@
+#include "common.h"
+
+#include <library/cpp/tvmauth/client/mocked_updater.h>
+#include <library/cpp/tvmauth/client/misc/checker.h>
+#include <library/cpp/tvmauth/client/misc/getter.h>
+#include <library/cpp/tvmauth/client/misc/api/threaded_updater.h>
+
+#include <library/cpp/tvmauth/type.h>
+
+#include <library/cpp/testing/unittest/registar.h>
+
+using namespace NTvmAuth;
+
+Y_UNIT_TEST_SUITE(ClientChecker) {
+    static const TTvmId OK_CLIENT = 100500;
+    static const TString PROD_TICKET = "3:user:CAsQ__________9_Gg4KAgh7EHsg0oXYzAQoAA:N8PvrDNLh-5JywinxJntLeQGDEHBUxfzjuvB8-_BEUv1x9CALU7do8irDlDYVeVVDr4AIpR087YPZVzWPAqmnBuRJS0tJXekmDDvrivLnbRrzY4IUXZ_fImB0fJhTyVetKv6RD11bGqnAJeDpIukBwPTbJc_EMvKDt8V490CJFw";
+    static const TString TEST_TICKET = "3:user:CA0Q__________9_Gg4KAgh7EHsg0oXYzAQoAQ:FSADps3wNGm92Vyb1E9IVq5M6ZygdGdt1vafWWEhfDDeCLoVA-sJesxMl2pGW4OxJ8J1r_MfpG3ZoBk8rLVMHUFrPa6HheTbeXFAWl8quEniauXvKQe4VyrpA1SPgtRoFqi5upSDIJzEAe1YRJjq1EClQ_slMt8R0kA_JjKUX54";
+    static const TString PROD_YATEAM_TICKET = "3:user:CAwQ__________9_Gg4KAgh7EHsg0oXYzAQoAg:M9dEFEWHLHXiL7brCsyfYlm254PE6VeshUjI62u2qMDRzt6-0jAoJTIdDiogerItht1YFYSn8fSqmMf23_rueGj-wkmvyNzbcBSk3jtK2U5sai_W0bK6OwukR9tzWzi1Gcgg9DrNEeIKFvs1EBqYCF4mPHWo5bgk0CR580Cgit4";
+    static const TString TEST_YATEAM_TICKET = "3:user:CA4Q__________9_Gg4KAgh7EHsg0oXYzAQoAw:IlaV3htk3jYrviIOz3k3Dfwz7p-bYYpbrgdn53GiUrMGdrT9eobHeuzNvPLrWB0yuYZAD46C3MGxok4GGmHhT73mki4XOCX8yWT4jW_hzcHBik1442tjWwh8IWqV_7q5j5496suVuLWjnZORWbb7I-2iwdIlU1BUiDfhoAolCq8";
+    static const TString STRESS_TICKET = "3:user:CA8Q__________9_Gg4KAgh7EHsg0oXYzAQoBA:GBuG_TLo6SL2OYFxp7Zly04HPNzmAF7Fu2E8E9SnwQDoxq9rf7VThSPtTmnBSAl5UVRRPkMsRtzzHZ87qtj6l-PvF0K7PrDu7-yS_xiFTgAl9sEfXAIHJVzZLoksGRgpoBtpBUg9vVaJsPns0kWFKJgq8M-Mk9agrSk7sb2VUeQ";
+    static const TString SRV_TICKET = "3:serv:CBAQ__________9_IgYIexCUkQY:GioCM49Ob6_f80y6FY0XBVN4hLXuMlFeyMvIMiDuQnZkbkLpRpQOuQo5YjWoBjM0Vf-XqOm8B7xtrvxSYHDD7Q4OatN2l-Iwg7i71lE3scUeD36x47st3nd0OThvtjrFx_D8mw_c0GT5KcniZlqq1SjhLyAk1b_zJsx8viRAhCU";
+
+    Y_UNIT_TEST(User) {
+        NTvmApi::TClientSettings s;
+        s.SetSelfTvmId(OK_CLIENT);
+        s.EnableServiceTicketChecking();
+        s.SetDiskCacheDir(GetCachePath());
+
+        auto l = MakeIntrusive<TLogger>();
+        {
+            auto u = NTvmApi::TThreadedUpdater::Create(s, l);
+            UNIT_ASSERT_VALUES_EQUAL(TClientStatus::Ok, u->GetStatus().GetCode());
+            UNIT_ASSERT_EXCEPTION(TUserTicketChecker(u), TBrokenTvmClientSettings);
+        }
+        UNIT_ASSERT_C(l->Stream.Str().find("was successfully fetched") == TString::npos, l->Stream.Str());
+
+        s.EnableUserTicketChecking(EBlackboxEnv::Prod);
+        {
+            auto u = NTvmApi::TThreadedUpdater::Create(s, l);
+            UNIT_ASSERT_VALUES_EQUAL(TClientStatus::Ok, u->GetStatus().GetCode());
+            TUserTicketChecker c(u);
+            UNIT_ASSERT(c.Check(PROD_TICKET, {}));
+            UNIT_ASSERT(!c.Check(TEST_TICKET, {}));
+            UNIT_ASSERT(!c.Check(PROD_YATEAM_TICKET, {}));
+            UNIT_ASSERT(!c.Check(TEST_YATEAM_TICKET, {}));
+            UNIT_ASSERT(!c.Check(STRESS_TICKET, {}));
+
+            UNIT_ASSERT(!c.Check(PROD_TICKET, EBlackboxEnv::ProdYateam));
+            UNIT_ASSERT(!c.Check(TEST_TICKET, EBlackboxEnv::ProdYateam));
+            UNIT_ASSERT(c.Check(PROD_YATEAM_TICKET, EBlackboxEnv::ProdYateam));
+            UNIT_ASSERT(!c.Check(TEST_YATEAM_TICKET, EBlackboxEnv::ProdYateam));
+            UNIT_ASSERT(!c.Check(STRESS_TICKET, EBlackboxEnv::ProdYateam));
+
+            UNIT_ASSERT_EXCEPTION(c.Check(PROD_TICKET, EBlackboxEnv::Stress), TBrokenTvmClientSettings);
+        }
+
+        s.EnableUserTicketChecking(EBlackboxEnv::Test);
+        {
+            auto u = NTvmApi::TThreadedUpdater::Create(s, l);
+            UNIT_ASSERT_VALUES_EQUAL(TClientStatus::Ok, u->GetStatus().GetCode());
+            TUserTicketChecker c(u);
+            UNIT_ASSERT(!c.Check(PROD_TICKET, {}));
+            UNIT_ASSERT(c.Check(TEST_TICKET, {}));
+            UNIT_ASSERT(!c.Check(PROD_YATEAM_TICKET, {}));
+            UNIT_ASSERT(!c.Check(TEST_YATEAM_TICKET, {}));
+            UNIT_ASSERT(!c.Check(STRESS_TICKET, {}));
+        }
+
+        s.EnableUserTicketChecking(EBlackboxEnv::ProdYateam);
+        {
+            auto u = NTvmApi::TThreadedUpdater::Create(s, l);
+            UNIT_ASSERT_VALUES_EQUAL(TClientStatus::Ok, u->GetStatus().GetCode());
+            TUserTicketChecker c(u);
+            UNIT_ASSERT(!c.Check(PROD_TICKET, {}));
+            UNIT_ASSERT(!c.Check(TEST_TICKET, {}));
+            UNIT_ASSERT(c.Check(PROD_YATEAM_TICKET, {}));
+            UNIT_ASSERT(!c.Check(TEST_YATEAM_TICKET, {}));
+            UNIT_ASSERT(!c.Check(STRESS_TICKET, {}));
+        }
+
+        s.EnableUserTicketChecking(EBlackboxEnv::TestYateam);
+        {
+            auto u = NTvmApi::TThreadedUpdater::Create(s, l);
+            UNIT_ASSERT_VALUES_EQUAL(TClientStatus::Ok, u->GetStatus().GetCode());
+            TUserTicketChecker c(u);
+            UNIT_ASSERT(!c.Check(PROD_TICKET, {}));
+            UNIT_ASSERT(!c.Check(TEST_TICKET, {}));
+            UNIT_ASSERT(!c.Check(PROD_YATEAM_TICKET, {}));
+            UNIT_ASSERT(c.Check(TEST_YATEAM_TICKET, {}));
+            UNIT_ASSERT(!c.Check(STRESS_TICKET, {}));
+        }
+
+        s.EnableUserTicketChecking(EBlackboxEnv::Stress);
+        {
+            auto u = NTvmApi::TThreadedUpdater::Create(s, l);
+            UNIT_ASSERT_VALUES_EQUAL(TClientStatus::Ok, u->GetStatus().GetCode());
+            TUserTicketChecker c(u);
+            UNIT_ASSERT(c.Check(PROD_TICKET, {}));
+            UNIT_ASSERT(!c.Check(TEST_TICKET, {}));
+            UNIT_ASSERT(!c.Check(PROD_YATEAM_TICKET, {}));
+            UNIT_ASSERT(!c.Check(TEST_YATEAM_TICKET, {}));
+            UNIT_ASSERT(c.Check(STRESS_TICKET, {}));
+        }
+    }
+
+    Y_UNIT_TEST(Service) {
+        NTvmApi::TClientSettings s;
+        s.EnableUserTicketChecking(EBlackboxEnv::Stress);
+        s.SetSelfTvmId(OK_CLIENT);
+        s.SetDiskCacheDir(GetCachePath());
+
+        auto l = MakeIntrusive<TLogger>();
+        {
+            auto u = NTvmApi::TThreadedUpdater::Create(s, l);
+            UNIT_ASSERT_VALUES_EQUAL(TClientStatus::Ok, u->GetStatus().GetCode());
+            UNIT_ASSERT_EXCEPTION(TServiceTicketChecker(u), TBrokenTvmClientSettings);
+        }
+        UNIT_ASSERT_C(l->Stream.Str().find("was successfully fetched") == TString::npos, l->Stream.Str());
+
+        s.EnableServiceTicketChecking();
+        l = MakeIntrusive<TLogger>();
+        {
+            auto u = NTvmApi::TThreadedUpdater::Create(s, l);
+            UNIT_ASSERT_VALUES_EQUAL(TClientStatus::Ok, u->GetStatus().GetCode());
+            TServiceTicketChecker c(u);
+            UNIT_ASSERT(c.Check(SRV_TICKET));
+            UNIT_ASSERT(!c.Check(PROD_TICKET));
+        }
+        UNIT_ASSERT_C(l->Stream.Str().find("was successfully fetched") == TString::npos, l->Stream.Str());
+
+        s.SetSelfTvmId(17);
+        l = MakeIntrusive<TLogger>();
+        {
+            auto u = NTvmApi::TThreadedUpdater::Create(s, l);
+            UNIT_ASSERT_VALUES_EQUAL(TClientStatus::Ok, u->GetStatus().GetCode());
+            TServiceTicketChecker c(u);
+            UNIT_ASSERT(!c.Check(SRV_TICKET));
+            UNIT_ASSERT(!c.Check(PROD_TICKET));
+        }
+        UNIT_ASSERT_C(l->Stream.Str().find("was successfully fetched") == TString::npos, l->Stream.Str());
+    }
+
+    Y_UNIT_TEST(Tickets) {
+        NTvmApi::TClientSettings s;
+        s.SetSelfTvmId(OK_CLIENT);
+        s.EnableServiceTicketsFetchOptions("qwerty", {{"blackbox", 19}});
+        s.SetDiskCacheDir(GetCachePath());
+
+        auto l = MakeIntrusive<TLogger>();
+        {
+            auto u = NTvmApi::TThreadedUpdater::Create(s, l);
+            UNIT_ASSERT_VALUES_EQUAL(TClientStatus::Ok, u->GetStatus().GetCode());
+            TServiceTicketGetter g(u);
+            UNIT_ASSERT_VALUES_EQUAL("3:serv:CBAQ__________9_IgYIKhCUkQY:CX", g.GetTicket("blackbox"));
+            UNIT_ASSERT_EXCEPTION_CONTAINS(g.GetTicket("blackbox2"),
+                                           TBrokenTvmClientSettings,
+                                           "Destination 'blackbox2' was not specified in settings. Check your settings (if you use Qloud/YP/tvmtool - check it's settings)");
+        }
+        UNIT_ASSERT_C(l->Stream.Str().find("was successfully fetched") == TString::npos, l->Stream.Str());
+    }
+
+    Y_UNIT_TEST(ErrorForDst) {
+        TServiceTicketGetter g(new TMockedUpdater);
+
+        UNIT_ASSERT_VALUES_EQUAL(TMockedUpdater::TSettings::CreateDeafult().Backends.at(0).Value,
+                                 g.GetTicket("my_dest"));
+        UNIT_ASSERT_VALUES_EQUAL(TMockedUpdater::TSettings::CreateDeafult().Backends.at(0).Value,
+                                 g.GetTicket(42));
+        UNIT_ASSERT_EXCEPTION_CONTAINS(g.GetTicket("my_bad_dest"),
+                                       TMissingServiceTicket,
+                                       "Failed to get ticket for 'my_bad_dest': Dst is not found");
+        UNIT_ASSERT_EXCEPTION_CONTAINS(g.GetTicket(43),
+                                       TMissingServiceTicket,
+                                       "Failed to get ticket for '43': Dst is not found");
+    }
+}