jiri-test/internal/test/release_kube.go

// Copyright 2015 The Vanadium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package test

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"
	"time"

	"v.io/jiri"
	"v.io/jiri/collect"
	"v.io/x/devtools/internal/test"
)

var (
	releaseKubeDefaultReleaseTestTimeout = time.Minute * 5
	releaseKubeReleaseBucket             = "gs://vanadium-release"
)

func vanadiumReleaseKubeStaging(jirix *jiri.X, testName string, opts ...Opt) (_ *test.Result, e error) {
	manifestPath := os.Getenv(manifestEnvVar)
	if manifestPath == "" {
		return nil, fmt.Errorf("%s environment variable not set", manifestEnvVar)
	}
	version := cleanupVersionString(filepath.Base(manifestPath))
	return vanadiumReleaseKubeCommon(jirix, testName, "staging", version)
}

func vanadiumReleaseKubeProduction(jirix *jiri.X, testName string, opts ...Opt) (_ *test.Result, e error) {
	version := ""
	if snapshotTimestamp := os.Getenv(snapshotTimestampEnvVar); snapshotTimestamp != "" {
		version = cleanupVersionString(snapshotTimestamp)
	}
	return vanadiumReleaseKubeCommon(jirix, testName, "production", version)
}

func vanadiumReleaseKubeCommon(jirix *jiri.X, testName, updateType, version string) (_ *test.Result, e error) {
	cleanup, err := initTest(jirix, testName, []string{"v23:base"})
	if err != nil {
		return nil, newInternalError(err, "Init")
	}
	defer collect.Error(func() error { return cleanup() }, &e)

	// Build and run vprodupdater.
	s := jirix.NewSeq()
	if err := s.Last("jiri", "go", "install", "v.io/infrastructure/vprodupdater/"); err != nil {
		return nil, newInternalError(err, "Build vprodupdater")
	}
	vprodupdaterBin := filepath.Join(jirix.Root, "infrastructure", "go", "bin", "vprodupdater")
	args := []string{
		"update-all",
		fmt.Sprintf("-type=%s", updateType),
		fmt.Sprintf("-tag=%s", version),
	}
	if err := s.Capture(jirix.Stdout(), jirix.Stderr()).Last(vprodupdaterBin, args...); err != nil {
		return nil, newInternalError(err, "Run vprodupdater")
	}

	// Run "prod service test" to make sure everything is up and running.
	hostname := "dev.staging.v.io"
	if updateType == "production" {
		hostname = "dev.v.io"
	}
	jiriTestArgs := []string{
		"run",
		fmt.Sprintf("--v23.namespace.root=/ns.%s:8101", hostname),
		fmt.Sprintf("--blessings-root=%s", hostname),
		"vanadium-prod-services-test",
	}
	if err := s.
		Capture(jirix.Stdout(), jirix.Stderr()).
		Timeout(releaseKubeDefaultReleaseTestTimeout).
		Last("jiri-test", jiriTestArgs...); err != nil {
		return nil, newInternalError(err, "Prod service test")
	}

	// For staging release, write a latest file with the version as
	// its content.
	if updateType == "staging" {
		tmpDir, err := s.TempDir("", "")
		if err != nil {
			return nil, newInternalError(err, "Create tmp dir")
		}
		defer jirix.NewSeq().RemoveAll(tmpDir)
		latestFile := filepath.Join(tmpDir, "latest")
		gsutilArgs := []string{"-q", "cp", latestFile, fmt.Sprintf("%s/latest", releaseKubeReleaseBucket)}
		if err := s.
			WriteFile(latestFile, []byte(version), os.FileMode(0600)).
			Last("gsutil", gsutilArgs...); err != nil {
			return nil, newInternalError(err, "Write latest file")
		}
	}

	return &test.Result{Status: test.Passed}, nil
}

func cleanupVersionString(version string) string {
	// Remove all separators to make the version string look cleaner.
	for _, s := range []string{"-", ".", ":"} {
		version = strings.Replace(version, s, "", -1)
	}
	return "manifest-" + version
}