xp-testing/upgrade/nop_upgrade_test.go at main · AndresNico/xp-testing · 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
//go:build upgrade

package upgrade

import (
	"testing"
	"time"

	"github.com/crossplane-contrib/xp-testing/pkg/upgrade"
	"sigs.k8s.io/e2e-framework/pkg/features"
)

// TestUpgradProviderFeature demonstrates usage of the UpgradeFeatureBuilder
func TestUpgradeProviderFeature(t *testing.T) {
	upgradeTest := upgrade.UpgradeTest{
		ClusterName:         kindClusterName,
		ProviderName:        "provider-nop",
		FromProviderPackage: fromPackage,
		ToProviderPackage:   toPackage,
		ResourceDirectories: []string{
			"../e2e/crs/Nop",
		},
	}

	// use the upgrade test feature
	upgradeFeature := upgradeTest.UpgradeFeatureBuilder("upgrade provider-nop from 0.2.0 to 0.4.0", time.Minute*2)
	testenv.Test(t, upgradeFeature.Feature())
}

// TestUpgradProviderCustom demonstrates how to build a custom upgrade feature
func TestUpgradeProviderCustom(t *testing.T) {
	upgradeTest := upgrade.UpgradeTest{
		ClusterName:         kindClusterName,
		ProviderName:        "provider-nop",
		FromProviderPackage: fromPackage,
		ToProviderPackage:   toPackage,
		ResourceDirectories: []string{
			"../e2e/crs/Nop",
		},
	}

	// build the UpgradeFeatureBuilder scenario with a custom builder and e.g. different timeouts per assessment step
	// this enables you to implement additional setup, assessment and teardown steps like more sophisticated pre/post upgrade conditions, etc.
	// or to completely re-orchestrate the upgrade steps
	customFeatureUpgrade := features.New("custom build upgrade").
		Setup(upgrade.ApplyProvider(kindClusterName, upgradeTest.FromProviderInstallOptions())).
		Setup(upgrade.ImportResources(upgradeTest.ResourceDirectories)).
		Assess("verify resources before upgrade", upgrade.VerifyResources(upgradeTest.ResourceDirectories, time.Minute*4)).
		Assess("upgrade provider", upgrade.UpgradeProvider(upgrade.UpgradeProviderOptions{
			ClusterName:         kindClusterName,
			ProviderOptions:     upgradeTest.ToProviderInstallOptions(),
			ResourceDirectories: upgradeTest.ResourceDirectories,
			WaitForPause:        time.Minute * 1,
		})).
		Assess("verify resources after upgrade", upgrade.VerifyResources(upgradeTest.ResourceDirectories, time.Minute*3)).
		Teardown(upgrade.DeleteResources(upgradeTest.ResourceDirectories, time.Minute*2)).
		Teardown(upgrade.DeleteProvider(upgradeTest.ProviderName))

	testenv.Test(t, customFeatureUpgrade.Feature())
}