Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
104 changes: 104 additions & 0 deletions pkg/authorization/authorization_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,104 @@
package authorization

import (
"context"
"net/http"
"net/http/httptest"
"testing"

"github.com/stretchr/testify/require"
"github.com/workos/workos-go/v6/pkg/retryablehttp"
)

func TestSetAPIKey(t *testing.T) {
DefaultClient = &Client{}
SetAPIKey("test-key-123")
require.Equal(t, "test-key-123", DefaultClient.APIKey)
}

func TestAuthorizationGetResourceByExternalId(t *testing.T) {
response := AuthorizationResource{
Object: "authorization_resource",
Id: "rsrc_01H945H0YD4F97JN3MNHBFPG37",
ExternalId: "my-document-1",
Name: "My Document",
Description: "A test document",
ResourceTypeSlug: "document",
OrganizationId: "org_01H945H0YD4F97JN3MNHBFPG37",
CreatedAt: "2024-01-01T00:00:00.000Z",
UpdatedAt: "2024-01-01T00:00:00.000Z",
}

server := httptest.NewServer(http.HandlerFunc(jsonResponseHandler(nil, nil, response)))
defer server.Close()

DefaultClient = &Client{
HTTPClient: &retryablehttp.HttpClient{Client: *server.Client()},
Endpoint: server.URL,
}
SetAPIKey("test")

resource, err := GetResourceByExternalId(context.Background(), GetResourceByExternalIdOpts{
OrganizationId: "org_01H945H0YD4F97JN3MNHBFPG37",
ResourceTypeSlug: "document",
ExternalId: "my-document-1",
})
require.NoError(t, err)
require.Equal(t, "rsrc_01H945H0YD4F97JN3MNHBFPG37", resource.Id)
require.Equal(t, "my-document-1", resource.ExternalId)
}

func TestAuthorizationUpdateResourceByExternalId(t *testing.T) {
var capturedBody map[string]interface{}

response := AuthorizationResource{
Object: "authorization_resource",
Id: "rsrc_01H945H0YD4F97JN3MNHBFPG37",
ExternalId: "my-document-1",
Name: "Updated Document",
Description: "Updated description",
ResourceTypeSlug: "document",
OrganizationId: "org_01H945H0YD4F97JN3MNHBFPG37",
CreatedAt: "2024-01-01T00:00:00.000Z",
UpdatedAt: "2024-01-02T00:00:00.000Z",
}

server := httptest.NewServer(http.HandlerFunc(jsonResponseHandler(&capturedBody, nil, response)))
defer server.Close()

DefaultClient = &Client{
HTTPClient: &retryablehttp.HttpClient{Client: *server.Client()},
Endpoint: server.URL,
}
SetAPIKey("test")

newName := "Updated Document"
newDescription := "Updated description"
resource, err := UpdateResourceByExternalId(context.Background(), UpdateResourceByExternalIdOpts{
OrganizationId: "org_01H945H0YD4F97JN3MNHBFPG37",
ResourceTypeSlug: "document",
ExternalId: "my-document-1",
Name: &newName,
Description: &newDescription,
})
require.NoError(t, err)
require.Equal(t, "Updated Document", resource.Name)
}

func TestAuthorizationDeleteResourceByExternalId(t *testing.T) {
server := httptest.NewServer(http.HandlerFunc(noContentHandler(nil, nil)))
defer server.Close()

DefaultClient = &Client{
HTTPClient: &retryablehttp.HttpClient{Client: *server.Client()},
Endpoint: server.URL,
}
SetAPIKey("test")

err := DeleteResourceByExternalId(context.Background(), DeleteResourceByExternalIdOpts{
OrganizationId: "org_01H945H0YD4F97JN3MNHBFPG37",
ResourceTypeSlug: "document",
ExternalId: "my-document-1",
})
require.NoError(t, err)
}
111 changes: 108 additions & 3 deletions pkg/authorization/client.go
Original file line number Diff line number Diff line change
@@ -1,15 +1,20 @@
package authorization

import (
"bytes"
"context"
"encoding/json"
"errors"
"fmt"
"net/http"
"net/url"
"sync"
"time"

workos "github.com/workos/workos-go/v6/internal/workos"
"github.com/workos/workos-go/v6/pkg/common"
"github.com/workos/workos-go/v6/pkg/retryablehttp"
"github.com/workos/workos-go/v6/pkg/workos_errors"
)

