internal/criadb/db_test.go

// Copyright 2024 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package criadb

import (
	"context"
	"testing"

	"go.chromium.org/luci/server/auth/authtest"
)

func TestIsMemberOfAny(t *testing.T) {
	db := &AuthDatabase{}

	_, err := db.IsMemberOfAny(context.Background(), "test@golang.org", []string{"test-group"})
	if err == nil {
		t.Error("db.IsMemberOfAny returned nil error with uninitialized database")
	}

	db.db = authtest.NewFakeDB(authtest.MockMembership("user:test@golang.org", "test-group"))

	isMember, err := db.IsMemberOfAny(context.Background(), "user:test@golang.org", []string{"test-group", "random-group"})
	if err != nil {
		t.Errorf("db.IsMemberOfAny('user:test@golang.org', []string{'test-group', 'random-group'}) failed: %s", err)
	}
	if !isMember {
		t.Error("db.IsMemberOfAny('user:test@golang.org', []string{'test-group', 'random-group'}) returned false, expected true")
	}

	isMember, err = db.IsMemberOfAny(context.Background(), "user:nope@golang.org", []string{"test-group", "random-group"})
	if err != nil {
		t.Errorf("db.IsMemberOfAny('user:nope@golang.org', []string{'test-group', 'random-group'}) failed: %s", err)
	}
	if isMember {
		t.Error("db.IsMemberOfAny('user:nope@golang.org', []string{'test-group', 'random-group'}) returned true, expected false")
	}
}