Skip to content

Commit

Permalink
test: fix collector for machines not having an IP in status
Browse files Browse the repository at this point in the history
  • Loading branch information
chrischdi committed May 17, 2024
1 parent 3bd6c93 commit 718cb44
Show file tree
Hide file tree
Showing 5 changed files with 90 additions and 13 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -14,3 +14,10 @@ patches:
- target:
kind: VSphereMachineTemplate
path: ../commons/remove-storage-policy.yaml
# Replace ssh user to match expected user by the e2e machine collector
- target:
kind: KubeadmControlPlane
path: patch-user-kcp.yaml
- target:
kind: KubeadmConfigTemplate
path: patch-user-md.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
- op: add
path: /spec/kubeadmConfigSpec/users/0/name
value: "capv"
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
- op: add
path: /spec/template/spec/users/0/name
value: "capv"
5 changes: 4 additions & 1 deletion test/e2e/e2e_suite_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -266,7 +266,10 @@ var _ = SynchronizedBeforeSuite(func() []byte {
// vspherelog.MachineLogCollector tries to ssh to the machines to collect logs.
// This does not work when using vcsim because there are no real machines running ssh.
if testTarget != VCSimTestTarget {
clusterProxyOptions = append(clusterProxyOptions, framework.WithMachineLogCollector(vspherelog.MachineLogCollector{}))
clusterProxyOptions = append(clusterProxyOptions, framework.WithMachineLogCollector(&vspherelog.MachineLogCollector{
Client: vsphereClient,
Finder: vsphereFinder,
}))
}
bootstrapClusterProxy = framework.NewClusterProxy("bootstrap", kubeconfigPath, initScheme(), clusterProxyOptions...)

Expand Down
85 changes: 73 additions & 12 deletions test/framework/log/collector.go
Original file line number Diff line number Diff line change
Expand Up @@ -22,12 +22,18 @@ import (
"context"
"fmt"
"io"
"net"
"os"
"path/filepath"
"strings"

"github.com/pkg/errors"
"github.com/vmware/govmomi"
"github.com/vmware/govmomi/find"
"github.com/vmware/govmomi/vim25/mo"
"golang.org/x/crypto/ssh"
kerrors "k8s.io/apimachinery/pkg/util/errors"
"k8s.io/klog/v2"
clusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
expv1 "sigs.k8s.io/cluster-api/exp/api/v1beta1"
"sigs.k8s.io/controller-runtime/pkg/client"
Expand All @@ -39,20 +45,19 @@ const (
VSpherePrivateKeyFilePath = "VSPHERE_SSH_PRIVATE_KEY"
)

type MachineLogCollector struct{}
type MachineLogCollector struct {
Client *govmomi.Client
Finder *find.Finder
}

func (collector MachineLogCollector) CollectMachinePoolLog(_ context.Context, _ client.Client, _ *expv1.MachinePool, _ string) error {
func (c *MachineLogCollector) CollectMachinePoolLog(_ context.Context, _ client.Client, _ *expv1.MachinePool, _ string) error {
return nil
}

func (collector MachineLogCollector) CollectMachineLog(_ context.Context, _ client.Client, m *clusterv1.Machine, outputPath string) error {
var hostIPAddr string
for _, address := range m.Status.Addresses {
if address.Type != clusterv1.MachineExternalIP {
continue
}
hostIPAddr = address.Address
break
func (c *MachineLogCollector) CollectMachineLog(ctx context.Context, _ client.Client, m *clusterv1.Machine, outputPath string) error {
machineIPAddresses, err := c.machineIPAddresses(ctx, m)
if err != nil {
return err
}

captureLogs := func(hostFileName, command string, args ...string) func() error {
Expand All @@ -62,7 +67,20 @@ func (collector MachineLogCollector) CollectMachineLog(_ context.Context, _ clie
return err
}
defer f.Close()
return executeRemoteCommand(f, hostIPAddr, command, args...)
var errs []error
// Try with all available IPs unless it succeeded.
for _, machineIPAddress := range machineIPAddresses {
if err := executeRemoteCommand(f, machineIPAddress, command, args...); err != nil {
errs = append(errs, err)
continue
}
return nil
}

if err := kerrors.NewAggregate(errs); err != nil {
return errors.Wrapf(err, "failed to run command %s for machine %s on ips [%s]", command, klog.KObj(m), strings.Join(machineIPAddresses, ", "))
}
return nil
}
}

Expand All @@ -78,10 +96,53 @@ func (collector MachineLogCollector) CollectMachineLog(_ context.Context, _ clie
})
}

func (collector MachineLogCollector) CollectInfrastructureLogs(_ context.Context, _ client.Client, _ *clusterv1.Cluster, _ string) error {
func (c *MachineLogCollector) CollectInfrastructureLogs(_ context.Context, _ client.Client, _ *clusterv1.Cluster, _ string) error {
return nil
}

func (c *MachineLogCollector) machineIPAddresses(ctx context.Context, m *clusterv1.Machine) ([]string, error) {
for _, address := range m.Status.Addresses {
if address.Type != clusterv1.MachineExternalIP {
continue
}
return []string{address.Address}, nil
}

vmObj, err := c.Finder.VirtualMachine(ctx, m.GetName())
if err != nil {
return nil, err
}

var vm mo.VirtualMachine

if err := c.Client.RetrieveOne(ctx, vmObj.Reference(), []string{"guest.net"}, &vm); err != nil {
// We cannot get the properties e.g. when the vm already got deleted or is getting deleted.
return nil, errors.Errorf("error retrieving properties for machine %s", klog.KObj(m))
}

addresses := []string{}

// Return all IPs so we can try each of them until one succeeded.
for _, nic := range vm.Guest.Net {
if nic.IpConfig == nil {
continue
}
for _, ip := range nic.IpConfig.IpAddress {
netIP := net.ParseIP(ip.IpAddress)
ipv4 := netIP.To4()
if ipv4 != nil {
addresses = append(addresses, ip.IpAddress)
}
}
}

if len(addresses) == 0 {
return nil, errors.Errorf("unable to find IP Addresses for Machine %s", klog.KObj(m))
}

return addresses, nil
}

func createOutputFile(path string) (*os.File, error) {
if err := os.MkdirAll(filepath.Dir(path), os.ModePerm); err != nil {
return nil, err
Expand Down

0 comments on commit 718cb44

Please sign in to comment.