// DefaultListSize is the default number of records to return in list responses.
Expand Down Expand Up @@ -638,19 +643,119 @@ func (c *Client) ListResources(ctx context.Context, opts ListAuthorizationResour
// GetResourceByExternalId gets a resource by its external Id.
func (c *Client) GetResourceByExternalId(ctx context.Context, opts GetResourceByExternalIdOpts) (AuthorizationResource, error) {
c.once.Do(c.init)
return AuthorizationResource{}, errors.New("not implemented")

endpoint := fmt.Sprintf(
"%s/authorization/organizations/%s/resources/%s/%s",
c.Endpoint,
url.PathEscape(opts.OrganizationId),
url.PathEscape(opts.ResourceTypeSlug),
url.PathEscape(opts.ExternalId),
)

req, err := http.NewRequest(http.MethodGet, endpoint, nil)
if err != nil {
return AuthorizationResource{}, err
}

req = req.WithContext(ctx)
req.Header.Set("Authorization", "Bearer "+c.APIKey)
req.Header.Set("Content-Type", "application/json")
req.Header.Set("User-Agent", "workos-go/"+workos.Version)

res, err := c.HTTPClient.Do(req)
if err != nil {
return AuthorizationResource{}, err
}
defer res.Body.Close()

if err = workos_errors.TryGetHTTPError(res); err != nil {
return AuthorizationResource{}, err
}

var body AuthorizationResource
dec := json.NewDecoder(res.Body)
err = dec.Decode(&body)
return body, err
}

// UpdateResourceByExternalId updates a resource by its external Id.
func (c *Client) UpdateResourceByExternalId(ctx context.Context, opts UpdateResourceByExternalIdOpts) (AuthorizationResource, error) {
c.once.Do(c.init)
return AuthorizationResource{}, errors.New("not implemented")

data, err := c.JSONEncode(opts)
if err != nil {
return AuthorizationResource{}, err
}

endpoint := fmt.Sprintf(
"%s/authorization/organizations/%s/resources/%s/%s",
c.Endpoint,
url.PathEscape(opts.OrganizationId),
url.PathEscape(opts.ResourceTypeSlug),
url.PathEscape(opts.ExternalId),
)

req, err := http.NewRequest(http.MethodPatch, endpoint, bytes.NewBuffer(data))
if err != nil {
return AuthorizationResource{}, err
}

req = req.WithContext(ctx)
req.Header.Set("Authorization", "Bearer "+c.APIKey)
req.Header.Set("Content-Type", "application/json")
req.Header.Set("User-Agent", "workos-go/"+workos.Version)

res, err := c.HTTPClient.Do(req)
if err != nil {
return AuthorizationResource{}, err
}
defer res.Body.Close()

if err = workos_errors.TryGetHTTPError(res); err != nil {
return AuthorizationResource{}, err
}

var body AuthorizationResource
dec := json.NewDecoder(res.Body)
err = dec.Decode(&body)
return body, err
}

// DeleteResourceByExternalId deletes a resource by its external Id.
func (c *Client) DeleteResourceByExternalId(ctx context.Context, opts DeleteResourceByExternalIdOpts) error {
c.once.Do(c.init)
return errors.New("not implemented")

endpoint := fmt.Sprintf(
"%s/authorization/organizations/%s/resources/%s/%s",
c.Endpoint,
url.PathEscape(opts.OrganizationId),
url.PathEscape(opts.ResourceTypeSlug),
url.PathEscape(opts.ExternalId),
)

req, err := http.NewRequest(http.MethodDelete, endpoint, nil)
if err != nil {
return err
}

req = req.WithContext(ctx)
req.Header.Set("Authorization", "Bearer "+c.APIKey)
req.Header.Set("Content-Type", "application/json")
req.Header.Set("User-Agent", "workos-go/"+workos.Version)

if opts.CascadeDelete {
q := req.URL.Query()
q.Set("cascade_delete", "true")
req.URL.RawQuery = q.Encode()
}

res, err := c.HTTPClient.Do(req)
if err != nil {
return err
}
defer res.Body.Close()

return workos_errors.TryGetHTTPError(res)
}

// Check performs an authorization check.
Expand Down
Loading
Loading