|
| 1 | +// SPDX-License-Identifier: Apache-2.0 |
| 2 | +// SPDX-FileCopyrightText: 2021-Present The Zarf Authors |
| 3 | + |
| 4 | +// Package test provides e2e tests for Zarf. |
| 5 | +package test |
| 6 | + |
| 7 | +import ( |
| 8 | + "encoding/base64" |
| 9 | + "encoding/json" |
| 10 | + "fmt" |
| 11 | + "path/filepath" |
| 12 | + "sync" |
| 13 | + "testing" |
| 14 | + "time" |
| 15 | + |
| 16 | + "github.com/stretchr/testify/require" |
| 17 | + "github.com/zarf-dev/zarf/src/types" |
| 18 | +) |
| 19 | + |
| 20 | +func TestComponentStatus(t *testing.T) { |
| 21 | + t.Log("E2E: Component Status") |
| 22 | + tmpDir := t.TempDir() |
| 23 | + _, _, err := e2e.Zarf(t, "package", "create", "src/test/packages/37-component-status", "-o", tmpDir, "--confirm") |
| 24 | + require.NoError(t, err) |
| 25 | + packageName := fmt.Sprintf("zarf-package-component-status-%s.tar.zst", e2e.Arch) |
| 26 | + path := filepath.Join(tmpDir, packageName) |
| 27 | + // Stop channel getting the zarf state |
| 28 | + stop := make(chan bool) |
| 29 | + // Error channel to return any errors from the goroutine. Testify doesn't like require in a goroutine |
| 30 | + errCh := make(chan error, 1) |
| 31 | + var wg sync.WaitGroup |
| 32 | + wg.Add(1) |
| 33 | + // Goroutine to wait for the package to show "deploying" status |
| 34 | + go func() { |
| 35 | + defer wg.Done() |
| 36 | + // Give extra time to build and push the package |
| 37 | + ticker := time.NewTicker(30 * time.Second) |
| 38 | + for { |
| 39 | + select { |
| 40 | + case <-ticker.C: |
| 41 | + t.Log("Timed out waiting for package to deploy") |
| 42 | + errCh <- fmt.Errorf("timed out waiting for package to deploy") |
| 43 | + return |
| 44 | + case <-stop: |
| 45 | + return |
| 46 | + default: |
| 47 | + stdOut, _, err := e2e.Kubectl(t, "get", "secret", "zarf-package-component-status", "-n", "zarf", "-o", "jsonpath={.data.data}") |
| 48 | + if err != nil { |
| 49 | + // Wait for the secret to be created and try again |
| 50 | + time.Sleep(2 * time.Second) |
| 51 | + continue |
| 52 | + } |
| 53 | + deployedPackage, err := getDeployedPackage(stdOut) |
| 54 | + if err != nil { |
| 55 | + errCh <- err |
| 56 | + return |
| 57 | + } |
| 58 | + if len(deployedPackage.DeployedComponents) != 1 { |
| 59 | + errCh <- fmt.Errorf("expected 1 component got %d", len(deployedPackage.DeployedComponents)) |
| 60 | + return |
| 61 | + } |
| 62 | + status := deployedPackage.DeployedComponents[0].Status |
| 63 | + if status != types.ComponentStatusDeploying { |
| 64 | + errCh <- fmt.Errorf("expected %s got %s", types.ComponentStatusDeploying, status) |
| 65 | + return |
| 66 | + } |
| 67 | + time.Sleep(2 * time.Second) |
| 68 | + return |
| 69 | + } |
| 70 | + } |
| 71 | + }() |
| 72 | + stdOut, stdErr, err := e2e.Zarf(t, "package", "deploy", path, "--confirm") |
| 73 | + require.NoError(t, err, stdOut, stdErr) |
| 74 | + close(stop) |
| 75 | + wg.Wait() |
| 76 | + select { |
| 77 | + case err := <-errCh: |
| 78 | + require.NoError(t, err) |
| 79 | + default: |
| 80 | + } |
| 81 | + // Verify that the component status is "succeeded" |
| 82 | + stdOut, stdErr, err = e2e.Kubectl(t, "get", "secret", "zarf-package-component-status", "-n", "zarf", "-o", "jsonpath={.data.data}") |
| 83 | + require.NoError(t, err, stdOut, stdErr) |
| 84 | + deployedPackage, err := getDeployedPackage(stdOut) |
| 85 | + require.NoError(t, err) |
| 86 | + require.Len(t, deployedPackage.DeployedComponents, 1) |
| 87 | + require.Equal(t, types.ComponentStatusSucceeded, deployedPackage.DeployedComponents[0].Status) |
| 88 | + // Remove the package |
| 89 | + t.Cleanup(func() { |
| 90 | + stdOut, stdErr, err = e2e.Zarf(t, "package", "remove", "component-status", "--confirm") |
| 91 | + require.NoError(t, err, stdOut, stdErr) |
| 92 | + }) |
| 93 | +} |
| 94 | + |
| 95 | +func getDeployedPackage(secret string) (*types.DeployedPackage, error) { |
| 96 | + deployedPackage := &types.DeployedPackage{} |
| 97 | + decoded, err := base64.StdEncoding.DecodeString(secret) |
| 98 | + if err != nil { |
| 99 | + return nil, err |
| 100 | + } |
| 101 | + err = json.Unmarshal(decoded, &deployedPackage) |
| 102 | + if err != nil { |
| 103 | + return nil, err |
| 104 | + } |
| 105 | + return deployedPackage, nil |
| 106 | +} |
0 commit comments