Skip to content

Commit aca475c

Browse files
author
Dmitriy Matrenichev
committed
chore: small usability fixes
* Replace logging.Wrap(log.Writer()) with zaptest.NewLogger(suite.T()) where possible. * Replace reflect.DeepEqual with =|slices.Equal|bytes.Equal where possible. Signed-off-by: Dmitriy Matrenichev <dmitry.matrenichev@siderolabs.com>
1 parent 26cf566 commit aca475c

64 files changed

Lines changed: 157 additions & 223 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

internal/app/apid/main.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@ import (
1313
"fmt"
1414
"log"
1515
"os/signal"
16-
"reflect"
1716
"regexp"
17+
"slices"
1818
"syscall"
1919
"time"
2020

@@ -260,7 +260,7 @@ func verifyExtKeyUsage(rawCerts [][]byte, verifiedChains [][]*x509.Certificate)
260260
continue
261261
}
262262

263-
if !reflect.DeepEqual(cert.ExtKeyUsage, []x509.ExtKeyUsage{x509.ExtKeyUsageClientAuth}) {
263+
if !slices.Equal(cert.ExtKeyUsage, []x509.ExtKeyUsage{x509.ExtKeyUsageClientAuth}) {
264264
return fmt.Errorf("certificate %q is missing the client auth extended key usage", cert.Subject)
265265
}
266266
}

internal/app/machined/pkg/controllers/cluster/cluster_test.go

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@ package cluster_test
66

