Skip to content

Commit

Permalink
tests: add multiple items to custom_db_roles test (#2805)
Browse files Browse the repository at this point in the history
  • Loading branch information
blva authored May 29, 2024
1 parent 301cd36 commit 1f5dadb
Showing 1 changed file with 10 additions and 7 deletions.
17 changes: 10 additions & 7 deletions test/e2e/atlas/custom_db_roles_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ import (

const (
createPrivilege = "UPDATE"
findPrivilege = "FIND"
updatePrivilege = "LIST_SESSIONS"
enableShardingRole = "enableSharding"
enableShardingInheritedRole = "enableSharding@admin"
Expand All @@ -51,7 +52,7 @@ func TestDBRoles(t *testing.T) {
customDBRoleEntity,
"create",
roleName,
"--privilege", createPrivilege+"@db.collection",
"--privilege", fmt.Sprintf("%s@db.collection,%[email protected]", createPrivilege, findPrivilege),
"--inheritedRole", enableShardingInheritedRole,
"-o=json",
)
Expand All @@ -64,8 +65,10 @@ func TestDBRoles(t *testing.T) {

a := assert.New(t)
a.Equal(roleName, role.RoleName)
a.Len(role.GetActions(), 1)
a.Equal(createPrivilege, role.GetActions()[0].Action)
a.Len(role.GetActions(), 2)
a.ElementsMatch(
[]string{role.GetActions()[0].Action, role.GetActions()[1].Action},
[]string{createPrivilege, findPrivilege})
a.Len(role.GetInheritedRoles(), 1)
a.Equal(enableShardingRole, role.GetInheritedRoles()[0].Role)
})
Expand Down Expand Up @@ -100,7 +103,7 @@ func TestDBRoles(t *testing.T) {

a := assert.New(t)
a.Equal(roleName, role.RoleName)
a.Len(role.GetActions(), 1)
a.Len(role.GetActions(), 2)
a.Equal(createPrivilege, role.GetActions()[0].Action)
a.Len(role.GetInheritedRoles(), 1)
a.Equal(enableShardingRole, role.GetInheritedRoles()[0].Role)
Expand All @@ -125,10 +128,10 @@ func TestDBRoles(t *testing.T) {

a := assert.New(t)
a.Equal(roleName, role.RoleName)
a.Len(role.GetActions(), 2)
a.Len(role.GetActions(), 3)
a.ElementsMatch(
[]string{role.GetActions()[0].Action, role.GetActions()[1].Action},
[]string{updatePrivilege, createPrivilege})
[]string{role.GetActions()[0].Action, role.GetActions()[1].Action, role.GetActions()[2].Action},
[]string{updatePrivilege, createPrivilege, findPrivilege})
a.ElementsMatch(
[]string{enableShardingRole, readRole},
[]string{role.GetInheritedRoles()[0].Role, role.GetInheritedRoles()[1].Role})
Expand Down

0 comments on commit 1f5dadb

Please sign in to comment.