Skip to content

Commit

Permalink
refactor: creator
Browse files Browse the repository at this point in the history
Signed-off-by: Philip Laine <philip.laine@gmail.com>
  • Loading branch information
phillebaba committed Oct 7, 2024
1 parent 3f2ed2c commit a86edd5
Show file tree
Hide file tree
Showing 30 changed files with 1,971 additions and 40 deletions.
467 changes: 467 additions & 0 deletions src/internal/packager2/layout/create.go

Large diffs are not rendered by default.

109 changes: 109 additions & 0 deletions src/internal/packager2/layout/create_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,109 @@
// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: 2021-Present The Zarf Authors

package layout

import (
"os"
"path/filepath"
"testing"

"github.com/defenseunicorns/pkg/helpers/v2"
"github.com/stretchr/testify/require"

"github.com/zarf-dev/zarf/src/pkg/layout"
"github.com/zarf-dev/zarf/src/pkg/lint"
"github.com/zarf-dev/zarf/src/test/testutil"
)

func TestCreateSkeleton(t *testing.T) {
t.Parallel()

ctx := testutil.TestContext(t)

lint.ZarfSchema = testutil.LoadSchema(t, "../../../../zarf.schema.json")

opt := CreateOptions{}
path, err := CreateSkeleton(ctx, "./testdata/zarf-package", opt)
require.NoError(t, err)

pkgPath := layout.New(path)
_, warnings, err := pkgPath.ReadZarfYAML()
require.NoError(t, err)
require.Empty(t, warnings)
b, err := os.ReadFile(filepath.Join(pkgPath.Base, "checksums.txt"))
require.NoError(t, err)
expectedChecksum := `54f657b43323e1ebecb0758835b8d01a0113b61b7bab0f4a8156f031128d00f9 components/data-injections.tar
879bfe82d20f7bdcd60f9e876043cc4343af4177a6ee8b2660c304a5b6c70be7 components/files.tar
c497f1a56559ea0a9664160b32e4b377df630454ded6a3787924130c02f341a6 components/manifests.tar
fb7ebee94a4479bacddd71195030a483b0b0b96d4f73f7fcd2c2c8e0fce0c5c6 components/helm-charts.tar
`
require.Equal(t, expectedChecksum, string(b))
}

func TestGetChecksum(t *testing.T) {
t.Parallel()

tmpDir := t.TempDir()
files := map[string]string{
"empty.txt": "",
"foo": "bar",
"zarf.yaml": "Zarf Yaml Data",
"checksums.txt": "Old Checksum Data",
"nested/directory/file.md": "nested",
}
for k, v := range files {
err := os.MkdirAll(filepath.Join(tmpDir, filepath.Dir(k)), 0o700)
require.NoError(t, err)
err = os.WriteFile(filepath.Join(tmpDir, k), []byte(v), 0o600)
require.NoError(t, err)
}

checksumContent, checksumHash, err := getChecksum(tmpDir)
require.NoError(t, err)

expectedContent := `233562de1a0288b139c4fa40b7d189f806e906eeb048517aeb67f34ac0e2faf1 nested/directory/file.md
e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855 empty.txt
fcde2b2edba56bf408601fb721fe9b5c338d10ee429ea04fae5511b68fbf8fb9 foo
`
require.Equal(t, expectedContent, checksumContent)
require.Equal(t, "7c554cf67e1c2b50a1b728299c368cd56d53588300c37479623f29a52812ca3f", checksumHash)
}

func TestSignPackage(t *testing.T) {
t.Parallel()

tmpDir := t.TempDir()
yamlPath := filepath.Join(tmpDir, "zarf.yaml")
signedPath := filepath.Join(tmpDir, "zarf.yaml.sig")

err := os.WriteFile(yamlPath, []byte("foobar"), 0o644)
require.NoError(t, err)

err = signPackage(tmpDir, "", "")
require.NoError(t, err)
require.NoFileExists(t, signedPath)

err = signPackage(tmpDir, "./testdata/cosign.key", "wrongpassword")
require.EqualError(t, err, "reading key: decrypt: encrypted: decryption failed")

err = signPackage(tmpDir, "./testdata/cosign.key", "test")
require.NoError(t, err)
require.FileExists(t, signedPath)
}

func TestCreateReproducibleTarballFromDir(t *testing.T) {
t.Parallel()

tmpDir := t.TempDir()
err := os.WriteFile(filepath.Join(tmpDir, "test.txt"), []byte("hello world"), 0o600)
require.NoError(t, err)
tarPath := filepath.Join(t.TempDir(), "data.tar")

err = createReproducibleTarballFromDir(tmpDir, "", tarPath)
require.NoError(t, err)

shaSum, err := helpers.GetSHA256OfFile(tarPath)
require.NoError(t, err)
require.Equal(t, "c09d17f612f241cdf549e5fb97c9e063a8ad18ae7a9f3af066332ed6b38556ad", shaSum)
}
Loading

0 comments on commit a86edd5

Please sign in to comment.