src/chromiumos/tast/remote/bundles/cros/omaha/lts.go

// Copyright 2021 The Chromium OS 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 omaha

import (
	"context"
	"strconv"
	"strings"
	"time"

	"chromiumos/tast/common/fixture"
	"chromiumos/tast/remote/bundles/cros/omaha/params"
	"chromiumos/tast/remote/bundles/cros/omaha/request"
	"chromiumos/tast/testing"
)

func init() {
	testing.AddTest(&testing.Test{
		Func: LTS,
		Desc: "Check if an appropriate LTS version is being served",
		Contacts: []string{
			"vsavu@chromium.org", // Test author
			"chromeos-commercial-remote-management@google.com",
		},
		Attr:    []string{"group:omaha"},
		Fixture: fixture.Omaha,
	})
}

func LTS(ctx context.Context, s *testing.State) {
	state := s.FixtValue().(*params.FixtData)

	ltsChromeOsVersion := state.Config.ChromeOSVersionFromMilestone[state.Config.CurrentChromeOSLTS]
	ltsPrefix := strconv.FormatInt(int64(ltsChromeOsVersion), 10)

	req := request.New()
	req.GenSP(state.Device, state.Config.OldVersion)
	requestApp := request.GenerateRequestApp(state.Device, state.Config.OldVersion, request.Stable)
	requestApp.UpdateCheck.LTSTag = "lts"
	requestApp.UpdateCheck.TargetVersionPrefix = ltsPrefix
	req.Apps = append(req.Apps, requestApp)

	ctx, cancel := context.WithTimeout(ctx, 30*time.Second)
	defer cancel()

	res, err := request.Send(ctx, req, "lts")
	if err != nil {
		s.Fatal("Failed to send request: ", err)
	}

	if err := res.ValidateUpdateResponse(); err != nil {
		s.Fatal("Response is not an update: ", err)
	}

	chromeOSVersion, err := res.ChromeOSVersion()
	if err != nil {
		s.Error("Failed to get ChromeOS version: ", err)
	}
	if !request.MatchOneOfVersions(chromeOSVersion, ltsChromeOsVersion) {
		s.Errorf("ChromeOS Version %q does not match the requested prefix %q", chromeOSVersion, ltsPrefix)
	}

	minorVersion, err := strconv.Atoi(strings.Split(chromeOSVersion, ".")[1])
	if err != nil {
		s.Fatalf("Failed to read the minor version from %q: %v", chromeOSVersion, err)
	}

	if minorVersion < state.Config.CurrentChromeOSLTSMinor {
		s.Errorf("Minor version %d not an LTS minor version (>=%d)", minorVersion, state.Config.CurrentChromeOSLTSMinor)
	}
}