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

import (
	"context"

	"chromiumos/tast/local/wilco"
	"chromiumos/tast/testing"
	dtcpb "chromiumos/wilco_dtc"
)

func init() {
	testing.AddTest(&testing.Test{
		Func:         APIPerformWebRequestEnrolled,
		LacrosStatus: testing.LacrosVariantUnknown,
		Desc:         "Test sending PerformWebRequest to the Wilco DTC Support Daemon",
		Contacts: []string{
			"vsavu@google.com",  // Test author
			"lamzin@google.com", // wilco_dtc_supportd maintainer
			"chromeos-wilco@google.com",
			"bisakhmondal00@gmail.com",
		},
		Attr:         []string{"group:mainline", "informational"},
		SoftwareDeps: []string{"reboot", "vm_host", "wilco", "chrome"},
		Fixture:      "wilcoDTCAllowed",
	})
}

// APIPerformWebRequestEnrolled tests PerformWebRequest gRPC API.
func APIPerformWebRequestEnrolled(ctx context.Context, s *testing.State) {
	for _, tc := range []struct {
		name       string
		url        string
		httpMethod dtcpb.PerformWebRequestParameter_HttpMethod
		wantStatus dtcpb.PerformWebRequestResponse_Status
		// Don't fail the test as the website may not be reachable from DUT. Instead, just put a log.
		logOnly bool
	}{
		{
			name:       "google",
			url:        "https://google.com",
			httpMethod: dtcpb.PerformWebRequestParameter_HTTP_METHOD_GET,
			wantStatus: dtcpb.PerformWebRequestResponse_STATUS_OK,
			logOnly:    true,
		},
		{
			name:       "localhost",
			url:        "https://localhost/test",
			httpMethod: dtcpb.PerformWebRequestParameter_HTTP_METHOD_GET,
			// Requests to localhost are blocked.
			wantStatus: dtcpb.PerformWebRequestResponse_STATUS_NETWORK_ERROR,
		},
	} {
		s.Run(ctx, tc.name, func(ctx context.Context, s *testing.State) {
			response := dtcpb.PerformWebRequestResponse{}

			if err := wilco.DPSLSendMessage(ctx, "PerformWebRequest", &dtcpb.PerformWebRequestParameter{
				HttpMethod: tc.httpMethod,
				Url:        tc.url,
			}, &response); err != nil {
				s.Fatal("Failed to call PerformWebRequest API method: ", err)
			}

			if response.Status != tc.wantStatus {
				if tc.logOnly {
					s.Logf("Request for %s failed with invalid status = got %v, expect %v", tc.url, response.Status, tc.wantStatus)
					return
				}

				s.Errorf("Request for %s failed with invalid status = got %v, expect %v", tc.url, response.Status, tc.wantStatus)
			}
		})
	}
}