diff options
Diffstat (limited to 'internal/praefect/nodes/sql_elector_test.go')
| -rw-r--r-- | internal/praefect/nodes/sql_elector_test.go | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/internal/praefect/nodes/sql_elector_test.go b/internal/praefect/nodes/sql_elector_test.go index adc602952..42765fe59 100644 --- a/internal/praefect/nodes/sql_elector_test.go +++ b/internal/praefect/nodes/sql_elector_test.go @@ -11,6 +11,7 @@ import ( "github.com/stretchr/testify/require" "gitlab.com/gitlab-org/gitaly/internal/praefect/config" "gitlab.com/gitlab-org/gitaly/internal/praefect/datastore/glsql" + "gitlab.com/gitlab-org/gitaly/internal/praefect/middleware" "gitlab.com/gitlab-org/gitaly/internal/praefect/models" "gitlab.com/gitlab-org/gitaly/internal/testhelper" "gitlab.com/gitlab-org/gitaly/internal/testhelper/promtest" @@ -43,7 +44,7 @@ func TestGetPrimaryAndSecondaries(t *testing.T) { storageName := "default" mockHistogramVec0 := promtest.NewMockHistogramVec() - cs0 := newConnectionStatus(models.Node{Storage: storageName + "-0"}, cc0, testhelper.DiscardTestEntry(t), mockHistogramVec0) + cs0 := newConnectionStatus(models.Node{Storage: storageName + "-0"}, cc0, testhelper.DiscardTestEntry(t), middleware.NewErrors(0, 0, 0), mockHistogramVec0) ns := []*nodeStatus{cs0} elector := newSQLElector(shardName, conf, 1, defaultActivePraefectSeconds, db.DB, logger, ns) @@ -98,8 +99,8 @@ func TestBasicFailover(t *testing.T) { storageName := "default" mockHistogramVec0, mockHistogramVec1 := promtest.NewMockHistogramVec(), promtest.NewMockHistogramVec() - cs0 := newConnectionStatus(models.Node{Storage: storageName + "-0"}, cc0, testhelper.DiscardTestEntry(t), mockHistogramVec0) - cs1 := newConnectionStatus(models.Node{Storage: storageName + "-1"}, cc1, testhelper.DiscardTestEntry(t), mockHistogramVec1) + cs0 := newConnectionStatus(models.Node{Storage: storageName + "-0"}, cc0, testhelper.DiscardTestEntry(t), middleware.NewErrors(0, 0, 0), mockHistogramVec0) + cs1 := newConnectionStatus(models.Node{Storage: storageName + "-1"}, cc1, testhelper.DiscardTestEntry(t), middleware.NewErrors(0, 0, 0), mockHistogramVec1) ns := []*nodeStatus{cs0, cs1} failoverTimeSeconds := 1 |