src/chromiumos/tast/local/chrome/familylink/familylink.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 familylink is used for writing Family Link tests.
package familylink

import (
	"context"
	"time"

	"chromiumos/tast/errors"
	"chromiumos/tast/local/chrome"
	"chromiumos/tast/local/chrome/uiauto"
	"chromiumos/tast/local/chrome/uiauto/nodewith"
	"chromiumos/tast/local/chrome/uiauto/ossettings"
	"chromiumos/tast/local/chrome/uiauto/role"
	"chromiumos/tast/local/input"
	"chromiumos/tast/testing"
)

// AddEduSecondaryAccount opens the Edu Coexistence in-session flow
// and attempts to add a secondary account for a Family Link (FL)
// primary user. FL users can only have EDU secondary accounts. Trying
// to add other account types will fail.
// Precondition: The current logged in user must be FL (such as Geller
// or Unicorn).
func AddEduSecondaryAccount(ctx context.Context, cr *chrome.Chrome, tconn *chrome.TestConn,
	parentFirstName, parentLastName, parentUser, parentPass,
	secondUser, secondPass string) error {

	ui := uiauto.New(tconn)

	testing.ContextLog(ctx, "Checking logged in user is Family Link")
	if err := ui.Exists(nodewith.Name("This account is managed by Family Link").Role(role.Image))(ctx); err != nil {
		return errors.Wrap(err, "logged in user is not Family Link")
	}

	testing.ContextLog(ctx, "Launching the settings app")
	googleAccountsButton := nodewith.Name("Google Accounts").Role(role.Button)
	if _, err := ossettings.LaunchAtPageURL(ctx, tconn, cr, "people", ui.Exists(googleAccountsButton)); err != nil {
		return errors.Wrap(err, "failed to launch people settings page")
	}

	testing.ContextLog(ctx, "Opening the in-session EDU Coexistence flow")
	addSchoolAccountButton := nodewith.Name("Add school account").Role(role.Button)
	selectParentOption := nodewith.NameStartingWith(parentFirstName + " " + parentLastName).Role(role.ListBoxOption)
	if err := uiauto.Combine("open in-session edu coexistence flow",
		ui.WaitUntilExists(googleAccountsButton),
		ui.LeftClickUntil(googleAccountsButton, ui.Exists(addSchoolAccountButton)),
		ui.WithInterval(time.Second).LeftClickUntil(addSchoolAccountButton, ui.Exists(selectParentOption)),
	)(ctx); err != nil {
		return errors.Wrap(err, "failed to open in-session edu coexistence flow")
	}

	testing.ContextLog(ctx, "Clicking button that matches parent email: ", parentUser)
	if err := ui.LeftClickUntil(selectParentOption, ui.Exists(nodewith.Name("Parent password").Role(role.TextField)))(ctx); err != nil {
		return errors.Wrap(err, "failed to click button that matches parent email")
	}

	if err := NavigateEduCoexistenceFlow(ctx, tconn, parentPass, secondUser, secondPass); err != nil {
		return errors.Wrap(err, "failed to navigate in-session Edu Coexistence flow")
	}

	return nil
}

// NavigateEduCoexistenceFlow goes through the Edu Coexistence
// in-session flow and attempts to add a secondary account for a
// Family Link (FL) primary user. FL users can only have EDU secondary
// accounts. Trying to add other account types will fail.
// Precondition: The current logged in user must be FL (such as Geller
// or Unicorn).
func NavigateEduCoexistenceFlow(ctx context.Context, tconn *chrome.TestConn, parentPass,
	secondUser, secondPass string) error {
	ui := uiauto.New(tconn)

	testing.ContextLog(ctx, "Checking logged in user is Family Link")
	if err := ui.Exists(nodewith.Name("This account is managed by Family Link").Role(role.Image))(ctx); err != nil {
		return errors.Wrap(err, "logged in user is not Family Link")
	}

	testing.ContextLog(ctx, "Clicking the parent password text field")
	if err := ui.LeftClick(nodewith.Name("Parent password").Role(role.TextField))(ctx); err != nil {
		return errors.Wrap(err, "failed to click parent password text")
	}

	testing.ContextLog(ctx, "Setting up keyboard")
	kb, err := input.Keyboard(ctx)
	if err != nil {
		return errors.Wrap(err, "failed to get keyboard")
	}
	defer kb.Close()

	testing.ContextLog(ctx, "Typing the parent password")
	if err := kb.Type(ctx, parentPass+"\n"); err != nil {
		return errors.Wrap(err, "failed to type parent password")
	}

	testing.ContextLog(ctx, "Clicking next on school account information for parents and Google workspace for education information pages")
	nextButton := nodewith.Name("Next").Role(role.Button)
	enterSchoolEmailText := nodewith.Name("School email").Role(role.TextField)
	if err := uiauto.Combine("Clicking next",
		ui.WaitUntilExists(nextButton),
		ui.LeftClickUntil(nextButton, ui.Exists(enterSchoolEmailText)))(ctx); err != nil {
		return errors.Wrap(err, "failed to click Next button")
	}

	testing.ContextLog(ctx, "Clicking school email text field")
	if err := ui.LeftClick(enterSchoolEmailText)(ctx); err != nil {
		return errors.Wrap(err, "failed to click school email text field")
	}

	testing.ContextLog(ctx, "Typing school account email")
	if err := kb.Type(ctx, secondUser+"\n"); err != nil {
		return errors.Wrap(err, "failed to type Geller parent email")
	}

	testing.ContextLog(ctx, "Clicking school account password text field")
	schoolPasswordText := nodewith.Name("School password").Role(role.TextField)
	if err := uiauto.Combine("Clicking school account password text field",
		ui.WaitUntilExists(schoolPasswordText), ui.LeftClick(schoolPasswordText))(ctx); err != nil {
		return errors.Wrap(err, "failed to click school account password text field")
	}

	testing.ContextLog(ctx, "Typing school account password")
	if err := kb.Type(ctx, secondPass+"\n"); err != nil {
		return errors.Wrap(err, "failed to type edu user password")
	}

	return nil
}