pam/client-dummy: Add support for cancelling isAuthenticated request

When waiting we may want to be able to cancel it, so support it
ubuntu · Jul 3, 2024 · 7c4e2e4 · 7c4e2e4
1 parent c912c14
commit 7c4e2e4
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/pam/internal/pam_test/pam-client-dummy.go b/pam/internal/pam_test/pam-client-dummy.go
@@ -428,7 +428,14 @@ func (dc *DummyClient) IsAuthenticated(ctx context.Context, in *authd.IARequest,
 		if dc.isAuthenticatedWantWait == 0 {
 			return nil, errors.New("no wanted wait provided")
 		}
-		time.Sleep(dc.isAuthenticatedWantWait)
+		select {
+		case <-time.After(dc.isAuthenticatedWantWait):
+		case <-ctx.Done():
+			return &authd.IAResponse{
+				Access: brokers.AuthCancelled,
+				Msg:    fmt.Sprintf(`{"message": "Cancelled: %s"}`, dc.isAuthenticatedMessage),
+			}, nil
+		}
 		return &authd.IAResponse{
 			Access: brokers.AuthGranted,
 			Msg:    msg,