mirrors/talos
1
0
Fork 0
mirror of https://github.com/siderolabs/talos.git synced 2025-11-03 09:51:32 +01:00
Code Issues Packages Projects Releases Wiki Activity

test: workaround race in the tests with zaptest package

Browse Source 
Looks like `zaptest` package when used from the goroutine (like in gRPC
server) results in a potential data race on test tear down.

Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
This commit is contained in:
Andrey Smirnov 2021-09-28 23:05:09 +03:00
parent 9c67fde759
commit ecdd7757fb
No known key found for this signature in database
GPG Key ID: 7B26396447AB6DFD
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
internal/app/machined/pkg/controllers/cluster/discovery_service_test.go
View File

@ -8,6 +8,7 @@ import (
"bytes"
"context"
"crypto/aes"
"log"
"net"
"testing"
"time"
@ -23,11 +24,11 @@ import (
"github.com/talos-systems/discovery-service/pkg/client"
"github.com/talos-systems/discovery-service/pkg/server"
"github.com/talos-systems/go-retry/retry"
"go.uber.org/zap/zaptest"
"google.golang.org/grpc"
"inet.af/netaddr"
clusterctrl "github.com/talos-systems/talos/internal/app/machined/pkg/controllers/cluster"
"github.com/talos-systems/talos/pkg/logging"
"github.com/talos-systems/talos/pkg/machinery/config/types/v1alpha1/machine"
"github.com/talos-systems/talos/pkg/machinery/proto"
"github.com/talos-systems/talos/pkg/resources/cluster"
@ -44,7 +45,7 @@ func setupServer(t *testing.T) (address string) {
lis, err := net.Listen("tcp", "localhost:0")
require.NoError(t, err)
logger := zaptest.NewLogger(t)
logger := logging.Wrap(log.Writer())
serverOptions := []grpc.ServerOption{
grpc_middleware.WithUnaryServerChain(
@ -128,7 +129,7 @@ func (suite *DiscoveryServiceSuite) TestReconcile() {
defer cliCtxCancel()
go func() {
errCh <- cli.Run(cliCtx, zaptest.NewLogger(suite.T()), notifyCh)
errCh <- cli.Run(cliCtx, logging.Wrap(log.Writer()), notifyCh)
}()
suite.Assert().NoError(retry.Constant(3*time.Second, retry.WithUnits(100*time.Millisecond)).Retry(