From b16f27c54b8fcc808667e665d7bf43ade1afb363 Mon Sep 17 00:00:00 2001 From: Egon Elbre Date: Tue, 5 Feb 2019 19:03:16 +0200 Subject: [PATCH] Ensure discovery tests run against postgres (#1230) --- pkg/discovery/service_test.go | 28 +++++++++++----------------- 1 file changed, 11 insertions(+), 17 deletions(-) diff --git a/pkg/discovery/service_test.go b/pkg/discovery/service_test.go index f8d022d4e..8ebf3d368 100644 --- a/pkg/discovery/service_test.go +++ b/pkg/discovery/service_test.go @@ -14,23 +14,17 @@ import ( ) func TestCache_Refresh(t *testing.T) { - ctx := testcontext.New(t) - defer ctx.Cleanup() + testplanet.Run(t, testplanet.Config{ + SatelliteCount: 1, StorageNodeCount: 10, UplinkCount: 0, + }, func(t *testing.T, ctx *testcontext.Context, planet *testplanet.Planet) { + time.Sleep(5 * time.Second) - planet, err := testplanet.New(t, 1, 10, 0) - if err != nil { - t.Fatal(err) - } - defer ctx.Check(planet.Shutdown) - - planet.Start(ctx) - time.Sleep(5 * time.Second) - - satellite := planet.Satellites[0] - for _, storageNode := range planet.StorageNodes { - node, err := satellite.Overlay.Service.Get(ctx, storageNode.ID()) - if assert.NoError(t, err) { - assert.Equal(t, storageNode.Addr(), node.Address.Address) + satellite := planet.Satellites[0] + for _, storageNode := range planet.StorageNodes { + node, err := satellite.Overlay.Service.Get(ctx, storageNode.ID()) + if assert.NoError(t, err) { + assert.Equal(t, storageNode.Addr(), node.Address.Address) + } } - } + }) }