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

import (
	"context"
	"fmt"
	"time"

	"chromiumos/tast/local/chrome"
	"chromiumos/tast/local/chrome/uiauto"
	"chromiumos/tast/local/chrome/uiauto/faillog"
	"chromiumos/tast/local/chrome/uiauto/nodewith"
	"chromiumos/tast/local/chrome/uiauto/role"
	"chromiumos/tast/local/wallpaper"
	"chromiumos/tast/testing"
)

func init() {
	testing.AddTest(&testing.Test{
		Func: SetOnlineWallpaper,
		Desc: "Test setting online wallpapers in the new wallpaper app",
		Contacts: []string{
			"jasontt@google.com",
			"chromeos-sw-engprod@google.com",
			"assistive-eng@google.com",
		},
		Attr:         []string{"group:mainline", "informational"},
		SoftwareDeps: []string{"chrome"},
	})
}

func SetOnlineWallpaper(ctx context.Context, s *testing.State) {
	const (
		firstCollection  = "Solid colors"
		firstImage       = "Light Blue"
		secondCollection = "Colors"
		secondImage      = "Bubbly"
	)

	cr, err := chrome.New(ctx, chrome.EnableFeatures("WallpaperWebUI"))
	if err != nil {
		s.Fatal("Failed to connect to Chrome: ", err)
	}
	defer cr.Close(ctx)

	tconn, err := cr.TestAPIConn(ctx)
	if err != nil {
		s.Fatal("Failed to create Test API connection: ", err)
	}
	defer faillog.DumpUITreeOnError(ctx, s.OutDir(), s.HasError, tconn)

	// The test has a dependency of network speed, so we give uiauto.Context ample
	// time to wait for nodes to load.
	ui := uiauto.New(tconn).WithTimeout(30 * time.Second)

	if err := wallpaper.OpenWallpaperPicker(ui)(ctx); err != nil {
		s.Fatal("Failed to open wallpaper picker: ", err)
	}
	if err := wallpaper.SelectCollection(ui, firstCollection)(ctx); err != nil {
		s.Fatalf("Failed to select collection %q: %v", firstCollection, err)
	}
	if err := wallpaper.SelectImage(ui, firstImage)(ctx); err != nil {
		s.Fatalf("Failed to select image %q: %v", firstImage, err)
	}
	if err := ui.WaitUntilExists(nodewith.Name(fmt.Sprintf("Currently set %v", firstImage)).Role(role.Heading))(ctx); err != nil {
		s.Fatalf("Failed to validate selected wallpaper %q: %v", firstImage, err)
	}

	// Navigate back to collection view by clicking on the back arrow in breadcrumb.
	if err := ui.LeftClick(nodewith.Name("Back to Wallpaper").HasClass("icon-arrow-back").Role(role.Button))(ctx); err != nil {
		s.Fatal("Failed to navigate back to collection view: ", err)
	}
	if err := wallpaper.SelectCollection(ui, secondCollection)(ctx); err != nil {
		s.Fatalf("Failed to select collection %q: %v", secondCollection, err)
	}
	if err := wallpaper.SelectImage(ui, secondImage)(ctx); err != nil {
		s.Fatalf("Failed to select image %q: %v", secondImage, err)
	}
	if err := ui.WaitUntilExists(nodewith.Name(fmt.Sprintf("Currently set %v", secondImage)).Role(role.Heading))(ctx); err != nil {
		s.Fatalf("Failed to validate selected wallpaper %q: %v", secondImage, err)
	}
}