diff options
author | Paul Okstad <pokstad@gitlab.com> | 2020-03-31 21:39:42 +0300 |
---|---|---|
committer | Paul Okstad <pokstad@gitlab.com> | 2020-03-31 21:39:42 +0300 |
commit | 56d460969054483b3e5725d393619569f4299f32 (patch) | |
tree | 24024efde20e702474a74a4c2bd0187988783635 /internal/praefect/nodes/manager_test.go | |
parent | 6f4132638538078909c43ca7bb70405ec3267d45 (diff) | |
parent | 8c67a12861ca7d324eb00d00bc4af8988381736d (diff) |
Merge branch 'sh-refactor-praefect-node-manager' into 'master'
Refactor Praefect node manager
See merge request gitlab-org/gitaly!1940
Diffstat (limited to 'internal/praefect/nodes/manager_test.go')
-rw-r--r-- | internal/praefect/nodes/manager_test.go | 17 |
1 files changed, 11 insertions, 6 deletions
diff --git a/internal/praefect/nodes/manager_test.go b/internal/praefect/nodes/manager_test.go index 8f5e1443b..185c640d6 100644 --- a/internal/praefect/nodes/manager_test.go +++ b/internal/praefect/nodes/manager_test.go @@ -1,6 +1,7 @@ package nodes import ( + "context" "testing" "time" @@ -30,21 +31,25 @@ func TestNodeStatus(t *testing.T) { storageName := "default" cs := newConnectionStatus(models.Node{Storage: storageName}, cc, testhelper.DiscardTestEntry(t), mockHistogramVec) - require.False(t, cs.isHealthy()) - var expectedLabels [][]string for i := 0; i < healthcheckThreshold; i++ { - cs.check() + ctx := context.Background() + status, err := cs.check(ctx) + + require.NoError(t, err) + require.True(t, status) expectedLabels = append(expectedLabels, []string{storageName}) } - require.True(t, cs.isHealthy()) + require.Equal(t, expectedLabels, mockHistogramVec.LabelsCalled()) require.Len(t, mockHistogramVec.Observer().Observed(), healthcheckThreshold) healthSvr.SetServingStatus("", grpc_health_v1.HealthCheckResponse_NOT_SERVING) - cs.check() - require.False(t, cs.isHealthy()) + ctx := context.Background() + status, err := cs.check(ctx) + require.NoError(t, err) + require.False(t, status) } func TestNodeManager(t *testing.T) { |