go-infra/cmd/releasego/update-azure-linux_test.go

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

package main

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

	"github.com/microsoft/go-infra/buildmodel/buildassets"
	"github.com/microsoft/go-infra/goldentest"
	"github.com/microsoft/go-infra/goversion"
)

var assetsJsonPath = filepath.Join("testdata", "update-azure-linux", "assets.json")

func TestUpdateSpecFileContent(t *testing.T) {
	assets, err := loadBuildAssets(assetsJsonPath)
	if err != nil {
		t.Fatal(err)
	}

	specFilepPath := filepath.Join("testdata", "update-azure-linux", "golang.spec")
	specFile, err := os.ReadFile(specFilepPath)
	if err != nil {
		t.Fatalf("Error reading spec file from path %s, error is:%s", specFilepPath, err)
	}

	specFileContent := string(specFile)
	extractedGoFileVersion, err := extractGoArchiveNameFromSpecFile(specFileContent)
	if err != nil {
		t.Fatalf("Error extracting go archive name from spec file : %s", err)
	}

	if extractedGoFileVersion != "go1.22.4-20240604.2.src.tar.gz" {
		t.Fatalf("Expected extracted Go file version is not same as actual filename. Expected %s, returned %s", extractedGoFileVersion, "go1.22.4-20240604.2.src.tar.gz")
	}

	changelogTime, err := time.Parse("2006-01-02", "2024-08-12")
	if err != nil {
		t.Fatalf("Error parsing changelog time : %s", err)
	}

	updatedSpecFile, err := updateSpecFile(assets, changelogTime, specFileContent)
	if err != nil {
		t.Fatalf("Error updating Go revision in spec file : %s", err)
	}

	goldentest.Check(
		t, "TestUpdateSpecFileContent ",
		filepath.Join("testdata", "update-azure-linux", "updated_golang.golden.spec"),
		updatedSpecFile)
}

func TestUpdateSignaturesFileContent(t *testing.T) {
	assets, err := loadBuildAssets(assetsJsonPath)
	if err != nil {
		t.Fatal(err)
	}

	signaturesFilePath := filepath.Join("testdata", "update-azure-linux", "signatures.json")
	signaturesFile, err := os.ReadFile(signaturesFilePath)
	if err != nil {
		t.Fatalf("Error reading spec file from path %s, error is:%s", signaturesFilePath, err)
	}

	updatedSignatureFile, err := updateSignatureFile(signaturesFile, "go1.22.4-20240604.2.src.tar.gz", path.Base(assets.GoSrcURL), assets.GoSrcSHA256)
	if err != nil {
		t.Errorf("Error updating CG Manifest file : %s", err)
	}

	goldentest.Check(
		t, "TestUpdateSignaturesFileContent",
		filepath.Join("testdata", "update-azure-linux", "updated_signatures.golden.json"),
		string(updatedSignatureFile))
}

func TestUpdateCGManifestFileContent(t *testing.T) {
	assets, err := loadBuildAssets(assetsJsonPath)
	if err != nil {
		t.Fatal(err)
	}

	cgManifestFilePath := filepath.Join("testdata", "update-azure-linux", "cgmanifest.json")
	cgManifestFile, err := os.ReadFile(cgManifestFilePath)
	if err != nil {
		t.Fatalf("Error reading spec file from path %s, error is:%s", cgManifestFilePath, err)
	}

	updatedCgManifestFile, err := updateCGManifest(assets, cgManifestFile)
	if err != nil {
		t.Errorf("Error updating CG Manifest file : %s", err)
	}

	goldentest.Check(
		t, "TestUpdateCGManifestFileContent ",
		filepath.Join("testdata", "update-azure-linux", "updated_cgmanifest.golden.json"),
		string(updatedCgManifestFile))
}

func TestUpdateSpecVersion(t *testing.T) {
	type args struct {
		newGoVersion string
		oldVersion   string
		oldRelease   int
	}
	tests := []struct {
		name        string
		args        args
		wantVersion string
		wantRelease string
	}{
		{
			"patch",
			args{"1.22.4-1", "1.22.3", 1},
			"1.22.4", "1",
		},
		{
			"patch-modified-package",
			args{"1.22.4-1", "1.22.3", 2},
			"1.22.4", "1",
		},
		{
			"patch-msft-go-release",
			args{"1.22.3-2", "1.22.3", 1},
			"1.22.3", "2",
		},
		{
			"patch-msft-go-release-modified",
			args{"1.22.3-2", "1.22.3", 4},
			"1.22.3", "5",
		},
		{
			"go-major",
			args{"1.23.0-1", "1.22.8", 1},
			"1.23.0", "1",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			assets := &buildassets.BuildAssets{
				Version: tt.args.newGoVersion,
			}
			gotVersion, gotRelease := updateSpecVersion(
				assets,
				goversion.New(tt.args.oldVersion),
				tt.args.oldRelease,
			)
			if gotVersion != tt.wantVersion {
				t.Errorf("updateSpecVersion() gotVersion = %v, want %v", gotVersion, tt.wantVersion)
			}
			if gotRelease != tt.wantRelease {
				t.Errorf("updateSpecVersion() gotRelease = %v, want %v", gotRelease, tt.wantRelease)
			}
		})
	}
}

func TestPRBody(t *testing.T) {
	assets, err := loadBuildAssets(assetsJsonPath)
	if err != nil {
		t.Fatal(err)
	}

	type args struct {
		latestMajor bool
		security    bool
		notify      string
		prNumber    int
	}
	tests := []struct {
		name string
		args args
	}{
		{"essential", args{true, false, "a-go-developer", 0}},
		{"non-latest", args{false, false, "a-go-developer", 0}},
		{"security", args{true, true, "a-go-developer", 0}},
		{"no-dev", args{true, false, "", 0}},
		{"pr-number", args{true, false, "a-go-developer", 1234}},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := generatePRTitleFromAssets(assets, tt.args.security)
			got += "\n\n---\n\n"
			got += GeneratePRDescription(assets, tt.args.latestMajor, tt.args.security, tt.args.notify, tt.args.prNumber)
			goldentest.Check(
				t, "go test ./cmd/releasego -run "+t.Name(),
				filepath.Join("testdata", "update-azure-linux", "pr", "pr-description-"+tt.name+".golden.md"),
				got)
		})
	}
}