Add context to all necessary methods (#128)

okta · Apr 28, 2020 · 758fdac · 758fdac
1 parent 2b15ccf
commit 758fdac
Show file tree

Hide file tree

Showing 37 changed files with 859 additions and 824 deletions.
diff --git a/okta/appUser.go b/okta/appUser.go
@@ -19,6 +19,7 @@
 package okta
 
 import (
+	"context"
 	"fmt"
 	"github.com/okta/okta-sdk-golang/v2/okta/query"
 	"time"
@@ -52,7 +53,7 @@ func (a *AppUser) IsApplicationInstance() bool {
 }
 
 // Updates a user&#x27;s profile for an application
-func (m *AppUserResource) UpdateApplicationUser(appId string, userId string, body AppUser) (*AppUser, *Response, error) {
+func (m *AppUserResource) UpdateApplicationUser(ctx context.Context, appId string, userId string, body AppUser) (*AppUser, *Response, error) {
 	url := fmt.Sprintf("/api/v1/apps/%v/users/%v", appId, userId)
 
 	req, err := m.client.requestExecutor.WithAccept("application/json").WithContentType("application/json").NewRequest("POST", url, body)
@@ -62,7 +63,7 @@ func (m *AppUserResource) UpdateApplicationUser(appId string, userId string, bod
 
 	var appUser *AppUser
 
-	resp, err := m.client.requestExecutor.Do(req, &appUser)
+	resp, err := m.client.requestExecutor.Do(ctx, req, &appUser)
 	if err != nil {
 		return nil, resp, err
 	}
@@ -71,7 +72,7 @@ func (m *AppUserResource) UpdateApplicationUser(appId string, userId string, bod
 }
 
 // Removes an assignment for a user from an application.
-func (m *AppUserResource) DeleteApplicationUser(appId string, userId string, qp *query.Params) (*Response, error) {
+func (m *AppUserResource) DeleteApplicationUser(ctx context.Context, appId string, userId string, qp *query.Params) (*Response, error) {
 	url := fmt.Sprintf("/api/v1/apps/%v/users/%v", appId, userId)
 	if qp != nil {
 		url = url + qp.String()
@@ -82,7 +83,7 @@ func (m *AppUserResource) DeleteApplicationUser(appId string, userId string, qp
 		return nil, err
 	}
 
-	resp, err := m.client.requestExecutor.Do(req, nil)
+	resp, err := m.client.requestExecutor.Do(ctx, req, nil)
 	if err != nil {
 		return resp, err
 	}

diff --git a/okta/application.go b/okta/application.go
diff --git a/okta/applicationGroupAssignment.go b/okta/applicationGroupAssignment.go
@@ -19,6 +19,7 @@
 package okta
 
 import (
+	"context"
 	"fmt"
 	"time"
 )
@@ -43,15 +44,15 @@ func (a *ApplicationGroupAssignment) IsApplicationInstance() bool {
 }
 
 // Removes a group assignment from an application.
-func (m *ApplicationGroupAssignmentResource) DeleteApplicationGroupAssignment(appId string, groupId string) (*Response, error) {
+func (m *ApplicationGroupAssignmentResource) DeleteApplicationGroupAssignment(ctx context.Context, appId string, groupId string) (*Response, error) {
 	url := fmt.Sprintf("/api/v1/apps/%v/groups/%v", appId, groupId)
 
 	req, err := m.client.requestExecutor.WithAccept("application/json").WithContentType("application/json").NewRequest("DELETE", url, nil)
 	if err != nil {
 		return nil, err
 	}
 
-	resp, err := m.client.requestExecutor.Do(req, nil)
+	resp, err := m.client.requestExecutor.Do(ctx, req, nil)
 	if err != nil {
 		return resp, err
 	}