Function TestGetRoleIDs

db/indextest/indextest_test.go:189–247

(t *testing.T)

}

func TestGetRoleIDs(t *testing.T) {
	testCases := []struct {
		useLegacySyncDocsIndex bool
	}{
		{
			useLegacySyncDocsIndex: false,
		},
		{
			useLegacySyncDocsIndex: true,
		},
	}

	for _, testCase := range testCases {
		t.Run(fmt.Sprintf("useLegacySyncDocsIndex=%t", testCase.useLegacySyncDocsIndex), func(t *testing.T) {
			database := setupIndexAndDB(t, testIndexCreationOptions{
				useLegacySyncDocsIndex: testCase.useLegacySyncDocsIndex,
				numPartitions:          db.DefaultNumIndexPartitions,
				useXattrs:              base.TestUseXattrs(),
			})

			ctx := database.AddDatabaseLogContext(base.TestCtx(t))
			authenticator := database.Authenticator(ctx)

			rolename1 := uuid.NewString()
			rolename2 := uuid.NewString()

			role1, err := authenticator.NewRole(rolename1, nil)
			require.NoError(t, err)
			require.NoError(t, authenticator.Save(role1))

			role2, err := authenticator.NewRole(rolename2, nil)
			require.NoError(t, err)
			require.NoError(t, authenticator.Save(role2))

			err = database.DeleteRole(ctx, role2.Name(), false)
			require.NoError(t, err)
			roleGet, err := authenticator.GetRoleIncDeleted(role2.Name())
			require.NoError(t, err)
			require.True(t, roleGet.IsDeleted())

			t.Log("role1:", role1.Name())
			t.Log("role2:", role2.Name())

			for _, includeDeleted := range []bool{false, true} {
				t.Run(fmt.Sprintf("includeDeleted=%t", includeDeleted), func(t *testing.T) {
					// require roles
					roles, err := database.GetRoleIDs(ctx, database.UseViews(), includeDeleted)
					expectedRoles := []string{role1.Name()}
					if includeDeleted {
						expectedRoles = append(expectedRoles, role2.Name())
					}

					require.NoError(t, err)
					require.ElementsMatch(t, expectedRoles, roles)
				})
			}
		})
	}