src/chromiumos/tast/local/bundles/cros/filemanager/drivefs_search.go

// Copyright 2020 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 filemanager

import (
	"context"
	"io/ioutil"
	"path/filepath"
	"time"

	"chromiumos/tast/errors"
	"chromiumos/tast/local/bundles/cros/filemanager/pre"
	"chromiumos/tast/local/chrome/ui"
	"chromiumos/tast/local/chrome/ui/filesapp"
	"chromiumos/tast/local/chrome/uiauto/faillog"
	"chromiumos/tast/local/drivefs"
	"chromiumos/tast/testing"
)

func init() {
	testing.AddTest(&testing.Test{
		Func: DrivefsSearch,
		Desc: "Verify that exact file search for Google Drive returns correct value",
		Contacts: []string{
			"dats@chromium.org",
			"austinct@chromium.org",
			"benreich@chromium.org",
			"chromeos-files-syd@google.com",
		},
		SoftwareDeps: []string{
			"chrome",
			"chrome_internal",
			"drivefs",
		},
		Attr: []string{
			"group:mainline",
			"informational",
		},
		Pre: pre.DriveFsStarted,
		VarDeps: []string{
			"filemanager.user",
			"filemanager.password",
			"filemanager.drive_credentials",
		},
	})
}

func DrivefsSearch(ctx context.Context, s *testing.State) {
	mountPath := s.PreValue().(drivefs.PreData).MountPath
	tconn := s.PreValue().(drivefs.PreData).TestAPIConn

	// This test case is exercising the full-text search of DriveFS, keeping the name
	// fairly unique to avoid having it match as content search (not just file name).
	drivefsRoot := filepath.Join(mountPath, "root")
	const fileName = "verify-full-text-search-functionality-drivefs"
	if err := ioutil.WriteFile(filepath.Join(drivefsRoot, fileName), []byte("fake-content"), 0644); err != nil {
		s.Fatalf("Could not create the test file inside %q: %v", drivefsRoot, err)
	}
	// Don't delete the test file after the test as there may not be enough time
	// after the test for the deletion to be synced to Drive.

	defer faillog.DumpUITreeOnError(ctx, s.OutDir(), s.HasError, tconn)

	// Launch Files App
	filesApp, err := filesapp.Launch(ctx, tconn)
	if err != nil {
		s.Fatal("Could not launch the Files App: ", err)
	}
	defer filesApp.Release(ctx)

	// Navigate to Google Drive via the Files App ui.
	if err := filesApp.OpenDrive(ctx); err != nil {
		s.Fatal("Could not open Google Drive folder: ", err)
	}

	// Searches files app for the search term passed in.
	// Do not have to wait for the file to sync as search happens on remote, not locally.
	if err := filesApp.Search(ctx, fileName); err != nil {
		s.Fatalf("Failed search for %q: %v", fileName, err)
	}

	// Get all the containers that contain the file names.
	params := ui.FindParams{
		ClassName: "filename-label",
		Role:      ui.RoleTypeGenericContainer,
	}

	// Poll for for 1 result to appear in the search results or 15 seconds elapses.
	var resultsTextContainer ui.NodeSlice
	defer func() {
		resultsTextContainer.Release(ctx)
	}()
	if err := testing.Poll(ctx, func(ctx context.Context) error {
		rtc, err := filesApp.Root.Descendants(ctx, params)
		if err != nil {
			return testing.PollBreak(err)
		}
		defer func() {
			rtc.Release(ctx)
		}()

		// We expect 1 file to be returned given the filename length of the seed file.
		if len(rtc) != 1 {
			return errors.Wrapf(err, "unexpected number of search results: got %d, want 1", len(rtc))
		}

		resultsTextContainer = rtc
		rtc = nil
		return nil
	}, &testing.PollOptions{Timeout: 15 * time.Second, Interval: 2 * time.Second}); err != nil {
		s.Fatal("Failed waiting for search results: ", err)
	}

	// Get the actual text from the container.
	text, err := resultsTextContainer[0].Descendant(ctx, ui.FindParams{Role: ui.RoleTypeStaticText})
	if err != nil {
		s.Fatal("Failed finding descendant text box: ", err)
	}
	defer text.Release(ctx)

	if text.Name != fileName {
		s.Fatalf("Failed as search result is incorrect: got %q, want %q", text.Name, fileName)
	}
}