require on integration tests

liushuangls · Sep 3, 2024 · d77c1b0 · d77c1b0
1 parent ea3f090
commit d77c1b0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 15 deletions.
diff --git a/integrationtest/complete_test.go b/integrationtest/complete_test.go
@@ -5,10 +5,12 @@ import (
 	"testing"
 
 	"github.com/liushuangls/go-anthropic/v2"
+	"github.com/stretchr/testify/require"
 )
 
 func TestIntegrationComplete(t *testing.T) {
 	testAPIKey(t)
+	is := require.New(t)
 
 	t.Run("CreateComplete on real API", func(t *testing.T) {
 		client := anthropic.NewClient(APIKey)
@@ -21,9 +23,8 @@ func TestIntegrationComplete(t *testing.T) {
 		}
 
 		resp, err := client.CreateComplete(ctx, request)
-		if err != nil {
-			t.Fatalf("CreateComplete error: %s", err)
-		}
+		is.NoError(err)
+
 		t.Logf("CreateComplete resp: %+v", resp)
 
 		// RateLimitHeaders are not present on the Completions endpoint

diff --git a/integrationtest/messages_test.go b/integrationtest/messages_test.go
@@ -5,10 +5,13 @@ import (
 	"testing"
 
 	"github.com/liushuangls/go-anthropic/v2"
+	"github.com/stretchr/testify/require"
 )
 
 func TestIntegrationMessages(t *testing.T) {
 	testAPIKey(t)
+	is := require.New(t)
+
 	client := anthropic.NewClient(APIKey)
 	ctx := context.Background()
 
@@ -28,16 +31,14 @@ func TestIntegrationMessages(t *testing.T) {
 		newClient := anthropic.NewClient(APIKey, betaOpts)
 
 		resp, err := newClient.CreateMessages(ctx, request)
-		if err != nil {
-			t.Fatalf("CreateMessages error: %s", err)
-		}
+		is.NoError(err)
+
 		t.Logf("CreateMessages resp: %+v", resp)
 
 		t.Run("RateLimitHeaders are present", func(t *testing.T) {
 			rateLimHeader, err := resp.GetRateLimitHeaders()
-			if err != nil {
-				t.Fatalf("GetRateLimitHeaders error: %s", err)
-			}
+			is.NoError(err)
+
 			t.Logf("RateLimitHeaders: %+v", rateLimHeader)
 		})
 	})
@@ -47,16 +48,14 @@ func TestIntegrationMessages(t *testing.T) {
 			MessagesRequest: request,
 		}
 		resp, err := client.CreateMessagesStream(ctx, streamRequest)
-		if err != nil {
-			t.Fatalf("CreateMessagesStream error: %s", err)
-		}
+		is.NoError(err)
+
 		t.Logf("CreateMessagesStream resp: %+v", resp)
 
 		t.Run("RateLimitHeaders are present", func(t *testing.T) {
 			rateLimHeader, err := resp.GetRateLimitHeaders()
-			if err != nil {
-				t.Fatalf("GetRateLimitHeaders error: %s", err)
-			}
+			is.NoError(err)
+
 			t.Logf("RateLimitHeaders: %+v", rateLimHeader)
 		})
 	})