77 lines
1.8 KiB
Go
77 lines
1.8 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package e2eutil
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/hashicorp/nomad/testutil"
|
|
)
|
|
|
|
func WaitForLastDeploymentStatus(jobID, ns, status string, wc *WaitConfig) error {
|
|
var nsArg = []string{}
|
|
if ns != "" {
|
|
nsArg = []string{"-namespace", ns}
|
|
}
|
|
|
|
var got string
|
|
var err error
|
|
interval, retries := wc.OrDefault()
|
|
testutil.WaitForResultRetries(retries, func() (bool, error) {
|
|
time.Sleep(interval)
|
|
|
|
cmd := []string{"nomad", "job", "status"}
|
|
cmd = append(cmd, nsArg...)
|
|
cmd = append(cmd, jobID)
|
|
|
|
out, err := Command(cmd[0], cmd[1:]...)
|
|
if err != nil {
|
|
return false, fmt.Errorf("could not get job status: %v\n%v", err, out)
|
|
}
|
|
|
|
section, err := GetSection(out, "Latest Deployment")
|
|
if err != nil {
|
|
return false, fmt.Errorf("could not find Latest Deployment section: %w", err)
|
|
}
|
|
|
|
fields, err := ParseFields(section)
|
|
if err != nil {
|
|
return false, fmt.Errorf("could not parse Latest Deployment section: %w", err)
|
|
}
|
|
|
|
got = fields["Status"]
|
|
return got == status, nil
|
|
}, func(e error) {
|
|
err = fmt.Errorf("deployment status check failed: got %#v", got)
|
|
})
|
|
return err
|
|
}
|
|
|
|
func LastDeploymentID(jobID, ns string) (string, error) {
|
|
|
|
var nsArg = []string{}
|
|
if ns != "" {
|
|
nsArg = []string{"-namespace", ns}
|
|
}
|
|
|
|
cmd := []string{"nomad", "deployment", "list"}
|
|
cmd = append(cmd, nsArg...)
|
|
|
|
out, err := Command(cmd[0], cmd[1:]...)
|
|
if err != nil {
|
|
return "", fmt.Errorf("could not get deployment list: %v\n%v", err, out)
|
|
}
|
|
rows, err := ParseColumns(out)
|
|
if err != nil {
|
|
return "", fmt.Errorf("could not parse deployment list output: %w", err)
|
|
}
|
|
for _, row := range rows {
|
|
if row["Job ID"] == jobID {
|
|
return row["ID"], nil
|
|
}
|
|
}
|
|
return "", fmt.Errorf("could not find a recent deployment for job")
|
|
}
|