diff options
author | Sophia Tan <sophia_tll@hotmail.com> | 2022-01-20 19:18:49 +0000 |
---|---|---|
committer | Evergreen Agent <no-reply@evergreen.mongodb.com> | 2022-01-20 21:36:30 +0000 |
commit | 0f90f936d206032693578db712376489619c0d62 (patch) | |
tree | 8b6f973a9dcfeff7d39ffddb2441e43d3873a43f /src/mongo/rpc | |
parent | 08aa07493315218bb47578ec77c94f2d728fb347 (diff) | |
download | mongo-0f90f936d206032693578db712376489619c0d62.tar.gz |
SERVER-62441 Define a new type mongo::TenantId for tenant id
Diffstat (limited to 'src/mongo/rpc')
-rw-r--r-- | src/mongo/rpc/metadata/security_token_metadata_test.cpp | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/mongo/rpc/metadata/security_token_metadata_test.cpp b/src/mongo/rpc/metadata/security_token_metadata_test.cpp index 1319dae55c7..9fcb2a55adc 100644 --- a/src/mongo/rpc/metadata/security_token_metadata_test.cpp +++ b/src/mongo/rpc/metadata/security_token_metadata_test.cpp @@ -36,6 +36,7 @@ #include "mongo/db/client.h" #include "mongo/db/concurrency/locker_noop_service_context_test_fixture.h" #include "mongo/db/multitenancy_gen.h" +#include "mongo/db/tenant_id.h" #include "mongo/rpc/op_msg_test.h" #include "mongo/unittest/unittest.h" @@ -65,7 +66,7 @@ class SecurityTokenMetadataTest : public LockerNoopServiceContextTest {}; TEST_F(SecurityTokenMetadataTest, SecurityTokenNotAccepted) { const auto kPingBody = BSON(kPingFieldName << 1); - const auto kTokenBody = makeSecurityToken(UserName("user", "admin", OID::gen())); + const auto kTokenBody = makeSecurityToken(UserName("user", "admin", TenantId(OID::gen()))); gMultitenancySupport = false; auto msgBytes = OpMsgBytes{0, kBodySection, kPingBody, kSecurityTokenSection, kTokenBody}; @@ -76,9 +77,9 @@ TEST_F(SecurityTokenMetadataTest, SecurityTokenNotAccepted) { } TEST_F(SecurityTokenMetadataTest, BasicSuccess) { - const auto kOid = OID::gen(); + const auto kTenantId = TenantId(OID::gen()); const auto kPingBody = BSON(kPingFieldName << 1); - const auto kTokenBody = makeSecurityToken(UserName("user", "admin", kOid)); + const auto kTokenBody = makeSecurityToken(UserName("user", "admin", kTenantId)); gMultitenancySupport = true; auto msg = OpMsgBytes{0, kBodySection, kPingBody, kSecurityTokenSection, kTokenBody}.parse(); @@ -97,7 +98,7 @@ TEST_F(SecurityTokenMetadataTest, BasicSuccess) { ASSERT_EQ(authedUser.getUser(), "user"); ASSERT_EQ(authedUser.getDB(), "admin"); ASSERT_TRUE(authedUser.getTenant() != boost::none); - ASSERT_EQ(authedUser.getTenant().get(), kOid); + ASSERT_EQ(authedUser.getTenant().get(), kTenantId); } } // namespace |