src/chromiumos/tast/local/bundles/cros/smartlock/login.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 smartlock contains tests for the Smart Lock feature in ChromeOS.
package smartlock

import (
	"context"
	"time"

	"chromiumos/tast/local/chrome"
	"chromiumos/tast/local/chrome/ash"
	"chromiumos/tast/local/chrome/crossdevice"
	"chromiumos/tast/local/chrome/crossdevice/smartlock"
	"chromiumos/tast/local/chrome/uiauto/faillog"
	"chromiumos/tast/local/chrome/uiauto/lockscreen"
	"chromiumos/tast/local/input"
	"chromiumos/tast/testing"
)

func init() {
	testing.AddTest(&testing.Test{
		Func:         Login,
		LacrosStatus: testing.LacrosVariantUnneeded,
		Desc:         "Tests ability to sign-in with Smart lock",
		Contacts: []string{
			"dhaddock@chromium.org",
			"chromeos-sw-engprod@google.com",
			"chromeos-cross-device-eng@google.com",
		},
		Attr:         []string{"group:cross-device"},
		VarDeps:      []string{"ui.signinProfileTestExtensionManifestKey"},
		SoftwareDeps: []string{"chrome"},
		Fixture:      "crossdeviceOnboardedNoLock",
		Timeout:      3 * time.Minute,
	})
}

// Login tests logging into ChromeOS using Smart Lock feature.
func Login(ctx context.Context, s *testing.State) {
	cr := s.FixtValue().(*crossdevice.FixtData).Chrome
	tconn := s.FixtValue().(*crossdevice.FixtData).TestConn
	androidDevice := s.FixtValue().(*crossdevice.FixtData).AndroidDevice
	username := s.FixtValue().(*crossdevice.FixtData).Username
	password := s.FixtValue().(*crossdevice.FixtData).Password

	// This test does a few login/logout cycles. We want all the chrome sessions to have the same setup as the fixture.
	opts := s.FixtValue().(*crossdevice.FixtData).ChromeOptions

	// Ensure all logins do not clear existing users.
	opts = append(opts, chrome.KeepState())

	// Smart Lock requires the Android phone to have a PIN code. Set it here and defer removing it.
	if err := androidDevice.SetPIN(ctx); err != nil {
		s.Fatal("Failed to set lock screen PIN on Android: ", err)
	}
	defer androidDevice.ClearPIN(ctx)

	s.Log("Enabling Smart Lock for login")
	if err := smartlock.EnableSmartLockLogin(ctx, tconn, cr, password); err != nil {
		s.Fatal("Failed to enable smart lock login: ", err)
	}

	// Logout is done by keyboard shortcut. So setup one to reuse throughout the test.
	kb, err := input.Keyboard(ctx)
	if err != nil {
		s.Fatal("Failed to get keyboard: ", err)
	}
	defer kb.Close()

	if err := smartlock.SignOut(ctx, cr, kb); err != nil {
		s.Fatal("Failed to sign out")
	}

	s.Log("Signed out after enabling Smart Lock sign-in")
	loginOpts := append(opts, chrome.GAIALogin(chrome.Creds{User: username, Pass: password}))
	s.Log("Logging in again with password before Smart Lock will work")
	cr, err = chrome.New(
		ctx,
		loginOpts...,
	)
	if err != nil {
		s.Fatal("Failed to login to Chrome: ", err)
	}

	tconn, err = cr.TestAPIConn(ctx)
	if err != nil {
		s.Fatal("Creating test API connection failed: ", err)
	}
	if _, err := smartlock.OpenSmartLockSubpage(ctx, tconn, cr); err != nil {
		s.Fatal("Failed to open Smart lock sub page: ", err)
	}
	if err := testing.Sleep(ctx, 10*time.Second); err != nil {
		s.Fatal("Failed to sleep on Smart Lock subpage: ", err)
	}
	if err := smartlock.SignOut(ctx, cr, kb); err != nil {
		s.Fatal("Failed to sign out")
	}
	s.Log("Signed out after signing in with password")

	s.Log("Now trying to login with Smart Lock")
	nologinOpts := append(opts, chrome.NoLogin())
	nologinOpts = append(nologinOpts, chrome.LoadSigninProfileExtension(s.RequiredVar("ui.signinProfileTestExtensionManifestKey")))
	cr, err = chrome.New(
		ctx,
		nologinOpts...,
	)
	defer cr.Close(ctx)
	if err != nil {
		s.Fatal("Failed to start chrome: ", err)
	}
	tconn, err = cr.SigninProfileTestAPIConn(ctx)
	if err != nil {
		s.Fatal("Getting API connection failed: ", err)
	}
	defer faillog.DumpUITreeOnError(ctx, s.OutDir(), s.HasError, tconn)

	s.Log("Waiting for the Smart Lock ready indicator")
	if err := lockscreen.WaitForSmartLockReady(ctx, tconn); err != nil {
		s.Fatal("Failed waiting for Smart Lock icon to turn green: ", err)
	}
	s.Log("Smart Unlock available. Clicking Smart Lock arrow button to log back in")
	if err := lockscreen.ClickSmartLockArrowButton(ctx, tconn); err != nil {
		s.Fatal("Failed to click arrow button on the ChromeOS lock screen: ", err)
	}

	// Check for shelf to ensure we are logged back in.
	if err := ash.WaitForShelf(ctx, tconn, 10*time.Second); err != nil {
		s.Fatal("Shelf did not appear after logging in: ", err)
	}
}