李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
63 lines
2.3 KiB
Go
63 lines
2.3 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
"github.com/alicebob/miniredis/v2"
|
|
"github.com/redis/go-redis/v9"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGatewayCacheLiveCallIdentityAndController(t *testing.T) {
|
|
redisServer := miniredis.RunT(t)
|
|
client := redis.NewClient(&redis.Options{Addr: redisServer.Addr()})
|
|
cache, ok := NewGatewayCache(client).(service.LiveCallStore)
|
|
require.True(t, ok)
|
|
otherInstance, ok := NewGatewayCache(client).(service.LiveCallStore)
|
|
require.True(t, ok)
|
|
record := &service.LiveCallRecord{
|
|
CallID: "call_secret",
|
|
CallHash: HashLiveCallID("call_secret"),
|
|
AccountID: 11,
|
|
APIKeyID: 22,
|
|
UserID: 33,
|
|
GroupID: 44,
|
|
LeaseID: "lease",
|
|
Model: "gpt-live-test",
|
|
AttestationCiphertext: "encrypted-attestation",
|
|
CreatedAt: time.Now(),
|
|
ExpiresAt: time.Now().Add(time.Hour),
|
|
Controller: service.LiveControllerPending,
|
|
}
|
|
require.NoError(t, cache.SaveLiveCall(context.Background(), record, time.Hour))
|
|
|
|
loaded, err := otherInstance.GetLiveCall(context.Background(), record.CallHash)
|
|
require.NoError(t, err)
|
|
require.Equal(t, record.CallID, loaded.CallID)
|
|
require.Equal(t, record.AccountID, loaded.AccountID)
|
|
require.Equal(t, record.AttestationCiphertext, loaded.AttestationCiphertext)
|
|
|
|
claimed, err := cache.ClaimLiveController(context.Background(), record.CallHash, service.LiveControllerObserver, "observer-1")
|
|
require.NoError(t, err)
|
|
require.True(t, claimed)
|
|
claimed, err = cache.ClaimLiveController(context.Background(), record.CallHash, service.LiveControllerProxy, "proxy-1")
|
|
require.NoError(t, err)
|
|
require.True(t, claimed)
|
|
controller, err := cache.GetLiveController(context.Background(), record.CallHash)
|
|
require.NoError(t, err)
|
|
require.Equal(t, service.LiveControllerProxy, controller)
|
|
|
|
released, err := cache.ReleaseLiveController(context.Background(), record.CallHash, "proxy-1")
|
|
require.NoError(t, err)
|
|
require.True(t, released)
|
|
closed, err := cache.MarkLiveCallClosed(context.Background(), record.CallHash, time.Hour)
|
|
require.NoError(t, err)
|
|
require.True(t, closed)
|
|
closed, err = cache.MarkLiveCallClosed(context.Background(), record.CallHash, time.Hour)
|
|
require.NoError(t, err)
|
|
require.False(t, closed)
|
|
}
|