summaryrefslogtreecommitdiff
path: root/src/mongo/rpc
diff options
context:
space:
mode:
authorSophia Tan <sophia_tll@hotmail.com>2022-01-20 19:18:49 +0000
committerEvergreen Agent <no-reply@evergreen.mongodb.com>2022-01-20 21:36:30 +0000
commit0f90f936d206032693578db712376489619c0d62 (patch)
tree8b6f973a9dcfeff7d39ffddb2441e43d3873a43f /src/mongo/rpc
parent08aa07493315218bb47578ec77c94f2d728fb347 (diff)
downloadmongo-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.cpp9
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