mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-21 06:31:11 +02:00
0081ac5fac
This extracts Docker Talos cluster provisioner as common code which might be shared between `osctl cluster` and integration-test. There should be almost no functional changes. As proof of concept, abstract cluster readiness checks were implemented based on provisioned cluster state. It implements same checks as `basic-integration.sh` in pure Go via Talos/K8s clients. `conditions` package was promoted from machined-internal to `internal/pkg` as it is used to run the checks. Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
48 lines
1.3 KiB
Go
48 lines
1.3 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
// Package check provides set of checks to verify cluster readiness.
|
|
package check
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/talos-systems/talos/internal/pkg/provision"
|
|
)
|
|
|
|
// ServiceStateAssertion checks whether service reached some specified state.
|
|
func ServiceStateAssertion(ctx context.Context, cluster provision.ClusterAccess, service, state string) error {
|
|
client, err := cluster.Client()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
servicesInfo, err := client.ServiceInfo(ctx, service)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
serviceOk := false
|
|
|
|
for _, serviceInfo := range servicesInfo {
|
|
if len(serviceInfo.Service.Events.Events) == 0 {
|
|
return fmt.Errorf("no events recorded yet for service %q", service)
|
|
}
|
|
|
|
lastEvent := serviceInfo.Service.Events.Events[len(serviceInfo.Service.Events.Events)-1]
|
|
if lastEvent.State != state {
|
|
return fmt.Errorf("service %q not in expected state %q: current state [%s] %s", service, state, lastEvent.State, lastEvent.Msg)
|
|
}
|
|
|
|
serviceOk = true
|
|
}
|
|
|
|
if !serviceOk {
|
|
return fmt.Errorf("service %q not found", service)
|
|
}
|
|
|
|
return nil
|
|
}
|