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 (
"context"
"errors"
"fmt"
"testing"
"time"
@@ -69,7 +70,7 @@ func (m *mockStore) GetProxyByAccountID(ctx context.Context, accountID string) (
if m.getProxyByAccountIDFunc != nil {
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) {
if m.countProxiesByAccountIDFunc != nil {

View File

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