github.com/anthropics/anthropic-sdk-go@v1.70.1/betaorganizationworkspace_test.go (about)

     1  // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
     2  
     3  package anthropic_test
     4  
     5  import (
     6  	"context"
     7  	"errors"
     8  	"os"
     9  	"testing"
    10  
    11  	"github.com/anthropics/anthropic-sdk-go"
    12  	"github.com/anthropics/anthropic-sdk-go/internal/testutil"
    13  	"github.com/anthropics/anthropic-sdk-go/option"
    14  	"github.com/anthropics/anthropic-sdk-go/shared/constant"
    15  )
    16  
    17  func TestBetaOrganizationWorkspaceNewWithOptionalParams(t *testing.T) {
    18  	baseURL := "http://localhost:4010"
    19  	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
    20  		baseURL = envURL
    21  	}
    22  	if !testutil.CheckTestServer(t, baseURL) {
    23  		return
    24  	}
    25  	client := anthropic.NewClient(
    26  		option.WithBaseURL(baseURL),
    27  		option.WithAPIKey("my-anthropic-api-key"),
    28  	)
    29  	_, err := client.Beta.Organization.Workspaces.New(context.TODO(), anthropic.BetaOrganizationWorkspaceNewParams{
    30  		Name: "x",
    31  		DataResidency: anthropic.BetaDataResidencyCreateConfigParam{
    32  			AllowedInferenceGeos: anthropic.BetaDataResidencyCreateConfigAllowedInferenceGeosUnionParam{
    33  				OfUnrestricted: constant.ValueOf[constant.Unrestricted](),
    34  			},
    35  			DefaultInferenceGeo: anthropic.BetaDataResidencyCreateConfigDefaultInferenceGeoGlobal,
    36  			WorkspaceGeo:        anthropic.BetaDataResidencyCreateConfigWorkspaceGeoUs,
    37  		},
    38  		DisplayColor:  anthropic.String("#6C5BB9"),
    39  		ExternalKeyID: anthropic.String("ekey_01SDCCSbTxrXDpWc1phhtcfK"),
    40  		Tags: map[string]string{
    41  			"env":  "prod",
    42  			"team": "platform",
    43  		},
    44  		Betas: []anthropic.AnthropicBeta{anthropic.AnthropicBetaMessageBatches2024_09_24},
    45  	})
    46  	if err != nil {
    47  		var apierr *anthropic.Error
    48  		if errors.As(err, &apierr) {
    49  			t.Log(string(apierr.DumpRequest(true)))
    50  		}
    51  		t.Fatalf("err should be nil: %s", err.Error())
    52  	}
    53  }
    54  
    55  func TestBetaOrganizationWorkspaceGet(t *testing.T) {
    56  	baseURL := "http://localhost:4010"
    57  	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
    58  		baseURL = envURL
    59  	}
    60  	if !testutil.CheckTestServer(t, baseURL) {
    61  		return
    62  	}
    63  	client := anthropic.NewClient(
    64  		option.WithBaseURL(baseURL),
    65  		option.WithAPIKey("my-anthropic-api-key"),
    66  	)
    67  	_, err := client.Beta.Organization.Workspaces.Get(context.TODO(), "workspace_id")
    68  	if err != nil {
    69  		var apierr *anthropic.Error
    70  		if errors.As(err, &apierr) {
    71  			t.Log(string(apierr.DumpRequest(true)))
    72  		}
    73  		t.Fatalf("err should be nil: %s", err.Error())
    74  	}
    75  }
    76  
    77  func TestBetaOrganizationWorkspaceUpdateWithOptionalParams(t *testing.T) {
    78  	baseURL := "http://localhost:4010"
    79  	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
    80  		baseURL = envURL
    81  	}
    82  	if !testutil.CheckTestServer(t, baseURL) {
    83  		return
    84  	}
    85  	client := anthropic.NewClient(
    86  		option.WithBaseURL(baseURL),
    87  		option.WithAPIKey("my-anthropic-api-key"),
    88  	)
    89  	_, err := client.Beta.Organization.Workspaces.Update(
    90  		context.TODO(),
    91  		"workspace_id",
    92  		anthropic.BetaOrganizationWorkspaceUpdateParams{
    93  			DataResidency: anthropic.BetaDataResidencyUpdateConfigParam{
    94  				AllowedInferenceGeos: anthropic.BetaDataResidencyUpdateConfigAllowedInferenceGeosUnionParam{
    95  					OfUnrestricted: constant.ValueOf[constant.Unrestricted](),
    96  				},
    97  				DefaultInferenceGeo: anthropic.BetaDataResidencyUpdateConfigDefaultInferenceGeoGlobal,
    98  			},
    99  			DisplayColor:  anthropic.String("#6C5BB9"),
   100  			ExternalKeyID: anthropic.String("ekey_01SDCCSbTxrXDpWc1phhtcfK"),
   101  			Name:          anthropic.String("x"),
   102  			Tags: map[string]string{
   103  				"env":  "prod",
   104  				"team": "platform",
   105  			},
   106  		},
   107  	)
   108  	if err != nil {
   109  		var apierr *anthropic.Error
   110  		if errors.As(err, &apierr) {
   111  			t.Log(string(apierr.DumpRequest(true)))
   112  		}
   113  		t.Fatalf("err should be nil: %s", err.Error())
   114  	}
   115  }
   116  
   117  func TestBetaOrganizationWorkspaceListWithOptionalParams(t *testing.T) {
   118  	baseURL := "http://localhost:4010"
   119  	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
   120  		baseURL = envURL
   121  	}
   122  	if !testutil.CheckTestServer(t, baseURL) {
   123  		return
   124  	}
   125  	client := anthropic.NewClient(
   126  		option.WithBaseURL(baseURL),
   127  		option.WithAPIKey("my-anthropic-api-key"),
   128  	)
   129  	_, err := client.Beta.Organization.Workspaces.List(context.TODO(), anthropic.BetaOrganizationWorkspaceListParams{
   130  		AfterID:         anthropic.String("after_id"),
   131  		BeforeID:        anthropic.String("before_id"),
   132  		IncludeArchived: anthropic.Bool(true),
   133  		Limit:           anthropic.Int(1),
   134  	})
   135  	if err != nil {
   136  		var apierr *anthropic.Error
   137  		if errors.As(err, &apierr) {
   138  			t.Log(string(apierr.DumpRequest(true)))
   139  		}
   140  		t.Fatalf("err should be nil: %s", err.Error())
   141  	}
   142  }
   143  
   144  func TestBetaOrganizationWorkspaceArchive(t *testing.T) {
   145  	baseURL := "http://localhost:4010"
   146  	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
   147  		baseURL = envURL
   148  	}
   149  	if !testutil.CheckTestServer(t, baseURL) {
   150  		return
   151  	}
   152  	client := anthropic.NewClient(
   153  		option.WithBaseURL(baseURL),
   154  		option.WithAPIKey("my-anthropic-api-key"),
   155  	)
   156  	_, err := client.Beta.Organization.Workspaces.Archive(context.TODO(), "workspace_id")
   157  	if err != nil {
   158  		var apierr *anthropic.Error
   159  		if errors.As(err, &apierr) {
   160  			t.Log(string(apierr.DumpRequest(true)))
   161  		}
   162  		t.Fatalf("err should be nil: %s", err.Error())
   163  	}
   164  }