mirror of
https://github.com/idanoo/autobrr
synced 2025-07-23 08:49:13 +00:00
6a94ecacca
* fix(auth): implement invalid cookie handling * that escalated quickly * refactor(http): auth handlers * add tests for auth handler * refactor methods * chore(tests): add header and build tag * add build tag integration * chore(tests): run in ci --------- Co-authored-by: ze0s <ze0s@riseup.net>
93 lines
2.9 KiB
Go
93 lines
2.9 KiB
Go
// Copyright (c) 2021 - 2023, Ludvig Lundgren and the autobrr contributors.
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
//go:build integration
|
|
|
|
package database
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/autobrr/autobrr/internal/domain"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestAPIRepo_Store(t *testing.T) {
|
|
for dbType, db := range testDBs {
|
|
log := setupLoggerForTest()
|
|
repo := NewAPIRepo(log, db)
|
|
|
|
t.Run(fmt.Sprintf("Store_Succeeds_With_Valid_Key [%s]", dbType), func(t *testing.T) {
|
|
key := &domain.APIKey{Name: "TestKey", Key: "123", Scopes: []string{"read", "write"}}
|
|
err := repo.Store(context.Background(), key)
|
|
assert.NoError(t, err)
|
|
assert.NotZero(t, key.CreatedAt)
|
|
// Cleanup
|
|
_ = repo.Delete(context.Background(), key.Key)
|
|
})
|
|
|
|
t.Run(fmt.Sprintf("Store_Fails_If_No_Name_Or_Scopes [%s]", dbType), func(t *testing.T) {
|
|
key := &domain.APIKey{Key: "456"}
|
|
err := repo.Store(context.Background(), key)
|
|
assert.Error(t, err) // Should fail when trying to insert a key without scopes (null constraint)
|
|
// Cleanup
|
|
_ = repo.Delete(context.Background(), key.Key)
|
|
})
|
|
|
|
t.Run(fmt.Sprintf("Store_Fails_If_Duplicate_Key [%s]", dbType), func(t *testing.T) {
|
|
key := &domain.APIKey{Key: "789", Scopes: []string{}}
|
|
err1 := repo.Store(context.Background(), key)
|
|
err2 := repo.Store(context.Background(), key)
|
|
assert.NoError(t, err1)
|
|
assert.Error(t, err2) // Should fail when trying to insert a duplicate key
|
|
// Cleanup
|
|
_ = repo.Delete(context.Background(), key.Key)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestAPIRepo_Delete(t *testing.T) {
|
|
for dbType, db := range testDBs {
|
|
log := setupLoggerForTest()
|
|
repo := NewAPIRepo(log, db)
|
|
|
|
t.Run(fmt.Sprintf("Delete_Succeeds_With_Existing_Key [%s]", dbType), func(t *testing.T) {
|
|
key := &domain.APIKey{Name: "TestKey", Key: "123", Scopes: []string{"read", "write"}}
|
|
_ = repo.Store(context.Background(), key)
|
|
err := repo.Delete(context.Background(), key.Key)
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run(fmt.Sprintf("Delete_Succeeds_If_Key_Does_Not_Exist [%s]", dbType), func(t *testing.T) {
|
|
err := repo.Delete(context.Background(), "nonexistent")
|
|
assert.NoError(t, err)
|
|
})
|
|
}
|
|
|
|
}
|
|
|
|
func TestAPIRepo_GetKeys(t *testing.T) {
|
|
for dbType, db := range testDBs {
|
|
log := setupLoggerForTest()
|
|
repo := NewAPIRepo(log, db)
|
|
|
|
t.Run(fmt.Sprintf("GetKeys_Returns_Keys_If_Exists [%s]", dbType), func(t *testing.T) {
|
|
key := &domain.APIKey{Name: "TestKey", Key: "123", Scopes: []string{"read", "write"}}
|
|
_ = repo.Store(context.Background(), key)
|
|
keys, err := repo.GetKeys(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Greater(t, len(keys), 0)
|
|
// Cleanup
|
|
_ = repo.Delete(context.Background(), key.Key)
|
|
})
|
|
|
|
t.Run(fmt.Sprintf("GetKeys_Returns_Empty_If_No_Keys [%s]", dbType), func(t *testing.T) {
|
|
keys, err := repo.GetKeys(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 0, len(keys))
|
|
})
|
|
}
|
|
}
|