@@ -75,6 +75,7 @@ import (
7575 snotation "github.com/fluxcd/source-controller/internal/oci/notation"
7676 sreconcile "github.com/fluxcd/source-controller/internal/reconcile"
7777 "github.com/fluxcd/source-controller/internal/reconcile/summarize"
78+ "github.com/fluxcd/source-controller/internal/storage"
7879)
7980
8081func TestHelmChartReconciler_deleteBeforeFinalizer (t * testing.T ) {
@@ -330,7 +331,7 @@ func TestHelmChartReconciler_Reconcile(t *testing.T) {
330331func TestHelmChartReconciler_reconcileStorage (t * testing.T ) {
331332 tests := []struct {
332333 name string
333- beforeFunc func (obj * sourcev1.HelmChart , storage * Storage ) error
334+ beforeFunc func (obj * sourcev1.HelmChart , storage * storage. Storage ) error
334335 want sreconcile.Result
335336 wantErr bool
336337 assertArtifact * sourcev1.Artifact
@@ -339,7 +340,7 @@ func TestHelmChartReconciler_reconcileStorage(t *testing.T) {
339340 }{
340341 {
341342 name : "garbage collects" ,
342- beforeFunc : func (obj * sourcev1.HelmChart , storage * Storage ) error {
343+ beforeFunc : func (obj * sourcev1.HelmChart , storage * storage. Storage ) error {
343344 revisions := []string {"a" , "b" , "c" , "d" }
344345 for n := range revisions {
345346 v := revisions [n ]
@@ -389,7 +390,7 @@ func TestHelmChartReconciler_reconcileStorage(t *testing.T) {
389390 },
390391 {
391392 name : "notices missing artifact in storage" ,
392- beforeFunc : func (obj * sourcev1.HelmChart , storage * Storage ) error {
393+ beforeFunc : func (obj * sourcev1.HelmChart , storage * storage. Storage ) error {
393394 obj .Status .Artifact = & sourcev1.Artifact {
394395 Path : "/reconcile-storage/invalid.txt" ,
395396 Revision : "d" ,
@@ -408,7 +409,7 @@ func TestHelmChartReconciler_reconcileStorage(t *testing.T) {
408409 },
409410 {
410411 name : "notices empty artifact digest" ,
411- beforeFunc : func (obj * sourcev1.HelmChart , storage * Storage ) error {
412+ beforeFunc : func (obj * sourcev1.HelmChart , storage * storage. Storage ) error {
412413 f := "empty-digest.txt"
413414
414415 obj .Status .Artifact = & sourcev1.Artifact {
@@ -439,7 +440,7 @@ func TestHelmChartReconciler_reconcileStorage(t *testing.T) {
439440 },
440441 {
441442 name : "notices artifact digest mismatch" ,
442- beforeFunc : func (obj * sourcev1.HelmChart , storage * Storage ) error {
443+ beforeFunc : func (obj * sourcev1.HelmChart , storage * storage. Storage ) error {
443444 f := "digest-mismatch.txt"
444445
445446 obj .Status .Artifact = & sourcev1.Artifact {
@@ -470,7 +471,7 @@ func TestHelmChartReconciler_reconcileStorage(t *testing.T) {
470471 },
471472 {
472473 name : "updates hostname on diff from current" ,
473- beforeFunc : func (obj * sourcev1.HelmChart , storage * Storage ) error {
474+ beforeFunc : func (obj * sourcev1.HelmChart , storage * storage. Storage ) error {
474475 obj .Status .Artifact = & sourcev1.Artifact {
475476 Path : "/reconcile-storage/hostname.txt" ,
476477 Revision : "f" ,
@@ -568,7 +569,7 @@ func TestHelmChartReconciler_reconcileSource(t *testing.T) {
568569
569570 tmpDir := t .TempDir ()
570571
571- storage , err := NewStorage (tmpDir , "example.com" , retentionTTL , retentionRecords )
572+ storage , err := storage . New (tmpDir , "example.com" , retentionTTL , retentionRecords )
572573 g .Expect (err ).ToNot (HaveOccurred ())
573574
574575 gitArtifact := & sourcev1.Artifact {
@@ -1185,7 +1186,7 @@ func TestHelmChartReconciler_buildFromOCIHelmRepository(t *testing.T) {
11851186 metadata , err := loadTestChartToOCI (chartData , testRegistryServer , "" , "" , "" )
11861187 g .Expect (err ).NotTo (HaveOccurred ())
11871188
1188- storage , err := NewStorage (tmpDir , "example.com" , retentionTTL , retentionRecords )
1189+ storage , err := storage . New (tmpDir , "example.com" , retentionTTL , retentionRecords )
11891190 g .Expect (err ).ToNot (HaveOccurred ())
11901191
11911192 cachedArtifact := & sourcev1.Artifact {
@@ -1408,7 +1409,7 @@ func TestHelmChartReconciler_buildFromTarballArtifact(t *testing.T) {
14081409
14091410 tmpDir := t .TempDir ()
14101411
1411- storage , err := NewStorage (tmpDir , "example.com" , retentionTTL , retentionRecords )
1412+ storage , err := storage . New (tmpDir , "example.com" , retentionTTL , retentionRecords )
14121413 g .Expect (err ).ToNot (HaveOccurred ())
14131414
14141415 chartsArtifact := & sourcev1.Artifact {
@@ -2884,7 +2885,7 @@ func TestHelmChartReconciler_reconcileSourceFromOCI_verifySignatureNotation(t *t
28842885 metadata , err := loadTestChartToOCI (chartData , server , "" , "" , "" )
28852886 g .Expect (err ).NotTo (HaveOccurred ())
28862887
2887- storage , err := NewStorage (tmpDir , server .registryHost , retentionTTL , retentionRecords )
2888+ storage , err := storage . New (tmpDir , server .registryHost , retentionTTL , retentionRecords )
28882889 g .Expect (err ).ToNot (HaveOccurred ())
28892890
28902891 cachedArtifact := & sourcev1.Artifact {
@@ -3208,7 +3209,7 @@ func TestHelmChartReconciler_reconcileSourceFromOCI_verifySignatureCosign(t *tes
32083209 metadata , err := loadTestChartToOCI (chartData , server , "" , "" , "" )
32093210 g .Expect (err ).NotTo (HaveOccurred ())
32103211
3211- storage , err := NewStorage (tmpDir , server .registryHost , retentionTTL , retentionRecords )
3212+ storage , err := storage . New (tmpDir , server .registryHost , retentionTTL , retentionRecords )
32123213 g .Expect (err ).ToNot (HaveOccurred ())
32133214
32143215 cachedArtifact := & sourcev1.Artifact {
0 commit comments