barge/sync_dag_test.go at main · frantjc/barge · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
//go:build dagger

package barge_test

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

	"github.com/Masterminds/semver/v3"
	"github.com/frantjc/barge"
	_ "github.com/frantjc/barge/internal/archive"
	_ "github.com/frantjc/barge/internal/chartmuseum"
	_ "github.com/frantjc/barge/internal/directory"
	_ "github.com/frantjc/barge/internal/file"
	_ "github.com/frantjc/barge/internal/oci"
	_ "github.com/frantjc/barge/internal/repo"
	"github.com/google/uuid"
	"github.com/stretchr/testify/require"
	"sigs.k8s.io/yaml"
)

func TestSyncDistribution(t *testing.T) {
	ctx := Context(t)

	namespace := uuid.NewString()
	dag := Dag(t)
	distributionURL := Distribution(t, dag)

	// Sync from archive into OCI registry.
	archiveChart, archiveURL := Archive(t)
	archiveSyncCfg, err := os.CreateTemp(t.TempDir(), "archive.yml")
	require.NoError(t, err)
	b, err := yaml.Marshal(&barge.SyncConfig{
		Sources: []barge.SourceConfig{
			{
				URL: barge.URL(*archiveURL),
			},
			{
				URL:       barge.URL(*archiveURL),
				Namespace: namespace,
			},
		},
	})
	require.NoError(t, err)
	_, err = archiveSyncCfg.Write(b)
	require.NoError(t, err)
	require.NoError(t, archiveSyncCfg.Close())

	require.NoError(t, barge.Sync(ctx, archiveSyncCfg.Name(), distributionURL.String()))

	// Sync from OCI registry into a directory.
	constraints, err := semver.NewConstraint(archiveChart.Metadata.Version)
	require.NoError(t, err)

	ociSyncCfg, err := os.CreateTemp(t.TempDir(), "oci.yml")
	require.NoError(t, err)
	b, err = yaml.Marshal(&barge.SyncConfig{
		Sources: []barge.SourceConfig{
			{
				URL: barge.URL(*distributionURL),
				Charts: map[string]barge.Constraints{
					archiveChart.Name(): barge.Constraints(*constraints),
				},
			},
			{
				URL: barge.URL(*distributionURL.JoinPath(namespace)),
				Charts: map[string]barge.Constraints{
					archiveChart.Name(): barge.Constraints(*constraints),
				},
			},
		},
	})
	require.NoError(t, err)
	_, err = ociSyncCfg.Write(b)
	require.NoError(t, err)
	require.NoError(t, ociSyncCfg.Close())

	require.NoError(t, barge.Sync(ctx, ociSyncCfg.Name(), t.TempDir()))
}

func TestSyncChartmuseum(t *testing.T) {
	ctx := Context(t)

	_, archiveURL := Archive(t)
	archiveSyncCfg, err := os.CreateTemp(t.TempDir(), "archive.yml")
	require.NoError(t, err)
	b, err := yaml.Marshal(&barge.SyncConfig{
		Sources: []barge.SourceConfig{
			{
				URL: barge.URL(*archiveURL),
			},
		},
	})
	require.NoError(t, err)
	_, err = archiveSyncCfg.Write(b)
	require.NoError(t, err)
	require.NoError(t, archiveSyncCfg.Close())

	dag := Dag(t)
	chartmuseumURL := Chartmuseum(t, dag)

	require.NoError(t, barge.Sync(ctx, archiveSyncCfg.Name(), chartmuseumURL.String()))
}

func TestSyncErrorMissingConfig(t *testing.T) {
	ctx := Context(t)
	dag := Dag(t)
	distributionURL := Distribution(t, dag)
	require.Error(t, barge.Sync(ctx, filepath.Join(t.TempDir(), "does-not-exist.yaml"), distributionURL.String()))
}