| // 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 typec |
| |
| import ( |
| "context" |
| "strconv" |
| "strings" |
| "time" |
| |
| "chromiumos/tast/common/testexec" |
| "chromiumos/tast/errors" |
| "chromiumos/tast/local/bundles/cros/typec/typecutils" |
| "chromiumos/tast/local/chrome" |
| "chromiumos/tast/testing" |
| "chromiumos/tast/testing/hwdep" |
| ) |
| |
| func init() { |
| testing.AddTest(&testing.Test{ |
| Func: ModeCrash, |
| Desc: "Checks USB Type C mode switch behaviour when typecd crashes", |
| Contacts: []string{"pmalani@chromium.org", "chromeos-power@google.com"}, |
| Attr: []string{"group:typec"}, |
| SoftwareDeps: []string{"chrome"}, |
| VarDeps: []string{"ui.signinProfileTestExtensionManifestKey"}, |
| Data: []string{"testcert.p12"}, |
| Params: []testing.Param{ |
| // For running manually. |
| { |
| Name: "manual", |
| ExtraAttr: []string{"typec_lab"}, |
| Val: false, |
| }, |
| // For automated testing. |
| { |
| Name: "smoke", |
| ExtraAttr: []string{"group:mainline", "informational"}, |
| ExtraHardwareDeps: hwdep.D(hwdep.Model("volteer", "voxel")), |
| Val: true, |
| }}, |
| }) |
| } |
| |
| // ModeCrash does the following: |
| // - Login. |
| // - Validate that TBT alt mode is working correctly. |
| // - Kill typecd. |
| // - Logout. |
| // - Validate that the dock in in USB+Dp alt mode. |
| // |
| // This test requires the following H/W topology to run. |
| // |
| // |
| // DUT ------> Thunderbolt3 (>= Titan Ridge) dock -----> DP monitor. |
| // (USB4) |
| // |
| func ModeCrash(ctx context.Context, s *testing.State) { |
| // This check is for test executions which take place on |
| // CQ (where TBT peripherals aren't connected). |
| port, err := typecutils.CheckPortsForTBTPartner(ctx) |
| if err != nil { |
| s.Fatal("Couldn't determine TBT device from PD identity: ", err) |
| } |
| |
| // Return early for smoke testing (CQ). |
| if smoke := s.Param().(bool); smoke { |
| return |
| } |
| |
| if port == -1 { |
| s.Fatal("No TBT device connected to DUT") |
| } |
| |
| // Get to the Chrome login screen. |
| loadSignInProfileOption := chrome.LoadSigninProfileExtension(s.RequiredVar("ui.signinProfileTestExtensionManifestKey")) |
| cr, err := chrome.New(ctx, chrome.DeferLogin(), loadSignInProfileOption) |
| if err != nil { |
| s.Fatal("Failed to start Chrome at login screen: ", err) |
| } |
| |
| testConn, err := cr.SigninProfileTestAPIConn(ctx) |
| if err != nil { |
| s.Fatal("Failed to get Test API connection: ", err) |
| } |
| defer testConn.Close() |
| |
| if err := typecutils.EnablePeripheralDataAccess(ctx, s.DataPath("testcert.p12")); err != nil { |
| s.Fatal("Failed to enable peripheral data access setting: ", err) |
| } |
| |
| if err := cr.ContinueLogin(ctx); err != nil { |
| s.Fatal("Failed to login: ", err) |
| } |
| |
| s.Log("Verifying that TBT device & DP monitor enumerated correctly") |
| if err := typecutils.CheckTBTAndDP(ctx, testConn); err != nil { |
| s.Fatal("Failed to verify TBT & DP after login: ", err) |
| } |
| |
| oldPID, err := typecdPID(ctx) |
| if err != nil { |
| s.Fatal("Failed to get original typecd PID: ", err) |
| } |
| |
| s.Log("Killing typecd") |
| if err := testexec.CommandContext(ctx, "pkill", "typecd").Run(); err != nil { |
| s.Fatal("Failed to kill typecd: ", err) |
| } |
| |
| s.Log("Checking that typecd restarted") |
| if err := testing.Poll(ctx, func(ctx context.Context) error { |
| newPID, err := typecdPID(ctx) |
| if err != nil { |
| return errors.Wrap(err, "failed to get new typecd PID") |
| } |
| |
| if newPID == oldPID { |
| return errors.Errorf("typecd still has old PID: %d", newPID) |
| } |
| |
| return nil |
| }, &testing.PollOptions{Timeout: 5 * time.Second}); err != nil { |
| s.Fatal("Failed to verify typecd restarted: ", err) |
| } |
| |
| // Log out; |
| if err := cr.Close(ctx); err != nil { |
| s.Fatal("Failed to close Chrome: ", err) |
| } |
| |
| cr, err = chrome.New(ctx, |
| chrome.DeferLogin(), |
| chrome.KeepState(), |
| loadSignInProfileOption) |
| if err != nil { |
| s.Fatal("Failed to start Chrome with login: ", err) |
| } |
| defer cr.Close(ctx) |
| |
| testConn, err = cr.SigninProfileTestAPIConn(ctx) |
| if err != nil { |
| s.Fatal("Failed to get Test API connection: ", err) |
| } |
| |
| s.Log("Verifying that no TBT devices enumerated") |
| if err := testing.Poll(ctx, func(ctx context.Context) error { |
| return typecutils.CheckTBTDevice(false) |
| }, &testing.PollOptions{Timeout: 10 * time.Second}); err != nil { |
| s.Fatal("Failed to verify no TBT devices connected at login screen: ", err) |
| } |
| |
| s.Log("Verifying that DP monitor enumerated correctly") |
| if err := testing.Poll(ctx, func(ctx context.Context) error { |
| return typecutils.FindConnectedDPMonitor(ctx, testConn) |
| }, &testing.PollOptions{Timeout: 20 * time.Second}); err != nil { |
| s.Fatal("Failed to verify DP monitor working at login screen: ", err) |
| } |
| } |
| |
| // typecdPID returns the `typecd` process ID. |
| func typecdPID(ctx context.Context) (int, error) { |
| out, err := testexec.CommandContext(ctx, "pgrep", "typecd").Output(testexec.DumpLogOnError) |
| if err != nil { |
| return 0, errors.Wrap(err, "failed to run pgrep to check whether typecd is running") |
| } |
| |
| return strconv.Atoi(strings.TrimSpace(string(out))) |
| } |