src/chromiumos/tast/local/bundles/cros/ui/tab_switch_cuj2.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 ui

import (
	"context"
	"time"

	"chromiumos/tast/ctxutil"
	"chromiumos/tast/local/bundles/cros/ui/cuj"
	"chromiumos/tast/local/bundles/cros/ui/tabswitchcuj"
	"chromiumos/tast/local/chrome"
	"chromiumos/tast/local/chrome/ash"
	"chromiumos/tast/local/wpr"
	"chromiumos/tast/testing"
)

type tabSwitchParam struct {
	level    tabswitchcuj.Level
	wprProxy bool
}

func init() {
	testing.AddTest(&testing.Test{
		Func:         TabSwitchCUJ2,
		Desc:         "Measures the performance of tab-switching CUJ, scrolling content with trackpad",
		Contacts:     []string{"abergman@google.com", "tclaiborne@chromium.org", "xliu@cienet.com", "alfredyu@cienet.com"},
		SoftwareDeps: []string{"chrome"},
		Vars: []string{
			"ui.cuj_mute",
			"ui.cuj_mode", // Expecting "tablet" or "clamshell".
			// WPR addresses are only required when running with WPR Proxy.
			"ui.wpr_http_addr",
			"ui.wpr_https_addr",
		},
		Params: []testing.Param{
			{
				Name:    "basic",
				Timeout: 30 * time.Minute,
				Val:     tabSwitchParam{level: tabswitchcuj.Basic, wprProxy: true},
				Pre:     wpr.RemoteReplayMode(),
			}, {
				Name:    "plus",
				Timeout: 35 * time.Minute,
				Val:     tabSwitchParam{level: tabswitchcuj.Plus, wprProxy: true},
				Pre:     wpr.RemoteReplayMode(),
			}, {
				Name:    "premium",
				Timeout: 40 * time.Minute,
				Val:     tabSwitchParam{level: tabswitchcuj.Premium, wprProxy: true},
				Pre:     wpr.RemoteReplayMode(),
			}, {
				Name:              "basic_noproxy",
				Timeout:           35 * time.Minute,
				Val:               tabSwitchParam{level: tabswitchcuj.Basic, wprProxy: false},
				Fixture:           "loggedInAndKeepState",
				ExtraSoftwareDeps: []string{"arc"},
			}, {
				Name:              "plus_noproxy",
				Timeout:           40 * time.Minute,
				Val:               tabSwitchParam{level: tabswitchcuj.Plus, wprProxy: false},
				Fixture:           "loggedInAndKeepState",
				ExtraSoftwareDeps: []string{"arc"},
			}, {
				Name:              "premium_noproxy",
				Timeout:           45 * time.Minute,
				Val:               tabSwitchParam{level: tabswitchcuj.Premium, wprProxy: false},
				Fixture:           "loggedInAndKeepState",
				ExtraSoftwareDeps: []string{"arc"},
			},
		},
	})
}

// TabSwitchCUJ2 measures the performance of tab-switching CUJ.
//
// WPR server should be running in a remote server. TabSwitchCUJRecorder2 case can be used to record
// WPR content for this test in the remote server.
func TabSwitchCUJ2(ctx context.Context, s *testing.State) {
	p := s.Param().(tabSwitchParam)

	var cr *chrome.Chrome
	if !p.wprProxy {
		cr = s.FixtValue().(cuj.FixtureData).Chrome
	} else {
		cr = s.PreValue().(*chrome.Chrome)
	}

	tconn, err := cr.TestAPIConn(ctx)
	if err != nil {
		s.Fatal("Failed to connect to test API: ", err)
	}

	var tabletMode bool
	if mode, ok := s.Var("ui.cuj_mode"); ok {
		tabletMode = mode == "tablet"
		cleanup, err := ash.EnsureTabletModeEnabled(ctx, tconn, tabletMode)
		if err != nil {
			s.Fatalf("Failed to enable tablet mode to %v: %v", tabletMode, err)
		}
		defer cleanup(ctx)
	} else {
		// Use default screen mode of the DUT.
		tabletMode, err = ash.TabletModeEnabled(ctx, tconn)
		if err != nil {
			s.Fatal("Failed to get DUT default screen mode: ", err)
		}
	}
	s.Log("Running test with tablet mode: ", tabletMode)

	// Shorten context a bit to allow for cleanup if Run fails.
	ctx, cancel := ctxutil.Shorten(ctx, 3*time.Second)
	defer cancel()
	tabswitchcuj.Run2(ctx, s, cr, p.level, tabletMode)
}