Mirrors/gitea
0
0
Fork 0
mirror of https://github.com/go-gitea/gitea.git synced 2026-08-08 14:14:54 +02:00
Code Issues Activity

fix testing context

Browse Source
This commit is contained in:
techknowlogick
2025-09-12 13:31:53 -04:00
parent e26e6282e2
commit 8579b08c9b
1 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/repo/mirror_ssh_keypair_test.go
+13 -14
View File
@@ -7,7 +7,6 @@ import (
"crypto/ed25519"
"testing"
"code.gitea.io/gitea/models/db"
repo_model "code.gitea.io/gitea/models/repo"
"code.gitea.io/gitea/models/unittest"
"code.gitea.io/gitea/modules/setting"
@@ -22,7 +21,7 @@ func TestUserSSHKeypair(t *testing.T) {
t.Run("CreateUserSSHKeypair", func(t *testing.T) {
// Test creating a new SSH keypair for a user
keypair, err := repo_model.CreateUserSSHKeypair(db.DefaultContext, 1)
keypair, err := repo_model.CreateUserSSHKeypair(t.Context(), 1)
require.NoError(t, err)
assert.NotNil(t, keypair)
assert.Equal(t, int64(1), keypair.OwnerID)
@@ -34,7 +33,7 @@ func TestUserSSHKeypair(t *testing.T) {
assert.Contains(t, keypair.PublicKey, "ssh-ed25519")
// Test creating a keypair for an organization
orgKeypair, err := repo_model.CreateUserSSHKeypair(db.DefaultContext, 2)
orgKeypair, err := repo_model.CreateUserSSHKeypair(t.Context(), 2)
require.NoError(t, err)
assert.NotNil(t, orgKeypair)
assert.Equal(t, int64(2), orgKeypair.OwnerID)
@@ -46,18 +45,18 @@ func TestUserSSHKeypair(t *testing.T) {
t.Run("GetUserSSHKeypairByOwner", func(t *testing.T) {
// Create a keypair first
created, err := repo_model.CreateUserSSHKeypair(db.DefaultContext, 3)
created, err := repo_model.CreateUserSSHKeypair(t.Context(), 3)
require.NoError(t, err)
// Test retrieving the keypair
retrieved, err := repo_model.GetUserSSHKeypairByOwner(db.DefaultContext, 3)
retrieved, err := repo_model.GetUserSSHKeypairByOwner(t.Context(), 3)
require.NoError(t, err)
assert.Equal(t, created.OwnerID, retrieved.OwnerID)
assert.Equal(t, created.PublicKey, retrieved.PublicKey)
assert.Equal(t, created.Fingerprint, retrieved.Fingerprint)
// Test retrieving non-existent keypair
_, err = repo_model.GetUserSSHKeypairByOwner(db.DefaultContext, 999)
_, err = repo_model.GetUserSSHKeypairByOwner(t.Context(), 999)
assert.ErrorIs(t, err, util.ErrNotExist)
})
@@ -68,7 +67,7 @@ func TestUserSSHKeypair(t *testing.T) {
}
// Create a keypair
keypair, err := repo_model.CreateUserSSHKeypair(db.DefaultContext, 4)
keypair, err := repo_model.CreateUserSSHKeypair(t.Context(), 4)
require.NoError(t, err)
// Test decrypting the private key
@@ -84,29 +83,29 @@ func TestUserSSHKeypair(t *testing.T) {
t.Run("DeleteUserSSHKeypair", func(t *testing.T) {
// Create a keypair
_, err := repo_model.CreateUserSSHKeypair(db.DefaultContext, 5)
_, err := repo_model.CreateUserSSHKeypair(t.Context(), 5)
require.NoError(t, err)
// Verify it exists
_, err = repo_model.GetUserSSHKeypairByOwner(db.DefaultContext, 5)
_, err = repo_model.GetUserSSHKeypairByOwner(t.Context(), 5)
require.NoError(t, err)
// Delete it
err = repo_model.DeleteUserSSHKeypair(db.DefaultContext, 5)
err = repo_model.DeleteUserSSHKeypair(t.Context(), 5)
require.NoError(t, err)
// Verify it's gone
_, err = repo_model.GetUserSSHKeypairByOwner(db.DefaultContext, 5)
_, err = repo_model.GetUserSSHKeypairByOwner(t.Context(), 5)
assert.ErrorIs(t, err, util.ErrNotExist)
})
t.Run("RegenerateUserSSHKeypair", func(t *testing.T) {
// Create initial keypair
original, err := repo_model.CreateUserSSHKeypair(db.DefaultContext, 6)
original, err := repo_model.CreateUserSSHKeypair(t.Context(), 6)
require.NoError(t, err)
// Regenerate it
regenerated, err := repo_model.RegenerateUserSSHKeypair(db.DefaultContext, 6)
regenerated, err := repo_model.RegenerateUserSSHKeypair(t.Context(), 6)
require.NoError(t, err)
// Verify it's different
@@ -126,7 +125,7 @@ func TestUserSSHKeypairConcurrency(t *testing.T) {
// Test concurrent creation of keypairs to ensure no race conditions
t.Run("ConcurrentCreation", func(t *testing.T) {
ctx := db.DefaultContext
ctx := t.Context()
results := make(chan error, 10)
// Start multiple goroutines creating keypairs for different owners