@@ -14,18 +14,18 @@ import (
14
14
"os"
15
15
"strings"
16
16
"sync"
17
- "time"
18
17
19
18
"github.com/gobuffalo/flect"
20
19
"github.com/pkg/errors"
21
20
bootstrapv1alpha3 "github.com/siderolabs/cluster-api-bootstrap-provider-talos/api/v1alpha3"
22
21
"github.com/siderolabs/crypto/tls"
23
22
"github.com/siderolabs/crypto/x509"
24
- "github.com/siderolabs/talos/pkg/grpc/ gen"
23
+ "github.com/siderolabs/gen/slices "
25
24
"github.com/siderolabs/talos/pkg/machinery/api/common"
26
25
"github.com/siderolabs/talos/pkg/machinery/api/machine"
27
26
"github.com/siderolabs/talos/pkg/machinery/api/storage"
28
- "github.com/siderolabs/talos/pkg/machinery/config/types/v1alpha1/generate"
27
+ "github.com/siderolabs/talos/pkg/machinery/config/generate"
28
+ "github.com/siderolabs/talos/pkg/machinery/config/generate/secrets"
29
29
"github.com/siderolabs/talos/pkg/machinery/constants"
30
30
"github.com/siderolabs/talos/pkg/machinery/proto"
31
31
"google.golang.org/grpc"
@@ -34,7 +34,6 @@ import (
34
34
"google.golang.org/grpc/status"
35
35
"google.golang.org/protobuf/types/known/anypb"
36
36
"google.golang.org/protobuf/types/known/emptypb"
37
- "gopkg.in/typ.v4/slices"
38
37
appsv1 "k8s.io/api/apps/v1"
39
38
corev1 "k8s.io/api/core/v1"
40
39
apiextensionsv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
@@ -95,19 +94,18 @@ func newReconciler(client client.Client, opts ...reconcilerOption) *controllers.
95
94
}
96
95
97
96
func newFakeClient (initObjs ... client.Object ) client.Client {
98
- return & fakeClient {
99
- startTime : time .Now (),
100
- Client : fake .NewClientBuilder ().WithObjects (initObjs ... ).WithScheme (fakeScheme ).Build (),
101
- }
102
- }
103
-
104
- type fakeClient struct {
105
- startTime time.Time
106
- mux sync.Mutex
107
- client.Client
97
+ return fake .NewClientBuilder ().
98
+ WithObjects (initObjs ... ).
99
+ WithScheme (fakeScheme ).
100
+ WithStatusSubresource (
101
+ & clusterv1.Cluster {},
102
+ & clusterv1.Machine {},
103
+ & controlplanev1.TalosControlPlane {},
104
+ ).
105
+ Build ()
108
106
}
109
107
110
- func createSecrets (ctx context.Context , obj client.Client , cluster * clusterv1.Cluster , secretsBundle * generate. SecretsBundle , machineAddress string ) error {
108
+ func createSecrets (ctx context.Context , obj client.Client , cluster * clusterv1.Cluster , secretsBundle * secrets. Bundle , machineAddress string ) error {
111
109
ca := & corev1.Secret {
112
110
TypeMeta : metav1.TypeMeta {
113
111
Kind : "Secret" ,
@@ -123,12 +121,12 @@ func createSecrets(ctx context.Context, obj client.Client, cluster *clusterv1.Cl
123
121
return err
124
122
}
125
123
126
- input , err := generate .NewInput (cluster .Name , "https://localhost:6443" , constants .DefaultKubernetesVersion , secretsBundle )
124
+ input , err := generate .NewInput (cluster .Name , "https://localhost:6443" , constants .DefaultKubernetesVersion , generate . WithSecretsBundle ( secretsBundle ) )
127
125
if err != nil {
128
126
return err
129
127
}
130
128
131
- config , err := generate .Talosconfig (input )
129
+ config , err := input .Talosconfig ()
132
130
if err != nil {
133
131
return err
134
132
}
@@ -330,7 +328,7 @@ func generateCRD(gvk schema.GroupVersionKind, properties map[string]apiextension
330
328
}
331
329
}
332
330
333
- func startMachineServer (ctx context.Context , secretsBundle * generate. SecretsBundle ) (* machineService , string , error ) {
331
+ func startMachineServer (ctx context.Context , secretsBundle * secrets. Bundle ) (* machineService , string , error ) {
334
332
listener , err := net .Listen ("tcp" , "localhost:0" )
335
333
if err != nil {
336
334
return nil , "" , err
@@ -344,7 +342,7 @@ func startMachineServer(ctx context.Context, secretsBundle *generate.SecretsBund
344
342
345
343
var generator tls.Generator
346
344
347
- generator , err = gen . NewLocalGenerator (ca .Key , ca .Crt )
345
+ generator , err = NewLocalGenerator (ca .Key , ca .Crt )
348
346
if err != nil {
349
347
return nil , "" , err
350
348
}
0 commit comments