Function TestRoleQuery

db/indextest/indextest_test.go:27–104

(t *testing.T)

)

func TestRoleQuery(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)
			require.NotNil(t, authenticator, "database.Authenticator(ctx) returned nil")
			// Add roles
			for i := 1; i <= 5; i++ {
				role, err := authenticator.NewRole(fmt.Sprintf("role%d", i), base.SetOf("ABC"))
				require.NoError(t, err, "Error creating new role")
				require.NoError(t, authenticator.Save(role))
			}

			// Delete 1 role
			role1, err := authenticator.NewRole("role1", base.SetOf("ABC"))
			require.NoError(t, err)
			require.NoError(t, database.DeleteRole(ctx, role1.Name(), false))

			testCases := []struct {
				name          string
				expectedRoles int
			}{
				{
					name:          "QueryRoles",
					expectedRoles: 4,
				},
				{
					name:          "QueryAllRoles",
					expectedRoles: 5,
				},
			}
			for _, testCase := range testCases {
				t.Run(testCase.name, func(t *testing.T) {
					var results sgbucket.QueryResultIterator
					var queryErr error
					switch testCase.name {
					case "QueryRoles":
						// Standard query
						results, queryErr = database.QueryRoles(ctx, "", 0)
					case "QueryAllRoles":
						results, queryErr = database.QueryAllRoles(ctx, "", 0)