#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"); } }