owncast/storage/userrepository/externalAPIUser_test.go

package userrepository

import (
	"testing"

	"github.com/owncast/owncast/storage/configrepository"
	"github.com/owncast/owncast/storage/data"
)

const (
	tokenName = "test token name"
	token     = "test-token-123"
)

var (
	testScopes       = []string{"test-scope"}
	userRepository   *SqlUserRepository
	configRepository configrepository.ConfigRepository
)

func TestMain(m *testing.M) {
	ds, err := data.NewStore(":memory:")
	if err != nil {
		panic(err)
	}

	userRepository = New(ds)

	m.Run()
}

func TestCreateExternalAPIUser(t *testing.T) {
	if err := userRepository.InsertExternalAPIUser(token, tokenName, 0, testScopes); err != nil {
		t.Fatal(err)
	}

	user := userRepository.GetUserByToken(token)
	if user == nil {
		t.Fatal("api user not found after creating")
	}

	if user.DisplayName != tokenName {
		t.Errorf("expected display name %q, got %q", tokenName, user.DisplayName)
	}

	if user.Scopes[0] != testScopes[0] {
		t.Errorf("expected scopes %q, got %q", testScopes, user.Scopes)
	}
}

func TestDeleteExternalAPIUser(t *testing.T) {
	if err := userRepository.DeleteExternalAPIUser(token); err != nil {
		t.Fatal(err)
	}
}

func TestVerifyTokenDisabled(t *testing.T) {
	users, err := userRepository.GetExternalAPIUser()
	if err != nil {
		t.Fatal(err)
	}

	if len(users) > 0 {
		t.Fatal("disabled user returned in list of all API users")
	}
}

func TestVerifyGetUserTokenDisabled(t *testing.T) {
	user := userRepository.GetUserByToken(token)
	if user == nil {
		t.Fatal("user not returned in GetUserByToken after disabling")
	}

	if user.DisabledAt == nil {
		t.Fatal("user returned in GetUserByToken after disabling")
	}
}

func TestVerifyGetExternalAPIUserForAccessTokenAndScopeTokenDisabled(t *testing.T) {
	user, _ := userRepository.GetExternalAPIUserForAccessTokenAndScope(token, testScopes[0])

	if user != nil {
		t.Fatal("user returned in GetExternalAPIUserForAccessTokenAndScope after disabling")
	}
}

func TestCreateAdditionalAPIUser(t *testing.T) {
	if err := userRepository.InsertExternalAPIUser("ignore-me", "token-to-be-ignored", 0, testScopes); err != nil {
		t.Fatal(err)
	}
}

func TestAgainVerifyGetExternalAPIUserForAccessTokenAndScopeTokenDisabled(t *testing.T) {
	user, _ := userRepository.GetExternalAPIUserForAccessTokenAndScope(token, testScopes[0])

	if user != nil {
		t.Fatal("user returned in TestAgainVerifyGetExternalAPIUserForAccessTokenAndScopeTokenDisabled after disabling")
	}
}