Skip to content

Commit 220ee92

Browse files
author
Mikalai Radchuk
committed
fixup! Turn installed package variable source into a func
1 parent deb7a6f commit 220ee92

File tree

1 file changed

+42
-24
lines changed

1 file changed

+42
-24
lines changed

internal/resolution/variablesources/installed_package_test.go

+42-24
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ package variablesources_test
22

33
import (
44
"encoding/json"
5-
"fmt"
65
"testing"
76

87
"github.com/operator-framework/deppy/pkg/deppy"
@@ -202,37 +201,36 @@ func TestMakeInstalledPackageVariables(t *testing.T) {
202201
},
203202
}
204203

205-
fakeBundleDeployments := func(bundleImages ...string) []rukpakv1alpha1.BundleDeployment {
206-
bundleDeployments := []rukpakv1alpha1.BundleDeployment{}
207-
for idx, bundleImage := range bundleImages {
208-
bd := rukpakv1alpha1.BundleDeployment{
209-
ObjectMeta: metav1.ObjectMeta{
210-
Name: fmt.Sprintf("bd-%d", idx),
211-
},
212-
Spec: rukpakv1alpha1.BundleDeploymentSpec{
213-
Template: &rukpakv1alpha1.BundleTemplate{
214-
Spec: rukpakv1alpha1.BundleSpec{
215-
Source: rukpakv1alpha1.BundleSource{
216-
Image: &rukpakv1alpha1.ImageSource{
217-
Ref: bundleImage,
218-
},
204+
fakeBundleDeployment := func(name, bundleImage string) rukpakv1alpha1.BundleDeployment {
205+
return rukpakv1alpha1.BundleDeployment{
206+
ObjectMeta: metav1.ObjectMeta{
207+
Name: name,
208+
},
209+
Spec: rukpakv1alpha1.BundleDeploymentSpec{
210+
Template: &rukpakv1alpha1.BundleTemplate{
211+
Spec: rukpakv1alpha1.BundleSpec{
212+
Source: rukpakv1alpha1.BundleSource{
213+
Image: &rukpakv1alpha1.ImageSource{
214+
Ref: bundleImage,
219215
},
220216
},
221217
},
222218
},
223-
}
224-
bundleDeployments = append(bundleDeployments, bd)
219+
},
225220
}
226-
227-
return bundleDeployments
228221
}
229222

230223
t.Run("with ForceSemverUpgradeConstraints feature gate enabled", func(t *testing.T) {
231224
defer featuregatetesting.SetFeatureGateDuringTest(t, features.OperatorControllerFeatureGate, features.ForceSemverUpgradeConstraints, true)()
232225

233226
t.Run("with non-zero major version", func(t *testing.T) {
234227
const bundleImage = "registry.io/repo/[email protected]"
235-
installedPackages, err := variablesources.MakeInstalledPackageVariables(allBundles, fakeBundleDeployments(bundleImage))
228+
installedPackages, err := variablesources.MakeInstalledPackageVariables(
229+
allBundles,
230+
[]rukpakv1alpha1.BundleDeployment{
231+
fakeBundleDeployment("test-bd", bundleImage),
232+
},
233+
)
236234
require.NoError(t, err)
237235

238236
require.Len(t, installedPackages, 1)
@@ -250,7 +248,12 @@ func TestMakeInstalledPackageVariables(t *testing.T) {
250248
t.Run("with zero major version", func(t *testing.T) {
251249
t.Run("with zero minor version", func(t *testing.T) {
252250
const bundleImage = "registry.io/repo/[email protected]"
253-
installedPackages, err := variablesources.MakeInstalledPackageVariables(allBundles, fakeBundleDeployments(bundleImage))
251+
installedPackages, err := variablesources.MakeInstalledPackageVariables(
252+
allBundles,
253+
[]rukpakv1alpha1.BundleDeployment{
254+
fakeBundleDeployment("test-bd", bundleImage),
255+
},
256+
)
254257
require.NoError(t, err)
255258

256259
require.Len(t, installedPackages, 1)
@@ -265,7 +268,12 @@ func TestMakeInstalledPackageVariables(t *testing.T) {
265268

266269
t.Run("with non-zero minor version", func(t *testing.T) {
267270
const bundleImage = "registry.io/repo/[email protected]"
268-
installedPackages, err := variablesources.MakeInstalledPackageVariables(allBundles, fakeBundleDeployments(bundleImage))
271+
installedPackages, err := variablesources.MakeInstalledPackageVariables(
272+
allBundles,
273+
[]rukpakv1alpha1.BundleDeployment{
274+
fakeBundleDeployment("test-bd", bundleImage),
275+
},
276+
)
269277
require.NoError(t, err)
270278

271279
require.Len(t, installedPackages, 1)
@@ -285,7 +293,12 @@ func TestMakeInstalledPackageVariables(t *testing.T) {
285293
defer featuregatetesting.SetFeatureGateDuringTest(t, features.OperatorControllerFeatureGate, features.ForceSemverUpgradeConstraints, false)()
286294

287295
const bundleImage = "registry.io/repo/[email protected]"
288-
installedPackages, err := variablesources.MakeInstalledPackageVariables(allBundles, fakeBundleDeployments(bundleImage))
296+
installedPackages, err := variablesources.MakeInstalledPackageVariables(
297+
allBundles,
298+
[]rukpakv1alpha1.BundleDeployment{
299+
fakeBundleDeployment("test-bd", bundleImage),
300+
},
301+
)
289302
require.NoError(t, err)
290303

291304
require.Len(t, installedPackages, 1)
@@ -301,7 +314,12 @@ func TestMakeInstalledPackageVariables(t *testing.T) {
301314

302315
t.Run("installed bundle not found", func(t *testing.T) {
303316
const bundleImage = "registry.io/repo/[email protected]"
304-
installedPackages, err := variablesources.MakeInstalledPackageVariables(allBundles, fakeBundleDeployments(bundleImage))
317+
installedPackages, err := variablesources.MakeInstalledPackageVariables(
318+
allBundles,
319+
[]rukpakv1alpha1.BundleDeployment{
320+
fakeBundleDeployment("test-bd", bundleImage),
321+
},
322+
)
305323
assert.Nil(t, installedPackages)
306324
assert.ErrorContains(t, err, `bundleImage "registry.io/repo/[email protected]" not found`)
307325
})

0 commit comments

Comments
 (0)