This commit is contained in:
crn4
2026-03-24 14:58:03 +01:00
parent 38fd73fad6
commit 94149a9441
2 changed files with 5 additions and 3 deletions

View File

@@ -3,6 +3,7 @@ package manager
import ( import (
"context" "context"
"errors" "errors"
"fmt"
"testing" "testing"
"time" "time"
@@ -69,7 +70,7 @@ func (m *mockStore) GetProxyByAccountID(ctx context.Context, accountID string) (
if m.getProxyByAccountIDFunc != nil { if m.getProxyByAccountIDFunc != nil {
return m.getProxyByAccountIDFunc(ctx, accountID) return m.getProxyByAccountIDFunc(ctx, accountID)
} }
return nil, nil return nil, fmt.Errorf("proxy not found for account %s", accountID)
} }
func (m *mockStore) CountProxiesByAccountID(ctx context.Context, accountID string) (int64, error) { func (m *mockStore) CountProxiesByAccountID(ctx context.Context, accountID string) (int64, error) {
if m.countProxiesByAccountIDFunc != nil { if m.countProxiesByAccountIDFunc != nil {

View File

@@ -6,6 +6,7 @@ import (
"crypto/rand" "crypto/rand"
"encoding/base64" "encoding/base64"
"errors" "errors"
"fmt"
"net" "net"
"sync" "sync"
"sync/atomic" "sync/atomic"
@@ -229,8 +230,8 @@ func (m *testProxyManager) CleanupStale(_ context.Context, _ time.Duration) erro
return nil return nil
} }
func (m *testProxyManager) GetAccountProxy(_ context.Context, _ string) (*nbproxy.Proxy, error) { func (m *testProxyManager) GetAccountProxy(_ context.Context, accountID string) (*nbproxy.Proxy, error) {
return nil, nil return nil, fmt.Errorf("proxy not found for account %s", accountID)
} }
func (m *testProxyManager) CountAccountProxies(_ context.Context, _ string) (int64, error) { func (m *testProxyManager) CountAccountProxies(_ context.Context, _ string) (int64, error) {