plugin-barman-cloud/test/e2e/internal/deployment/deployment.go
Francesco Canovai 74bc9e221e
test(e2e): environment setup (#43)
Create the CI and testing infrastructure for e2e testing. Running the ci
task now will push the plugin and sidecar images to a local registry,
start kind, install the CloudNativePG and cert-manager operators, and
then install the plugin-barman-cloud one.

No actual test is implemented.

Signed-off-by: Francesco Canovai <francesco.canovai@enterprisedb.com>
2024-11-14 09:44:18 +01:00

70 lines
1.9 KiB
Go

/*
Copyright 2024.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package deployment
import (
"context"
"fmt"
"time"
appsv1 "k8s.io/api/apps/v1"
"k8s.io/apimachinery/pkg/types"
"k8s.io/apimachinery/pkg/util/wait"
"sigs.k8s.io/controller-runtime/pkg/client"
)
// IsReady checks if the deployment is ready.
func IsReady(ctx context.Context, cl client.Client, name types.NamespacedName) (bool, error) {
deployment := &appsv1.Deployment{}
err := cl.Get(ctx, name, deployment)
if err != nil {
return false, fmt.Errorf("failed to get %s deployment: %w", name, err)
}
// Check if the deployment is ready
ready := false
for _, condition := range deployment.Status.Conditions {
if condition.Type == appsv1.DeploymentAvailable && condition.Status == "True" {
ready = true
break
}
}
if !ready {
return false, nil
}
return true, nil
}
// WaitForDeploymentReady waits for the deployment to be ready. ctx should have a timeout set.
func WaitForDeploymentReady(
ctx context.Context, cl client.Client, namespacedName types.NamespacedName, interval time.Duration,
) error {
return wait.PollUntilContextCancel(ctx, interval, false,
func(ctx context.Context) (bool, error) {
ready, err := IsReady(ctx, cl, namespacedName)
if err != nil {
return false, fmt.Errorf("failed to check if %s is ready: %w", namespacedName, err)
}
if ready {
return true, nil
}
return false, nil
})
}