fix_: address PR comments
This commit is contained in:
parent
137920cd1c
commit
eb10b20137
|
@ -323,11 +323,11 @@ func TestCircuitBreaker_SuccessCallStatus(t *testing.T) {
|
|||
assert.Len(t, result.FunctorCallStatuses(), 1)
|
||||
|
||||
status := result.FunctorCallStatuses()[0]
|
||||
if status.name != "successCircuit" {
|
||||
t.Errorf("Expected functor name to be 'successCircuit', got %s", status.name)
|
||||
if status.Name != "successCircuit" {
|
||||
t.Errorf("Expected functor name to be 'successCircuit', got %s", status.Name)
|
||||
}
|
||||
if status.err != nil {
|
||||
t.Errorf("Expected no error in functor status, got %v", status.err)
|
||||
if status.Err != nil {
|
||||
t.Errorf("Expected no error in functor status, got %v", status.Err)
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -350,11 +350,11 @@ func TestCircuitBreaker_ErrorCallStatus(t *testing.T) {
|
|||
assert.Len(t, result.FunctorCallStatuses(), 1)
|
||||
|
||||
status := result.FunctorCallStatuses()[0]
|
||||
if status.name != "errorCircuit" {
|
||||
t.Errorf("Expected functor name to be 'errorCircuit', got %s", status.name)
|
||||
if status.Name != "errorCircuit" {
|
||||
t.Errorf("Expected functor name to be 'errorCircuit', got %s", status.Name)
|
||||
}
|
||||
if !errors.Is(status.err, expectedError) {
|
||||
t.Errorf("Expected functor error to be '%v', got '%v'", expectedError, status.err)
|
||||
if !errors.Is(status.Err, expectedError) {
|
||||
t.Errorf("Expected functor error to be '%v', got '%v'", expectedError, status.Err)
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -405,11 +405,11 @@ func TestCircuitBreaker_MultipleFunctorsResult(t *testing.T) {
|
|||
statuses := result.FunctorCallStatuses()
|
||||
require.Len(t, statuses, 2)
|
||||
|
||||
require.Equal(t, statuses[0].name, "circuit1")
|
||||
require.NotNil(t, statuses[0].err)
|
||||
require.Equal(t, statuses[0].Name, "circuit1")
|
||||
require.NotNil(t, statuses[0].Err)
|
||||
|
||||
require.Equal(t, statuses[1].name, "circuit2")
|
||||
require.Nil(t, statuses[1].err)
|
||||
require.Equal(t, statuses[1].Name, "circuit2")
|
||||
require.Nil(t, statuses[1].Err)
|
||||
}
|
||||
|
||||
func TestCircuitBreaker_LastFunctorDirectExecution(t *testing.T) {
|
||||
|
@ -444,9 +444,9 @@ func TestCircuitBreaker_LastFunctorDirectExecution(t *testing.T) {
|
|||
statuses := result.FunctorCallStatuses()
|
||||
require.Len(t, statuses, 2)
|
||||
|
||||
require.Equal(t, statuses[0].name, "circuitName")
|
||||
require.NotNil(t, statuses[0].err)
|
||||
require.Equal(t, statuses[0].Name, "circuitName")
|
||||
require.NotNil(t, statuses[0].Err)
|
||||
|
||||
require.Equal(t, statuses[1].name, "circuitName")
|
||||
require.Nil(t, statuses[1].err)
|
||||
require.Equal(t, statuses[1].Name, "circuitName")
|
||||
require.Nil(t, statuses[1].Err)
|
||||
}
|
||||
|
|
|
@ -10,14 +10,14 @@ import (
|
|||
// Aggregator manages and aggregates the statuses of multiple providers.
|
||||
type Aggregator struct {
|
||||
mu sync.RWMutex
|
||||
Name string
|
||||
name string
|
||||
providerStatuses map[string]*rpcstatus.ProviderStatus
|
||||
}
|
||||
|
||||
// NewAggregator creates a new instance of Aggregator with the given name.
|
||||
func NewAggregator(name string) *Aggregator {
|
||||
return &Aggregator{
|
||||
Name: name,
|
||||
name: name,
|
||||
providerStatuses: make(map[string]*rpcstatus.ProviderStatus),
|
||||
}
|
||||
}
|
||||
|
@ -100,7 +100,7 @@ func (a *Aggregator) ComputeAggregatedStatus() rpcstatus.ProviderStatus {
|
|||
}
|
||||
|
||||
aggregatedStatus := rpcstatus.ProviderStatus{
|
||||
Name: a.Name,
|
||||
Name: a.name,
|
||||
LastSuccessAt: lastSuccessAt,
|
||||
LastErrorAt: lastErrorAt,
|
||||
LastError: lastError,
|
||||
|
|
|
@ -5,9 +5,10 @@ import (
|
|||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/status-im/status-go/healthmanager/rpcstatus"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/suite"
|
||||
|
||||
"github.com/status-im/status-go/healthmanager/rpcstatus"
|
||||
)
|
||||
|
||||
// StatusAggregatorTestSuite defines the test suite for Aggregator.
|
||||
|
@ -23,7 +24,7 @@ func (suite *StatusAggregatorTestSuite) SetupTest() {
|
|||
|
||||
// TestNewAggregator verifies that a new Aggregator is initialized correctly.
|
||||
func (suite *StatusAggregatorTestSuite) TestNewAggregator() {
|
||||
assert.Equal(suite.T(), "TestAggregator", suite.aggregator.Name, "Aggregator name should be set correctly")
|
||||
assert.Equal(suite.T(), "TestAggregator", suite.aggregator.name, "Aggregator name should be set correctly")
|
||||
assert.Empty(suite.T(), suite.aggregator.providerStatuses, "Aggregator should have no providers initially")
|
||||
}
|
||||
|
||||
|
@ -94,7 +95,7 @@ func (suite *StatusAggregatorTestSuite) TestUpdate() {
|
|||
func (suite *StatusAggregatorTestSuite) TestComputeAggregatedStatus_NoProviders() {
|
||||
aggStatus := suite.aggregator.ComputeAggregatedStatus()
|
||||
|
||||
assert.Equal(suite.T(), rpcstatus.StatusUnknown, aggStatus.Status, "Aggregated status should be 'unknown' when no providers are registered")
|
||||
assert.Equal(suite.T(), rpcstatus.StatusDown, aggStatus.Status, "Aggregated status should be 'down' when no providers are registered")
|
||||
assert.True(suite.T(), aggStatus.LastSuccessAt.IsZero(), "LastSuccessAt should be zero when no providers are registered")
|
||||
assert.True(suite.T(), aggStatus.LastErrorAt.IsZero(), "LastErrorAt should be zero when no providers are registered")
|
||||
}
|
||||
|
|
|
@ -2,15 +2,16 @@ package healthmanager
|
|||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"sync"
|
||||
|
||||
"github.com/status-im/status-go/healthmanager/aggregator"
|
||||
"github.com/status-im/status-go/healthmanager/rpcstatus"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// BlockchainFullStatus contains the full status of the blockchain, including provider statuses.
|
||||
type BlockchainFullStatus struct {
|
||||
Status rpcstatus.ProviderStatus `json:"status"`
|
||||
StatusPerChain map[uint64]rpcstatus.ProviderStatus `json:"statusPerChain"`
|
||||
StatusPerChainPerProvider map[uint64]map[string]rpcstatus.ProviderStatus `json:"statusPerChainPerProvider"`
|
||||
}
|
||||
|
||||
|
@ -28,7 +29,7 @@ type BlockchainHealthManager struct {
|
|||
|
||||
providers map[uint64]*ProvidersHealthManager
|
||||
cancelFuncs map[uint64]context.CancelFunc // Map chainID to cancel functions
|
||||
lastStatus BlockchainStatus
|
||||
lastStatus *BlockchainStatus
|
||||
wg sync.WaitGroup
|
||||
}
|
||||
|
||||
|
@ -43,30 +44,37 @@ func NewBlockchainHealthManager() *BlockchainHealthManager {
|
|||
}
|
||||
|
||||
// RegisterProvidersHealthManager registers the provider health manager.
|
||||
// It prevents registering the same provider twice for the same chain.
|
||||
// It removes any existing provider for the same chain before registering the new one.
|
||||
func (b *BlockchainHealthManager) RegisterProvidersHealthManager(ctx context.Context, phm *ProvidersHealthManager) error {
|
||||
b.mu.Lock()
|
||||
defer b.mu.Unlock()
|
||||
|
||||
// Check if the provider for the given chainID is already registered
|
||||
if _, exists := b.providers[phm.ChainID()]; exists {
|
||||
// Log a warning or return an error to indicate that the provider is already registered
|
||||
return fmt.Errorf("provider for chainID %d is already registered", phm.ChainID())
|
||||
chainID := phm.ChainID()
|
||||
|
||||
// Check if a provider for the given chainID is already registered and remove it
|
||||
if _, exists := b.providers[chainID]; exists {
|
||||
// Cancel the existing context
|
||||
if cancel, cancelExists := b.cancelFuncs[chainID]; cancelExists {
|
||||
cancel()
|
||||
}
|
||||
// Remove the old registration
|
||||
delete(b.providers, chainID)
|
||||
delete(b.cancelFuncs, chainID)
|
||||
}
|
||||
|
||||
// Proceed with the registration
|
||||
b.providers[phm.ChainID()] = phm
|
||||
b.providers[chainID] = phm
|
||||
|
||||
// Create a new context for the provider
|
||||
providerCtx, cancel := context.WithCancel(ctx)
|
||||
b.cancelFuncs[phm.ChainID()] = cancel
|
||||
b.cancelFuncs[chainID] = cancel
|
||||
|
||||
statusCh := phm.Subscribe()
|
||||
b.wg.Add(1)
|
||||
go func(phm *ProvidersHealthManager, statusCh chan struct{}, providerCtx context.Context) {
|
||||
defer func() {
|
||||
b.wg.Done()
|
||||
phm.Unsubscribe(statusCh)
|
||||
b.wg.Done()
|
||||
}()
|
||||
for {
|
||||
select {
|
||||
|
@ -91,6 +99,7 @@ func (b *BlockchainHealthManager) Stop() {
|
|||
cancel()
|
||||
}
|
||||
clear(b.cancelFuncs)
|
||||
clear(b.providers)
|
||||
|
||||
b.mu.Unlock()
|
||||
b.wg.Wait()
|
||||
|
@ -134,15 +143,15 @@ func (b *BlockchainHealthManager) aggregateAndUpdateStatus(ctx context.Context)
|
|||
b.aggregator.UpdateBatch(providerStatuses)
|
||||
|
||||
// Get the new aggregated full and short status
|
||||
newShortStatus := b.getShortStatus()
|
||||
newShortStatus := b.getStatusPerChain()
|
||||
b.mu.Unlock()
|
||||
|
||||
// Compare full and short statuses and emit if changed
|
||||
if !compareShortStatus(newShortStatus, b.lastStatus) {
|
||||
if b.lastStatus == nil || !compareShortStatus(newShortStatus, *b.lastStatus) {
|
||||
b.emitBlockchainHealthStatus(ctx)
|
||||
b.mu.Lock()
|
||||
defer b.mu.Unlock()
|
||||
b.lastStatus = newShortStatus
|
||||
b.lastStatus = &newShortStatus
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -192,15 +201,16 @@ func (b *BlockchainHealthManager) GetFullStatus() BlockchainFullStatus {
|
|||
statusPerChainPerProvider[chainID] = providerStatuses
|
||||
}
|
||||
|
||||
blockchainStatus := b.aggregator.GetAggregatedStatus()
|
||||
statusPerChain := b.getStatusPerChain()
|
||||
|
||||
return BlockchainFullStatus{
|
||||
Status: blockchainStatus,
|
||||
Status: statusPerChain.Status,
|
||||
StatusPerChain: statusPerChain.StatusPerChain,
|
||||
StatusPerChainPerProvider: statusPerChainPerProvider,
|
||||
}
|
||||
}
|
||||
|
||||
func (b *BlockchainHealthManager) getShortStatus() BlockchainStatus {
|
||||
func (b *BlockchainHealthManager) getStatusPerChain() BlockchainStatus {
|
||||
statusPerChain := make(map[uint64]rpcstatus.ProviderStatus)
|
||||
|
||||
for chainID, phm := range b.providers {
|
||||
|
@ -216,10 +226,10 @@ func (b *BlockchainHealthManager) getShortStatus() BlockchainStatus {
|
|||
}
|
||||
}
|
||||
|
||||
func (b *BlockchainHealthManager) GetShortStatus() BlockchainStatus {
|
||||
func (b *BlockchainHealthManager) GetStatusPerChain() BlockchainStatus {
|
||||
b.mu.RLock()
|
||||
defer b.mu.RUnlock()
|
||||
return b.getShortStatus()
|
||||
return b.getStatusPerChain()
|
||||
}
|
||||
|
||||
// Status returns the current aggregated status.
|
||||
|
|
|
@ -8,8 +8,9 @@ import (
|
|||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/status-im/status-go/healthmanager/rpcstatus"
|
||||
"github.com/stretchr/testify/suite"
|
||||
|
||||
"github.com/status-im/status-go/healthmanager/rpcstatus"
|
||||
)
|
||||
|
||||
type BlockchainHealthManagerSuite struct {
|
||||
|
@ -50,7 +51,8 @@ func (s *BlockchainHealthManagerSuite) assertBlockChainStatus(expected rpcstatus
|
|||
// Test registering a provider health manager
|
||||
func (s *BlockchainHealthManagerSuite) TestRegisterProvidersHealthManager() {
|
||||
phm := NewProvidersHealthManager(1) // Create a real ProvidersHealthManager
|
||||
s.manager.RegisterProvidersHealthManager(context.Background(), phm)
|
||||
err := s.manager.RegisterProvidersHealthManager(context.Background(), phm)
|
||||
s.Require().NoError(err)
|
||||
|
||||
// Verify that the provider is registered
|
||||
s.Require().NotNil(s.manager.providers[1])
|
||||
|
@ -59,7 +61,8 @@ func (s *BlockchainHealthManagerSuite) TestRegisterProvidersHealthManager() {
|
|||
// Test status updates and notifications
|
||||
func (s *BlockchainHealthManagerSuite) TestStatusUpdateNotification() {
|
||||
phm := NewProvidersHealthManager(1)
|
||||
s.manager.RegisterProvidersHealthManager(context.Background(), phm)
|
||||
err := s.manager.RegisterProvidersHealthManager(context.Background(), phm)
|
||||
s.Require().NoError(err)
|
||||
ch := s.manager.Subscribe()
|
||||
|
||||
// Update the provider status
|
||||
|
@ -75,8 +78,10 @@ func (s *BlockchainHealthManagerSuite) TestGetFullStatus() {
|
|||
phm1 := NewProvidersHealthManager(1)
|
||||
phm2 := NewProvidersHealthManager(2)
|
||||
ctx := context.Background()
|
||||
s.manager.RegisterProvidersHealthManager(ctx, phm1)
|
||||
s.manager.RegisterProvidersHealthManager(ctx, phm2)
|
||||
err := s.manager.RegisterProvidersHealthManager(ctx, phm1)
|
||||
s.Require().NoError(err)
|
||||
err = s.manager.RegisterProvidersHealthManager(ctx, phm2)
|
||||
s.Require().NoError(err)
|
||||
ch := s.manager.Subscribe()
|
||||
|
||||
// Update the provider status
|
||||
|
@ -120,7 +125,8 @@ func (s *BlockchainHealthManagerSuite) TestConcurrency() {
|
|||
defer cancel()
|
||||
for i := 1; i <= chainsCount; i++ {
|
||||
phm := NewProvidersHealthManager(uint64(i))
|
||||
s.manager.RegisterProvidersHealthManager(ctx, phm)
|
||||
err := s.manager.RegisterProvidersHealthManager(ctx, phm)
|
||||
s.Require().NoError(err)
|
||||
}
|
||||
|
||||
ch := s.manager.Subscribe()
|
||||
|
@ -161,7 +167,8 @@ func (s *BlockchainHealthManagerSuite) TestUnsubscribeOneOfMultipleSubscribers()
|
|||
// Create an instance of BlockchainHealthManager and register a provider manager
|
||||
phm := NewProvidersHealthManager(1)
|
||||
ctx, cancel := context.WithCancel(s.ctx)
|
||||
s.manager.RegisterProvidersHealthManager(ctx, phm)
|
||||
err := s.manager.RegisterProvidersHealthManager(ctx, phm)
|
||||
s.Require().NoError(err)
|
||||
|
||||
defer cancel()
|
||||
|
||||
|
@ -196,7 +203,8 @@ func (s *BlockchainHealthManagerSuite) TestUnsubscribeOneOfMultipleSubscribers()
|
|||
func (s *BlockchainHealthManagerSuite) TestMixedProviderStatusInSingleChain() {
|
||||
// Register a provider for chain 1
|
||||
phm := NewProvidersHealthManager(1)
|
||||
s.manager.RegisterProvidersHealthManager(s.ctx, phm)
|
||||
err := s.manager.RegisterProvidersHealthManager(s.ctx, phm)
|
||||
s.Require().NoError(err)
|
||||
|
||||
// Subscribe to status updates
|
||||
ch := s.manager.Subscribe()
|
||||
|
@ -212,7 +220,7 @@ func (s *BlockchainHealthManagerSuite) TestMixedProviderStatusInSingleChain() {
|
|||
s.waitForUpdate(ch, rpcstatus.StatusUp, 100*time.Millisecond)
|
||||
|
||||
// Verify that the short status reflects the chain as down, since one provider is down
|
||||
shortStatus := s.manager.GetShortStatus()
|
||||
shortStatus := s.manager.GetStatusPerChain()
|
||||
s.Equal(rpcstatus.StatusUp, shortStatus.Status.Status)
|
||||
s.Equal(rpcstatus.StatusUp, shortStatus.StatusPerChain[1].Status) // Chain 1 should be marked as down
|
||||
}
|
||||
|
|
|
@ -2,10 +2,11 @@ package provider_errors
|
|||
|
||||
import (
|
||||
"errors"
|
||||
"strings"
|
||||
|
||||
"github.com/ethereum/go-ethereum"
|
||||
"github.com/ethereum/go-ethereum/core"
|
||||
"github.com/ethereum/go-ethereum/rpc"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type RpcProviderErrorType string
|
||||
|
|
|
@ -5,24 +5,25 @@ import (
|
|||
"fmt"
|
||||
"sync"
|
||||
|
||||
statusaggregator "github.com/status-im/status-go/healthmanager/aggregator"
|
||||
"github.com/status-im/status-go/healthmanager/aggregator"
|
||||
"github.com/status-im/status-go/healthmanager/rpcstatus"
|
||||
)
|
||||
|
||||
type ProvidersHealthManager struct {
|
||||
mu sync.RWMutex
|
||||
chainID uint64
|
||||
aggregator *statusaggregator.Aggregator
|
||||
aggregator *aggregator.Aggregator
|
||||
subscribers []chan struct{}
|
||||
lastStatus *rpcstatus.ProviderStatus
|
||||
}
|
||||
|
||||
// NewProvidersHealthManager creates a new instance of ProvidersHealthManager with the given chain ID.
|
||||
func NewProvidersHealthManager(chainID uint64) *ProvidersHealthManager {
|
||||
aggregator := statusaggregator.NewAggregator(fmt.Sprintf("%d", chainID))
|
||||
agg := aggregator.NewAggregator(fmt.Sprintf("%d", chainID))
|
||||
|
||||
return &ProvidersHealthManager{
|
||||
chainID: chainID,
|
||||
aggregator: aggregator,
|
||||
aggregator: agg,
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -30,8 +31,6 @@ func NewProvidersHealthManager(chainID uint64) *ProvidersHealthManager {
|
|||
func (p *ProvidersHealthManager) Update(ctx context.Context, callStatuses []rpcstatus.RpcProviderCallStatus) {
|
||||
p.mu.Lock()
|
||||
|
||||
previousStatus := p.aggregator.GetAggregatedStatus()
|
||||
|
||||
// Update the aggregator with the new provider statuses
|
||||
for _, rpcCallStatus := range callStatuses {
|
||||
providerStatus := rpcstatus.NewRpcProviderStatus(rpcCallStatus)
|
||||
|
@ -40,7 +39,7 @@ func (p *ProvidersHealthManager) Update(ctx context.Context, callStatuses []rpcs
|
|||
|
||||
newStatus := p.aggregator.GetAggregatedStatus()
|
||||
|
||||
shouldEmit := newStatus.Status != previousStatus.Status
|
||||
shouldEmit := p.lastStatus == nil || p.lastStatus.Status != newStatus.Status
|
||||
p.mu.Unlock()
|
||||
|
||||
if !shouldEmit {
|
||||
|
@ -48,6 +47,10 @@ func (p *ProvidersHealthManager) Update(ctx context.Context, callStatuses []rpcs
|
|||
}
|
||||
|
||||
p.emitChainStatus(ctx)
|
||||
p.mu.Lock()
|
||||
defer p.mu.Unlock()
|
||||
p.lastStatus = &newStatus
|
||||
|
||||
}
|
||||
|
||||
// GetStatuses returns a copy of the current provider statuses.
|
||||
|
@ -95,7 +98,7 @@ func (p *ProvidersHealthManager) UnsubscribeAll() {
|
|||
func (p *ProvidersHealthManager) Reset() {
|
||||
p.mu.Lock()
|
||||
defer p.mu.Unlock()
|
||||
p.aggregator = statusaggregator.NewAggregator(fmt.Sprintf("%d", p.chainID))
|
||||
p.aggregator = aggregator.NewAggregator(fmt.Sprintf("%d", p.chainID))
|
||||
}
|
||||
|
||||
// Status Returns the current aggregated status
|
||||
|
|
|
@ -4,11 +4,13 @@ import (
|
|||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"github.com/status-im/status-go/healthmanager/rpcstatus"
|
||||
"github.com/stretchr/testify/suite"
|
||||
"sync"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/suite"
|
||||
|
||||
"github.com/status-im/status-go/healthmanager/rpcstatus"
|
||||
)
|
||||
|
||||
type ProvidersHealthManagerSuite struct {
|
||||
|
@ -139,9 +141,12 @@ func (s *BlockchainHealthManagerSuite) TestInterleavedChainStatusChanges() {
|
|||
phm1 := NewProvidersHealthManager(1)
|
||||
phm2 := NewProvidersHealthManager(2)
|
||||
phm3 := NewProvidersHealthManager(3)
|
||||
s.manager.RegisterProvidersHealthManager(s.ctx, phm1)
|
||||
s.manager.RegisterProvidersHealthManager(s.ctx, phm2)
|
||||
s.manager.RegisterProvidersHealthManager(s.ctx, phm3)
|
||||
err := s.manager.RegisterProvidersHealthManager(s.ctx, phm1)
|
||||
s.Require().NoError(err)
|
||||
err = s.manager.RegisterProvidersHealthManager(s.ctx, phm2)
|
||||
s.Require().NoError(err)
|
||||
err = s.manager.RegisterProvidersHealthManager(s.ctx, phm3)
|
||||
s.Require().NoError(err)
|
||||
|
||||
// Subscribe to status updates
|
||||
ch := s.manager.Subscribe()
|
||||
|
@ -163,7 +168,7 @@ func (s *BlockchainHealthManagerSuite) TestInterleavedChainStatusChanges() {
|
|||
s.waitForUpdate(ch, rpcstatus.StatusUp, 100*time.Millisecond)
|
||||
|
||||
// Check that short status correctly reflects the mixed state
|
||||
shortStatus := s.manager.GetShortStatus()
|
||||
shortStatus := s.manager.GetStatusPerChain()
|
||||
s.Equal(rpcstatus.StatusUp, shortStatus.Status.Status)
|
||||
s.Equal(rpcstatus.StatusDown, shortStatus.StatusPerChain[1].Status) // Chain 1 is down
|
||||
s.Equal(rpcstatus.StatusUp, shortStatus.StatusPerChain[2].Status) // Chain 2 is still up
|
||||
|
@ -174,8 +179,10 @@ func (s *BlockchainHealthManagerSuite) TestDelayedChainUpdate() {
|
|||
// Register providers for chains 1 and 2
|
||||
phm1 := NewProvidersHealthManager(1)
|
||||
phm2 := NewProvidersHealthManager(2)
|
||||
s.manager.RegisterProvidersHealthManager(s.ctx, phm1)
|
||||
s.manager.RegisterProvidersHealthManager(s.ctx, phm2)
|
||||
err := s.manager.RegisterProvidersHealthManager(s.ctx, phm1)
|
||||
s.Require().NoError(err)
|
||||
err = s.manager.RegisterProvidersHealthManager(s.ctx, phm2)
|
||||
s.Require().NoError(err)
|
||||
|
||||
// Subscribe to status updates
|
||||
ch := s.manager.Subscribe()
|
||||
|
@ -195,7 +202,7 @@ func (s *BlockchainHealthManagerSuite) TestDelayedChainUpdate() {
|
|||
s.waitForUpdate(ch, rpcstatus.StatusDown, 100*time.Millisecond)
|
||||
|
||||
// Check that short status reflects the final state where both chains are down
|
||||
shortStatus := s.manager.GetShortStatus()
|
||||
shortStatus := s.manager.GetStatusPerChain()
|
||||
s.Equal(rpcstatus.StatusDown, shortStatus.Status.Status)
|
||||
s.Equal(rpcstatus.StatusDown, shortStatus.StatusPerChain[1].Status) // Chain 1 is down
|
||||
s.Equal(rpcstatus.StatusDown, shortStatus.StatusPerChain[2].Status) // Chain 2 is down
|
||||
|
|
|
@ -17,11 +17,11 @@ const (
|
|||
|
||||
// ProviderStatus holds the status information for a single provider.
|
||||
type ProviderStatus struct {
|
||||
Name string
|
||||
LastSuccessAt time.Time
|
||||
LastErrorAt time.Time
|
||||
LastError error
|
||||
Status StatusType
|
||||
Name string `json:"name"`
|
||||
LastSuccessAt time.Time `json:"last_success_at"`
|
||||
LastErrorAt time.Time `json:"last_error_at"`
|
||||
LastError error `json:"last_error"`
|
||||
Status StatusType `json:"status"`
|
||||
}
|
||||
|
||||
// ProviderCallStatus represents the result of an arbitrary provider call.
|
||||
|
|
|
@ -2,9 +2,10 @@ package rpcstatus
|
|||
|
||||
import (
|
||||
"errors"
|
||||
"github.com/status-im/status-go/rpc/chain/rpclimiter"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/status-im/status-go/rpc/chain/rpclimiter"
|
||||
)
|
||||
|
||||
func TestNewRpcProviderStatus(t *testing.T) {
|
||||
|
|
|
@ -332,7 +332,16 @@ func (n *StatusNode) setupRPCClient() (err error) {
|
|||
},
|
||||
}
|
||||
|
||||
n.rpcClient, err = rpc.NewClient(gethNodeClient, n.config.NetworkID, n.config.UpstreamConfig, n.config.Networks, n.appDB, &n.walletFeed, providerConfigs)
|
||||
config := rpc.ClientConfig{
|
||||
Client: gethNodeClient,
|
||||
UpstreamChainID: n.config.NetworkID,
|
||||
UpstreamConfig: n.config.UpstreamConfig,
|
||||
Networks: n.config.Networks,
|
||||
DB: n.appDB,
|
||||
WalletFeed: &n.walletFeed,
|
||||
ProviderConfigs: providerConfigs,
|
||||
}
|
||||
n.rpcClient, err = rpc.NewClient(config)
|
||||
n.rpcClient.Start(context.Background())
|
||||
if err != nil {
|
||||
return
|
||||
|
|
|
@ -5,19 +5,22 @@ import (
|
|||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"github.com/status-im/status-go/healthmanager"
|
||||
"github.com/status-im/status-go/healthmanager/rpcstatus"
|
||||
mockEthclient "github.com/status-im/status-go/rpc/chain/ethclient/mock/client/ethclient"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/ethereum/go-ethereum/common"
|
||||
"github.com/ethereum/go-ethereum/core/types"
|
||||
"github.com/status-im/status-go/healthmanager"
|
||||
"github.com/status-im/status-go/healthmanager/rpcstatus"
|
||||
mockEthclient "github.com/status-im/status-go/rpc/chain/ethclient/mock/client/ethclient"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/stretchr/testify/suite"
|
||||
|
||||
"github.com/status-im/status-go/rpc/chain/ethclient"
|
||||
"github.com/ethereum/go-ethereum/common"
|
||||
"github.com/ethereum/go-ethereum/core/types"
|
||||
|
||||
"go.uber.org/mock/gomock"
|
||||
|
||||
"github.com/status-im/status-go/rpc/chain/ethclient"
|
||||
)
|
||||
|
||||
type BlockchainHealthManagerSuite struct {
|
||||
|
@ -53,7 +56,8 @@ func (s *BlockchainHealthManagerSuite) setupClients(chainIDs []uint64) {
|
|||
phm := healthmanager.NewProvidersHealthManager(chainID)
|
||||
client := NewClient([]ethclient.RPSLimitedEthClientInterface{mockEthClient}, chainID, phm)
|
||||
|
||||
s.blockchainHealthManager.RegisterProvidersHealthManager(ctx, phm)
|
||||
err := s.blockchainHealthManager.RegisterProvidersHealthManager(ctx, phm)
|
||||
require.NoError(s.T(), err)
|
||||
|
||||
s.mockProviders[chainID] = phm
|
||||
s.mockEthClients[chainID] = mockEthClient
|
||||
|
@ -272,7 +276,7 @@ func (s *BlockchainHealthManagerSuite) TestGetShortStatus() {
|
|||
s.waitForStatus(statusCh, rpcstatus.StatusUp)
|
||||
|
||||
// Get the short status from the BlockchainHealthManager
|
||||
shortStatus := s.blockchainHealthManager.GetShortStatus()
|
||||
shortStatus := s.blockchainHealthManager.GetStatusPerChain()
|
||||
|
||||
// Assert overall blockchain status
|
||||
require.Equal(s.T(), rpcstatus.StatusUp, shortStatus.Status.Status)
|
||||
|
|
|
@ -3,19 +3,21 @@ package chain
|
|||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"github.com/ethereum/go-ethereum/core/vm"
|
||||
"strconv"
|
||||
"testing"
|
||||
|
||||
"github.com/ethereum/go-ethereum/core/vm"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/stretchr/testify/suite"
|
||||
"go.uber.org/mock/gomock"
|
||||
|
||||
"github.com/ethereum/go-ethereum/common"
|
||||
"github.com/ethereum/go-ethereum/core/types"
|
||||
healthManager "github.com/status-im/status-go/healthmanager"
|
||||
"github.com/status-im/status-go/healthmanager/rpcstatus"
|
||||
"github.com/status-im/status-go/rpc/chain/ethclient"
|
||||
"github.com/status-im/status-go/rpc/chain/rpclimiter"
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/stretchr/testify/suite"
|
||||
"go.uber.org/mock/gomock"
|
||||
|
||||
mockEthclient "github.com/status-im/status-go/rpc/chain/ethclient/mock/client/ethclient"
|
||||
)
|
||||
|
@ -162,7 +164,7 @@ func (s *ClientWithFallbackSuite) TestVMErrorDoesNotMarkChainDown() {
|
|||
require.Equal(s.T(), providerStatuses["test0"].Status, rpcstatus.StatusUp)
|
||||
}
|
||||
|
||||
func (s *ClientWithFallbackSuite) TestNoClientsChainUnknown() {
|
||||
func (s *ClientWithFallbackSuite) TestNoClientsChainDown() {
|
||||
s.setupClients(0)
|
||||
|
||||
ctx := context.Background()
|
||||
|
@ -174,7 +176,7 @@ func (s *ClientWithFallbackSuite) TestNoClientsChainUnknown() {
|
|||
|
||||
// THEN
|
||||
chainStatus := s.providersHealthManager.Status()
|
||||
require.Equal(s.T(), rpcstatus.StatusUnknown, chainStatus.Status)
|
||||
require.Equal(s.T(), rpcstatus.StatusDown, chainStatus.Status)
|
||||
}
|
||||
|
||||
func (s *ClientWithFallbackSuite) TestAllClientsDifferentErrors() {
|
||||
|
@ -228,11 +230,11 @@ func (s *ClientWithFallbackSuite) TestAllClientsNetworkErrors() {
|
|||
require.Equal(s.T(), providerStatuses["test2"].Status, rpcstatus.StatusDown)
|
||||
}
|
||||
|
||||
func (s *ClientWithFallbackSuite) TestChainStatusUnknownWhenAllProvidersUnknown() {
|
||||
func (s *ClientWithFallbackSuite) TestChainStatusDownWhenInitial() {
|
||||
s.setupClients(2)
|
||||
|
||||
chainStatus := s.providersHealthManager.Status()
|
||||
require.Equal(s.T(), rpcstatus.StatusUnknown, chainStatus.Status)
|
||||
require.Equal(s.T(), rpcstatus.StatusDown, chainStatus.Status)
|
||||
}
|
||||
|
||||
func TestClientWithFallbackSuite(t *testing.T) {
|
||||
|
|
|
@ -124,35 +124,46 @@ type Client struct {
|
|||
// Is initialized in a build-tag-dependent module
|
||||
var verifProxyInitFn func(c *Client)
|
||||
|
||||
// ClientConfig holds the configuration for initializing a new Client.
|
||||
type ClientConfig struct {
|
||||
Client *gethrpc.Client
|
||||
UpstreamChainID uint64
|
||||
UpstreamConfig params.UpstreamRPCConfig
|
||||
Networks []params.Network
|
||||
DB *sql.DB
|
||||
WalletFeed *event.Feed
|
||||
ProviderConfigs []params.ProviderConfig
|
||||
}
|
||||
|
||||
// NewClient initializes Client and tries to connect to both,
|
||||
// upstream and local node.
|
||||
//
|
||||
// Client is safe for concurrent use and will automatically
|
||||
// reconnect to the server if connection is lost.
|
||||
func NewClient(client *gethrpc.Client, upstreamChainID uint64, upstream params.UpstreamRPCConfig, networks []params.Network, db *sql.DB, walletFeed *event.Feed, providerConfigs []params.ProviderConfig) (*Client, error) {
|
||||
func NewClient(config ClientConfig) (*Client, error) {
|
||||
var err error
|
||||
|
||||
log := log.New("package", "status-go/rpc.Client")
|
||||
networkManager := network.NewManager(db)
|
||||
networkManager := network.NewManager(config.DB)
|
||||
if networkManager == nil {
|
||||
return nil, errors.New("failed to create network manager")
|
||||
}
|
||||
|
||||
err = networkManager.Init(networks)
|
||||
err = networkManager.Init(config.Networks)
|
||||
if err != nil {
|
||||
log.Error("Network manager failed to initialize", "error", err)
|
||||
}
|
||||
|
||||
c := Client{
|
||||
local: client,
|
||||
local: config.Client,
|
||||
NetworkManager: networkManager,
|
||||
handlers: make(map[string]Handler),
|
||||
rpcClients: make(map[uint64]chain.ClientInterface),
|
||||
limiterPerProvider: make(map[string]*rpclimiter.RPCRpsLimiter),
|
||||
log: log,
|
||||
providerConfigs: providerConfigs,
|
||||
providerConfigs: config.ProviderConfigs,
|
||||
healthMgr: healthmanager.NewBlockchainHealthManager(),
|
||||
walletFeed: walletFeed,
|
||||
walletFeed: config.WalletFeed,
|
||||
}
|
||||
|
||||
var opts []gethrpc.ClientOption
|
||||
|
@ -162,10 +173,10 @@ func NewClient(client *gethrpc.Client, upstreamChainID uint64, upstream params.U
|
|||
}),
|
||||
)
|
||||
|
||||
if upstream.Enabled {
|
||||
c.UpstreamChainID = upstreamChainID
|
||||
c.upstreamEnabled = upstream.Enabled
|
||||
c.upstreamURL = upstream.URL
|
||||
if config.UpstreamConfig.Enabled {
|
||||
c.UpstreamChainID = config.UpstreamChainID
|
||||
c.upstreamEnabled = config.UpstreamConfig.Enabled
|
||||
c.upstreamURL = config.UpstreamConfig.URL
|
||||
upstreamClient, err := gethrpc.DialOptions(context.Background(), c.upstreamURL, opts...)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("dial upstream server: %s", err)
|
||||
|
@ -180,14 +191,17 @@ func NewClient(client *gethrpc.Client, upstreamChainID uint64, upstream params.U
|
|||
}
|
||||
|
||||
// Include the chain-id in the rpc client
|
||||
rpcName := fmt.Sprintf("%s-chain-id-%d", hostPortUpstream, upstreamChainID)
|
||||
rpcName := fmt.Sprintf("%s-chain-id-%d", hostPortUpstream, config.UpstreamChainID)
|
||||
|
||||
ethClients := []ethclient.RPSLimitedEthClientInterface{
|
||||
ethclient.NewRPSLimitedEthClient(upstreamClient, limiter, rpcName),
|
||||
}
|
||||
phm := healthmanager.NewProvidersHealthManager(upstreamChainID)
|
||||
c.healthMgr.RegisterProvidersHealthManager(context.Background(), phm)
|
||||
c.upstream = chain.NewClient(ethClients, upstreamChainID, phm)
|
||||
phm := healthmanager.NewProvidersHealthManager(config.UpstreamChainID)
|
||||
err = c.healthMgr.RegisterProvidersHealthManager(context.Background(), phm)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("register providers health manager: %s", err)
|
||||
}
|
||||
c.upstream = chain.NewClient(ethClients, config.UpstreamChainID, phm)
|
||||
}
|
||||
|
||||
c.router = newRouter(c.upstreamEnabled)
|
||||
|
@ -309,7 +323,10 @@ func (c *Client) getClientUsingCache(chainID uint64) (chain.ClientInterface, err
|
|||
}
|
||||
|
||||
phm := healthmanager.NewProvidersHealthManager(chainID)
|
||||
c.healthMgr.RegisterProvidersHealthManager(context.Background(), phm)
|
||||
err := c.healthMgr.RegisterProvidersHealthManager(context.Background(), phm)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("register providers health manager: %s", err)
|
||||
}
|
||||
|
||||
client := chain.NewClient(ethClients, chainID, phm)
|
||||
c.rpcClients[chainID] = client
|
||||
|
@ -456,7 +473,10 @@ func (c *Client) UpdateUpstreamURL(url string) error {
|
|||
ethclient.NewRPSLimitedEthClient(rpcClient, rpsLimiter, hostPortUpstream),
|
||||
}
|
||||
phm := healthmanager.NewProvidersHealthManager(c.UpstreamChainID)
|
||||
c.healthMgr.RegisterProvidersHealthManager(context.Background(), phm)
|
||||
err = c.healthMgr.RegisterProvidersHealthManager(context.Background(), phm)
|
||||
if err != nil {
|
||||
return fmt.Errorf("register providers health manager: %s", err)
|
||||
}
|
||||
c.upstream = chain.NewClient(ethClients, c.UpstreamChainID, phm)
|
||||
c.upstreamURL = url
|
||||
c.Unlock()
|
||||
|
|
|
@ -44,7 +44,16 @@ func TestBlockedRoutesCall(t *testing.T) {
|
|||
gethRPCClient, err := gethrpc.Dial(ts.URL)
|
||||
require.NoError(t, err)
|
||||
|
||||
c, err := NewClient(gethRPCClient, 1, params.UpstreamRPCConfig{Enabled: false, URL: ""}, []params.Network{}, db, nil, nil)
|
||||
config := ClientConfig{
|
||||
Client: gethRPCClient,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{Enabled: false, URL: ""},
|
||||
Networks: []params.Network{},
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
c, err := NewClient(config)
|
||||
require.NoError(t, err)
|
||||
|
||||
for _, m := range blockedMethods {
|
||||
|
@ -83,7 +92,16 @@ func TestBlockedRoutesRawCall(t *testing.T) {
|
|||
gethRPCClient, err := gethrpc.Dial(ts.URL)
|
||||
require.NoError(t, err)
|
||||
|
||||
c, err := NewClient(gethRPCClient, 1, params.UpstreamRPCConfig{Enabled: false, URL: ""}, []params.Network{}, db, nil, nil)
|
||||
config := ClientConfig{
|
||||
Client: gethRPCClient,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{Enabled: false, URL: ""},
|
||||
Networks: []params.Network{},
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
c, err := NewClient(config)
|
||||
require.NoError(t, err)
|
||||
|
||||
for _, m := range blockedMethods {
|
||||
|
@ -110,7 +128,16 @@ func TestUpdateUpstreamURL(t *testing.T) {
|
|||
gethRPCClient, err := gethrpc.Dial(ts.URL)
|
||||
require.NoError(t, err)
|
||||
|
||||
c, err := NewClient(gethRPCClient, 1, params.UpstreamRPCConfig{Enabled: true, URL: ts.URL}, []params.Network{}, db, nil, nil)
|
||||
config := ClientConfig{
|
||||
Client: gethRPCClient,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{Enabled: true, URL: ts.URL},
|
||||
Networks: []params.Network{},
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
c, err := NewClient(config)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, ts.URL, c.upstreamURL)
|
||||
|
||||
|
@ -183,7 +210,18 @@ func TestGetClientsUsingCache(t *testing.T) {
|
|||
DefaultFallbackURL2: server.URL + path3,
|
||||
},
|
||||
}
|
||||
c, err := NewClient(nil, 1, params.UpstreamRPCConfig{}, networks, db, nil, providerConfigs)
|
||||
|
||||
config := ClientConfig{
|
||||
Client: nil,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{},
|
||||
Networks: networks,
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: providerConfigs,
|
||||
}
|
||||
|
||||
c, err := NewClient(config)
|
||||
require.NoError(t, err)
|
||||
|
||||
// Networks from DB must pick up DefaultRPCURL, DefaultFallbackURL, DefaultFallbackURL2
|
||||
|
|
|
@ -48,7 +48,17 @@ func (s *ProxySuite) startRpcClient(infuraURL string) *Client {
|
|||
|
||||
db, close := setupTestNetworkDB(s.T())
|
||||
defer close()
|
||||
c, err := NewClient(gethRPCClient, 1, params.UpstreamRPCConfig{Enabled: true, URL: infuraURL}, []params.Network{}, db)
|
||||
|
||||
config := ClientConfig{
|
||||
Client: gethRPCClient,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{Enabled: true, URL: infuraURL},
|
||||
Networks: []params.Network{},
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
c, err := NewClient(config)
|
||||
require.NoError(s.T(), err)
|
||||
|
||||
return c
|
||||
|
|
|
@ -40,7 +40,16 @@ func setupTestAPI(t *testing.T) (*API, func()) {
|
|||
|
||||
_ = client
|
||||
|
||||
rpcClient, err := statusRPC.NewClient(nil, 1, upstreamConfig, nil, db, nil)
|
||||
config := statusRPC.ClientConfig{
|
||||
Client: nil,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: upstreamConfig,
|
||||
Networks: nil,
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
rpcClient, err := statusRPC.NewClient(config)
|
||||
require.NoError(t, err)
|
||||
|
||||
// import account keys
|
||||
|
|
|
@ -144,7 +144,16 @@ func TestAPI_GetAddressDetails(t *testing.T) {
|
|||
DefaultFallbackURL: serverWith1SecDelay.URL,
|
||||
},
|
||||
}
|
||||
c, err := rpc.NewClient(nil, chainID, params.UpstreamRPCConfig{}, networks, appDB, providerConfigs)
|
||||
config := rpc.ClientConfig{
|
||||
Client: nil,
|
||||
UpstreamChainID: chainID,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{},
|
||||
Networks: networks,
|
||||
DB: appDB,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: providerConfigs,
|
||||
}
|
||||
c, err := rpc.NewClient(config)
|
||||
require.NoError(t, err)
|
||||
|
||||
chainClient, err := c.EthClient(chainID)
|
||||
|
|
|
@ -405,7 +405,17 @@ func Test_removeBalanceHistoryOnEventAccountRemoved(t *testing.T) {
|
|||
txServiceMockCtrl := gomock.NewController(t)
|
||||
server, _ := fake.NewTestServer(txServiceMockCtrl)
|
||||
client := gethrpc.DialInProc(server)
|
||||
rpcClient, _ := rpc.NewClient(client, chainID, params.UpstreamRPCConfig{}, nil, appDB, nil)
|
||||
|
||||
config := rpc.ClientConfig{
|
||||
Client: client,
|
||||
UpstreamChainID: chainID,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{},
|
||||
Networks: nil,
|
||||
DB: appDB,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
rpcClient, _ := rpc.NewClient(config)
|
||||
rpcClient.UpstreamChainID = chainID
|
||||
|
||||
service := NewService(walletDB, accountsDB, &accountFeed, &walletFeed, rpcClient, nil, nil, nil)
|
||||
|
|
|
@ -51,7 +51,7 @@ func (s *MarketTestSuite) TestEventOnRpsError() {
|
|||
s.Require().Equal(event.Type, EventMarketStatusChanged)
|
||||
}
|
||||
|
||||
func (s *MarketTestSuite) TestNoEventOnNetworkError() {
|
||||
func (s *MarketTestSuite) TestEventOnNetworkError() {
|
||||
ctrl := gomock.NewController(s.T())
|
||||
defer ctrl.Finish()
|
||||
|
||||
|
@ -62,10 +62,11 @@ func (s *MarketTestSuite) TestNoEventOnNetworkError() {
|
|||
|
||||
_, err := manager.FetchPrices(s.symbols, s.currencies)
|
||||
s.Require().Error(err, "expected error from FetchPrices due to MockPriceProviderWithError")
|
||||
_, ok := s.feedSub.WaitForEvent(time.Millisecond * 500)
|
||||
event, ok := s.feedSub.WaitForEvent(500 * time.Millisecond)
|
||||
s.Require().True(ok, "expected an event, but none was received")
|
||||
|
||||
//THEN
|
||||
s.Require().False(ok, "expected no event, but one was received")
|
||||
// THEN
|
||||
s.Require().Equal(event.Type, EventMarketStatusChanged)
|
||||
}
|
||||
|
||||
func TestMarketTestSuite(t *testing.T) {
|
||||
|
|
|
@ -92,7 +92,16 @@ func setupTestNetworkDB(t *testing.T) (*sql.DB, func()) {
|
|||
func setupRouter(t *testing.T) (*Router, func()) {
|
||||
db, cleanTmpDb := setupTestNetworkDB(t)
|
||||
|
||||
client, _ := rpc.NewClient(nil, 1, params.UpstreamRPCConfig{Enabled: false, URL: ""}, defaultNetworks, db, nil)
|
||||
config := rpc.ClientConfig{
|
||||
Client: nil,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{Enabled: false, URL: ""},
|
||||
Networks: defaultNetworks,
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
client, _ := rpc.NewClient(config)
|
||||
|
||||
router := NewRouter(client, nil, nil, nil, nil, nil, nil, nil)
|
||||
|
||||
|
|
|
@ -331,7 +331,18 @@ func Test_removeTokenBalanceOnEventAccountRemoved(t *testing.T) {
|
|||
txServiceMockCtrl := gomock.NewController(t)
|
||||
server, _ := fake.NewTestServer(txServiceMockCtrl)
|
||||
client := gethrpc.DialInProc(server)
|
||||
rpcClient, _ := rpc.NewClient(client, chainID, params.UpstreamRPCConfig{}, nil, appDB, nil)
|
||||
|
||||
config := rpc.ClientConfig{
|
||||
Client: client,
|
||||
UpstreamChainID: chainID,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{},
|
||||
Networks: nil,
|
||||
DB: appDB,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
rpcClient, _ := rpc.NewClient(config)
|
||||
|
||||
rpcClient.UpstreamChainID = chainID
|
||||
nm := network.NewManager(appDB)
|
||||
mediaServer, err := mediaserver.NewMediaServer(appDB, nil, nil, walletDB)
|
||||
|
|
|
@ -10,6 +10,10 @@ import (
|
|||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/status-im/status-go/contracts"
|
||||
"github.com/status-im/status-go/services/wallet/blockchainstate"
|
||||
"github.com/status-im/status-go/t/utils"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
"github.com/stretchr/testify/mock"
|
||||
"go.uber.org/mock/gomock"
|
||||
|
@ -24,30 +28,26 @@ import (
|
|||
"github.com/ethereum/go-ethereum/event"
|
||||
"github.com/ethereum/go-ethereum/rpc"
|
||||
"github.com/status-im/status-go/appdatabase"
|
||||
"github.com/status-im/status-go/contracts"
|
||||
"github.com/status-im/status-go/contracts/balancechecker"
|
||||
"github.com/status-im/status-go/contracts/ethscan"
|
||||
"github.com/status-im/status-go/contracts/ierc20"
|
||||
ethtypes "github.com/status-im/status-go/eth-node/types"
|
||||
ethclient "github.com/status-im/status-go/rpc/chain/ethclient"
|
||||
mock_client "github.com/status-im/status-go/rpc/chain/mock/client"
|
||||
"github.com/status-im/status-go/rpc/chain/rpclimiter"
|
||||
mock_rpcclient "github.com/status-im/status-go/rpc/mock/client"
|
||||
"github.com/status-im/status-go/server"
|
||||
"github.com/status-im/status-go/services/wallet/async"
|
||||
"github.com/status-im/status-go/services/wallet/balance"
|
||||
"github.com/status-im/status-go/services/wallet/blockchainstate"
|
||||
"github.com/status-im/status-go/services/wallet/community"
|
||||
"github.com/status-im/status-go/t/helpers"
|
||||
"github.com/status-im/status-go/t/utils"
|
||||
|
||||
"github.com/status-im/status-go/multiaccounts/accounts"
|
||||
multicommon "github.com/status-im/status-go/multiaccounts/common"
|
||||
"github.com/status-im/status-go/params"
|
||||
statusRpc "github.com/status-im/status-go/rpc"
|
||||
ethclient "github.com/status-im/status-go/rpc/chain/ethclient"
|
||||
mock_client "github.com/status-im/status-go/rpc/chain/mock/client"
|
||||
"github.com/status-im/status-go/rpc/chain/rpclimiter"
|
||||
mock_rpcclient "github.com/status-im/status-go/rpc/mock/client"
|
||||
"github.com/status-im/status-go/rpc/network"
|
||||
"github.com/status-im/status-go/server"
|
||||
"github.com/status-im/status-go/services/wallet/async"
|
||||
"github.com/status-im/status-go/services/wallet/balance"
|
||||
walletcommon "github.com/status-im/status-go/services/wallet/common"
|
||||
"github.com/status-im/status-go/services/wallet/community"
|
||||
"github.com/status-im/status-go/services/wallet/token"
|
||||
"github.com/status-im/status-go/t/helpers"
|
||||
"github.com/status-im/status-go/transactions"
|
||||
"github.com/status-im/status-go/walletdatabase"
|
||||
)
|
||||
|
@ -1079,7 +1079,18 @@ func setupFindBlocksCommand(t *testing.T, accountAddress common.Address, fromBlo
|
|||
|
||||
return nil
|
||||
}
|
||||
client, _ := statusRpc.NewClient(nil, 1, params.UpstreamRPCConfig{Enabled: false, URL: ""}, []params.Network{}, db, nil)
|
||||
|
||||
config := statusRpc.ClientConfig{
|
||||
Client: nil,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{Enabled: false, URL: ""},
|
||||
Networks: []params.Network{},
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
client, _ := statusRpc.NewClient(config)
|
||||
|
||||
client.SetClient(tc.NetworkID(), tc)
|
||||
tokenManager := token.NewTokenManager(db, client, community.NewManager(appdb, nil, nil), network.NewManager(appdb), appdb, mediaServer, nil, nil, nil, token.NewPersistence(db))
|
||||
tokenManager.SetTokens([]*token.Token{
|
||||
|
@ -1342,7 +1353,17 @@ func TestFetchTransfersForLoadedBlocks(t *testing.T) {
|
|||
currentBlock: 100,
|
||||
}
|
||||
|
||||
client, _ := statusRpc.NewClient(nil, 1, params.UpstreamRPCConfig{Enabled: false, URL: ""}, []params.Network{}, db, nil)
|
||||
config := statusRpc.ClientConfig{
|
||||
Client: nil,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{Enabled: false, URL: ""},
|
||||
Networks: []params.Network{},
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
client, _ := statusRpc.NewClient(config)
|
||||
|
||||
client.SetClient(tc.NetworkID(), tc)
|
||||
tokenManager := token.NewTokenManager(db, client, community.NewManager(appdb, nil, nil), network.NewManager(appdb), appdb, mediaServer, nil, nil, nil, token.NewPersistence(db))
|
||||
|
||||
|
@ -1466,7 +1487,17 @@ func TestFetchNewBlocksCommand_findBlocksWithEthTransfers(t *testing.T) {
|
|||
currentBlock: 100,
|
||||
}
|
||||
|
||||
client, _ := statusRpc.NewClient(nil, 1, params.UpstreamRPCConfig{Enabled: false, URL: ""}, []params.Network{}, db, nil)
|
||||
config := statusRpc.ClientConfig{
|
||||
Client: nil,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{Enabled: false, URL: ""},
|
||||
Networks: []params.Network{},
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
client, _ := statusRpc.NewClient(config)
|
||||
|
||||
client.SetClient(tc.NetworkID(), tc)
|
||||
tokenManager := token.NewTokenManager(db, client, community.NewManager(appdb, nil, nil), network.NewManager(appdb), appdb, mediaServer, nil, nil, nil, token.NewPersistence(db))
|
||||
|
||||
|
@ -1546,7 +1577,17 @@ func TestFetchNewBlocksCommand_nonceDetection(t *testing.T) {
|
|||
mediaServer, err := server.NewMediaServer(appdb, nil, nil, db)
|
||||
require.NoError(t, err)
|
||||
|
||||
client, _ := statusRpc.NewClient(nil, 1, params.UpstreamRPCConfig{Enabled: false, URL: ""}, []params.Network{}, db, nil)
|
||||
config := statusRpc.ClientConfig{
|
||||
Client: nil,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{Enabled: false, URL: ""},
|
||||
Networks: []params.Network{},
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
client, _ := statusRpc.NewClient(config)
|
||||
|
||||
client.SetClient(tc.NetworkID(), tc)
|
||||
tokenManager := token.NewTokenManager(db, client, community.NewManager(appdb, nil, nil), network.NewManager(appdb), appdb, mediaServer, nil, nil, nil, token.NewPersistence(db))
|
||||
|
||||
|
@ -1660,7 +1701,17 @@ func TestFetchNewBlocksCommand(t *testing.T) {
|
|||
}
|
||||
//tc.printPreparedData = true
|
||||
|
||||
client, _ := statusRpc.NewClient(nil, 1, params.UpstreamRPCConfig{Enabled: false, URL: ""}, []params.Network{}, db, nil)
|
||||
config := statusRpc.ClientConfig{
|
||||
Client: nil,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{Enabled: false, URL: ""},
|
||||
Networks: []params.Network{},
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
client, _ := statusRpc.NewClient(config)
|
||||
|
||||
client.SetClient(tc.NetworkID(), tc)
|
||||
|
||||
tokenManager := token.NewTokenManager(db, client, community.NewManager(appdb, nil, nil), network.NewManager(appdb), appdb, mediaServer, nil, nil, nil, token.NewPersistence(db))
|
||||
|
@ -1799,7 +1850,17 @@ func TestLoadBlocksAndTransfersCommand_FiniteFinishedInfiniteRunning(t *testing.
|
|||
db, err := helpers.SetupTestMemorySQLDB(walletdatabase.DbInitializer{})
|
||||
require.NoError(t, err)
|
||||
|
||||
client, _ := statusRpc.NewClient(nil, 1, params.UpstreamRPCConfig{Enabled: false, URL: ""}, []params.Network{}, db, nil)
|
||||
config := statusRpc.ClientConfig{
|
||||
Client: nil,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{Enabled: false, URL: ""},
|
||||
Networks: []params.Network{},
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
client, _ := statusRpc.NewClient(config)
|
||||
|
||||
maker, _ := contracts.NewContractMaker(client)
|
||||
|
||||
wdb := NewDB(db)
|
||||
|
|
|
@ -45,7 +45,16 @@ func setupTestAPI(t *testing.T) (*API, func()) {
|
|||
server, _ := fake.NewTestServer(txServiceMockCtrl)
|
||||
client := gethrpc.DialInProc(server)
|
||||
|
||||
rpcClient, err := statusRPC.NewClient(client, 1, upstreamConfig, nil, db, nil)
|
||||
config := statusRPC.ClientConfig{
|
||||
Client: client,
|
||||
UpstreamChainID: 1,
|
||||
UpstreamConfig: upstreamConfig,
|
||||
Networks: nil,
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
rpcClient, err := statusRPC.NewClient(config)
|
||||
require.NoError(t, err)
|
||||
|
||||
// import account keys
|
||||
|
|
|
@ -10,6 +10,8 @@ import (
|
|||
"github.com/stretchr/testify/suite"
|
||||
"go.uber.org/mock/gomock"
|
||||
|
||||
statusRpc "github.com/status-im/status-go/rpc"
|
||||
|
||||
"github.com/ethereum/go-ethereum/common"
|
||||
"github.com/ethereum/go-ethereum/common/hexutil"
|
||||
gethtypes "github.com/ethereum/go-ethereum/core/types"
|
||||
|
@ -22,7 +24,6 @@ import (
|
|||
"github.com/status-im/status-go/eth-node/crypto"
|
||||
"github.com/status-im/status-go/eth-node/types"
|
||||
"github.com/status-im/status-go/params"
|
||||
"github.com/status-im/status-go/rpc"
|
||||
wallet_common "github.com/status-im/status-go/services/wallet/common"
|
||||
"github.com/status-im/status-go/sqlite"
|
||||
"github.com/status-im/status-go/t/utils"
|
||||
|
@ -56,7 +57,18 @@ func (s *TransactorSuite) SetupTest() {
|
|||
chainID := gethparams.AllEthashProtocolChanges.ChainID.Uint64()
|
||||
db, err := sqlite.OpenUnecryptedDB(sqlite.InMemoryPath) // dummy to make rpc.Client happy
|
||||
s.Require().NoError(err)
|
||||
rpcClient, _ := rpc.NewClient(s.client, chainID, params.UpstreamRPCConfig{}, nil, db, nil)
|
||||
|
||||
config := statusRpc.ClientConfig{
|
||||
Client: s.client,
|
||||
UpstreamChainID: chainID,
|
||||
UpstreamConfig: params.UpstreamRPCConfig{},
|
||||
Networks: nil,
|
||||
DB: db,
|
||||
WalletFeed: nil,
|
||||
ProviderConfigs: nil,
|
||||
}
|
||||
rpcClient, _ := statusRpc.NewClient(config)
|
||||
|
||||
rpcClient.UpstreamChainID = chainID
|
||||
nodeConfig, err := utils.MakeTestNodeConfigWithDataDir("", "/tmp", chainID)
|
||||
s.Require().NoError(err)
|
||||
|
|
Loading…
Reference in New Issue