6b137d0ac5
In case the job fails completely, resources might leak. This patch together with the one introducing predictable names for CI artifacts will make sure we wipe all resources after each CI run.
65 lines
2 KiB
Go
65 lines
2 KiB
Go
// +build integration
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/Azure/go-autorest/autorest/azure/auth"
|
|
"github.com/osbuild/osbuild-composer/internal/boot/azuretest"
|
|
)
|
|
|
|
func panicErr(err error) {
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func printErr(err error) {
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}
|
|
|
|
// GenerateCIArtifactName generates a new identifier for CI artifacts which is based
|
|
// on environment variables specified by Jenkins
|
|
// note: in case of migration to sth else like Github Actions, change it to whatever variables GH Action provides
|
|
func GenerateCIArtifactName(prefix string) (string, error) {
|
|
distroCode := os.Getenv("DISTRO_CODE")
|
|
changeId := os.Getenv("CHANGE_ID")
|
|
buildId := os.Getenv("BUILD_ID")
|
|
if changeId == "" || buildId == "" || distroCode == "" {
|
|
return "", fmt.Errorf("The environment variables must specify CHANGE_ID, BUILD_ID, and DISTRO_CODE")
|
|
}
|
|
|
|
return fmt.Sprintf("%s%s-%s-%s", prefix, distroCode, changeId, buildId), nil
|
|
}
|
|
|
|
func main() {
|
|
fmt.Println("Running a cloud cleanup")
|
|
|
|
// Load Azure credentials
|
|
creds, err := azuretest.GetAzureCredentialsFromEnv()
|
|
panicErr(err)
|
|
if creds == nil {
|
|
panic("empty credentials")
|
|
}
|
|
// Get test ID
|
|
testID, err := GenerateCIArtifactName("")
|
|
panicErr(err)
|
|
// Delete the vhd image
|
|
imageName := "image-" + testID + ".vhd"
|
|
fmt.Println("Running delete image from Azure, this should fail if the test succedded")
|
|
err = azuretest.DeleteImageFromAzure(creds, imageName)
|
|
printErr(err)
|
|
|
|
// Delete all remaining resources (see the full list in the CleanUpBootedVM function)
|
|
fmt.Println("Running clean up booted VM, this should fail if the test succedded")
|
|
parameters := azuretest.NewDeploymentParameters(creds, imageName, testID, "")
|
|
clientCredentialsConfig := auth.NewClientCredentialsConfig(creds.ClientID, creds.ClientSecret, creds.TenantID)
|
|
authorizer, err := clientCredentialsConfig.Authorizer()
|
|
panicErr(err)
|
|
err = azuretest.CleanUpBootedVM(creds, parameters, authorizer, testID)
|
|
printErr(err)
|
|
}
|