77
import (
88
"context"
9-
"log"
109
"sync"
1110
"time"
1211

@@ -17,8 +16,7 @@ import (
1716
"github.com/cosi-project/runtime/pkg/state/impl/namespaced"
1817
"github.com/siderolabs/go-retry/retry"
1918
"github.com/stretchr/testify/suite"
20-
21-
"github.com/siderolabs/talos/pkg/logging"
19+
"go.uber.org/zap/zaptest"
2220
)
2321

2422
type ClusterSuite struct {
@@ -40,9 +38,7 @@ func (suite *ClusterSuite) SetupTest() {
4038

4139
var err error
4240

43-
logger := logging.Wrap(log.Writer())
44-
45-
suite.runtime, err = runtime.NewRuntime(suite.state, logger)
41+
suite.runtime, err = runtime.NewRuntime(suite.state, zaptest.NewLogger(suite.T()))
4642
suite.Require().NoError(err)
4743
}
4844

internal/app/machined/pkg/controllers/cluster/discovery_service_test.go

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@ import (
1010
"crypto/rand"
1111
"encoding/base64"
1212
"io"
13-
"log"
1413
"net/netip"
1514
"net/url"
1615
"testing"
@@ -22,11 +21,11 @@ import (
2221
"github.com/siderolabs/go-retry/retry"
2322
"github.com/stretchr/testify/assert"
2423
"github.com/stretchr/testify/suite"
24+
"go.uber.org/zap/zaptest"
2525

2626
clusteradapter "github.com/siderolabs/talos/internal/app/machined/pkg/adapters/cluster"
2727
clusterctrl "github.com/siderolabs/talos/internal/app/machined/pkg/controllers/cluster"
2828
"github.com/siderolabs/talos/internal/app/machined/pkg/controllers/ctest"
29-
"github.com/siderolabs/talos/pkg/logging"
3029
"github.com/siderolabs/talos/pkg/machinery/config/machine"
3130
"github.com/siderolabs/talos/pkg/machinery/constants"
3231
"github.com/siderolabs/talos/pkg/machinery/proto"
@@ -113,7 +112,7 @@ func (suite *DiscoveryServiceSuite) TestReconcile() {
113112
defer cliCtxCancel()
114113

115114
go func() {
116-
errCh <- cli.Run(cliCtx, logging.Wrap(log.Writer()), notifyCh)
115+
errCh <- cli.Run(cliCtx, zaptest.NewLogger(suite.T()), notifyCh)
117116
}()
118117

119118
suite.Assert().NoError(retry.Constant(3*time.Second, retry.WithUnits(100*time.Millisecond)).Retry(
@@ -328,7 +327,7 @@ func (suite *DiscoveryServiceSuite) TestDisable() {
328327
defer cliCtxCancel()
329328

330329
go func() {
331-
errCh <- cli.Run(cliCtx, logging.Wrap(log.Writer()), notifyCh)
330+
errCh <- cli.Run(cliCtx, zaptest.NewLogger(suite.T()), notifyCh)
332331
}()
333332

334333
// inject some affiliate via our client, controller should publish it as an affiliate

internal/app/machined/pkg/controllers/cluster/endpoint.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ import (
88
"context"
99
"fmt"
1010
"net/netip"
11-
"reflect"
11+
"slices"
1212
"sort"
1313

1414
"github.com/cosi-project/runtime/pkg/controller"
@@ -82,7 +82,7 @@ func (ctrl *EndpointController) Run(ctx context.Context, r controller.Runtime, l
8282
r,
8383
k8s.NewEndpoint(k8s.ControlPlaneNamespaceName, k8s.ControlPlaneDiscoveredEndpointsID),
8484
func(r *k8s.Endpoint) error {
85-
if !reflect.DeepEqual(r.TypedSpec().Addresses, endpoints) {
85+
if !slices.Equal(r.TypedSpec().Addresses, endpoints) {
8686
logger.Debug("updated controlplane endpoints", zap.Any("endpoints", endpoints))
8787
}
8888

internal/app/machined/pkg/controllers/etcd/advertised_peer.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ import (
99
"errors"
1010
"fmt"
1111
"net/netip"
12-
"reflect"
12+
"slices"
1313
"time"
1414

1515
"github.com/cosi-project/runtime/pkg/controller"
@@ -153,7 +153,7 @@ func (ctrl *AdvertisedPeerController) updateAdvertisedPeers(ctx context.Context,
153153
})
154154
currentPeerURLs := localMember.PeerURLs
155155

156-
if reflect.DeepEqual(newPeerURLs, currentPeerURLs) {
156+
if slices.Equal(newPeerURLs, currentPeerURLs) {
157157
return nil
158158
}
159159

internal/app/machined/pkg/controllers/files/etcfile_test.go

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@ package files_test
66

77
import (
88
"context"
9-
"log"
109
"os"
1110
"path/filepath"
1211
"strconv"
@@ -22,9 +21,9 @@ import (
2221
"github.com/cosi-project/runtime/pkg/state/impl/namespaced"
2322
"github.com/siderolabs/go-retry/retry"
2423
"github.com/stretchr/testify/suite"
24+
"go.uber.org/zap/zaptest"
2525

2626
filesctrl "github.com/siderolabs/talos/internal/app/machined/pkg/controllers/files"
27-
"github.com/siderolabs/talos/pkg/logging"
2827
"github.com/siderolabs/talos/pkg/machinery/resources/files"
2928
)
3029

@@ -50,7 +49,7 @@ func (suite *EtcFileSuite) SetupTest() {
5049

5150
var err error
5251

53-
suite.runtime, err = runtime.NewRuntime(suite.state, logging.Wrap(log.Writer()))
52+
suite.runtime, err = runtime.NewRuntime(suite.state, zaptest.NewLogger(suite.T()))
5453
suite.Require().NoError(err)
5554

5655
suite.startRuntime()

internal/app/machined/pkg/controllers/hardware/hardware_test.go

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@ package hardware_test
66

77
import (
88
"context"
9-
"log"
109
"sync"
1110
"time"
1211

@@ -17,8 +16,7 @@ import (
1716
"github.com/cosi-project/runtime/pkg/state/impl/namespaced"
1817
"github.com/siderolabs/go-retry/retry"
1918
"github.com/stretchr/testify/suite"
20-
21-
"github.com/siderolabs/talos/pkg/logging"
19+
"go.uber.org/zap/zaptest"
2220
)
2321

2422
type HardwareSuite struct {
@@ -40,9 +38,7 @@ func (suite *HardwareSuite) SetupTest() {
4038

4139
var err error
4240

43-
logger := logging.Wrap(log.Writer())
44-
45-
suite.runtime, err = runtime.NewRuntime(suite.state, logger)
41+
suite.runtime, err = runtime.NewRuntime(suite.state, zaptest.NewLogger(suite.T()))
4642
suite.Require().NoError(err)
4743
}
4844

internal/app/machined/pkg/controllers/k8s/address_filter_test.go

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@ package k8s_test
88
import (
99
"context"
1010
"fmt"
11-
"log"
1211
"net/url"
1312
"sync"
1413
"testing"
@@ -21,9 +20,9 @@ import (
2120
"github.com/cosi-project/runtime/pkg/state/impl/namespaced"
2221
"github.com/siderolabs/go-retry/retry"
2322
"github.com/stretchr/testify/suite"
23+
"go.uber.org/zap/zaptest"
2424

2525
k8sctrl "github.com/siderolabs/talos/internal/app/machined/pkg/controllers/k8s"
26-
"github.com/siderolabs/talos/pkg/logging"
2726
"github.com/siderolabs/talos/pkg/machinery/config/container"
2827
"github.com/siderolabs/talos/pkg/machinery/config/types/v1alpha1"
2928
"github.com/siderolabs/talos/pkg/machinery/resources/config"
@@ -51,7 +50,7 @@ func (suite *K8sAddressFilterSuite) SetupTest() {
5150

5251
var err error
5352

54-
suite.runtime, err = runtime.NewRuntime(suite.state, logging.Wrap(log.Writer()))
53+
suite.runtime, err = runtime.NewRuntime(suite.state, zaptest.NewLogger(suite.T()))
5554
suite.Require().NoError(err)
5655

5756
suite.Require().NoError(suite.runtime.RegisterController(&k8sctrl.AddressFilterController{}))

internal/app/machined/pkg/controllers/k8s/control_plane_static_pod_test.go

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,7 @@ package k8s_test
77
import (
88
"context"
99
"fmt"
10-
"log"
11-
"reflect"
10+
"slices"
1211
"strconv"
1312
"strings"
1413
"sync"
@@ -23,13 +22,13 @@ import (
2322
"github.com/siderolabs/gen/xslices"
2423
"github.com/siderolabs/go-retry/retry"
2524
"github.com/stretchr/testify/suite"
25+
"go.uber.org/zap/zaptest"
2626
v1 "k8s.io/api/core/v1"
2727
apiresource "k8s.io/apimachinery/pkg/api/resource"
2828

2929
k8sadapter "github.com/siderolabs/talos/internal/app/machined/pkg/adapters/k8s"
3030
"github.com/siderolabs/talos/internal/app/machined/pkg/controllers/ctest"
3131
k8sctrl "github.com/siderolabs/talos/internal/app/machined/pkg/controllers/k8s"
32-
"github.com/siderolabs/talos/pkg/logging"
3332
"github.com/siderolabs/talos/pkg/machinery/constants"
3433
"github.com/siderolabs/talos/pkg/machinery/resources/k8s"
3534
"github.com/siderolabs/talos/pkg/machinery/resources/v1alpha1"
@@ -54,7 +53,7 @@ func (suite *ControlPlaneStaticPodSuite) SetupTest() {
5453

5554
var err error
5655

57-
suite.runtime, err = runtime.NewRuntime(suite.state, logging.Wrap(log.Writer()))
56+
suite.runtime, err = runtime.NewRuntime(suite.state, zaptest.NewLogger(suite.T()))
5857
suite.Require().NoError(err)
5958

6059
suite.Require().NoError(suite.runtime.RegisterController(&k8sctrl.ControlPlaneStaticPodController{}))
@@ -93,7 +92,7 @@ func (suite *ControlPlaneStaticPodSuite) assertControlPlaneStaticPods(manifests
9392

9493
ids := xslices.Map(resources.Items, func(r resource.Resource) string { return r.Metadata().ID() })
9594

96-
if !reflect.DeepEqual(manifests, ids) {
95+
if !slices.Equal(manifests, ids) {
9796
return retry.ExpectedErrorf("expected %q, got %q", manifests, ids)
9897
}
9998

internal/app/machined/pkg/controllers/k8s/endpoint.go

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@ import (
88
"context"
99
"fmt"
1010
"net/netip"
11-
"reflect"
1211
"slices"
1312
"time"
1413

@@ -213,7 +212,7 @@ func (ctrl *EndpointController) updateEndpointsResource(ctx context.Context, r c
213212
r,
214213
k8s.NewEndpoint(k8s.ControlPlaneNamespaceName, k8s.ControlPlaneAPIServerEndpointsID),
215214
func(r *k8s.Endpoint) error {
216-
if !reflect.DeepEqual(r.TypedSpec().Addresses, addrs) {
215+
if !slices.Equal(r.TypedSpec().Addresses, addrs) {
217216
logger.Debug("updated controlplane endpoints", zap.Any("endpoints", addrs))
218217
}
219218

0 commit comments

Comments
 (0)