further change

auth0 · Nov 15, 2024 · f86749c · f86749c
1 parent b9de027
commit f86749c
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/management/self_service_profiles_test.go b/management/self_service_profiles_test.go
@@ -4,7 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"net/http"
-	"strings"
+	"net/url"
 	"testing"
 	"time"
 
@@ -144,11 +144,13 @@ func TestSelfServiceProfileManager_CreateAndRevokeTicket(t *testing.T) {
 	assert.NotEmpty(t, ticket.GetTicket())
 
 	ticketURL := ticket.GetTicket()
-	ticketURLSliced := strings.Split(ticketURL, "=")
-	ticketID := ticketURLSliced[len(ticketURLSliced)-1]
-
-	err = api.SelfServiceProfile.RevokeTicket(context.Background(), ssop.GetID(), ticketID)
+	ticketIDMap, err := url.ParseQuery(ticketURL)
+	if err != nil {
+		ticketID := ticketIDMap["ticketId"][0]
+		err = api.SelfServiceProfile.RevokeTicket(context.Background(), ssop.GetID(), ticketID)
+	}
 	assert.NoError(t, err)
+
 }
 
 func TestSelfServiceProfileManager_MarshalJSON(t *testing.T) {