fix tests coverage

batch.go

@@ -4,6 +4,7 @@ import (
  "bytes"
  "context"
  "encoding/json"
+ "errors"
  "fmt"
  "net/http"
  "net/url"
@@ -148,6 +149,8 @@ type BatchResponse struct {
  Batch
 }
 
+var ErrUploadBatchFile = errors.New("upload batch file")
+
 // CreateBatch — API call to Create batch.
 func (c *Client) CreateBatch(
  ctx context.Context,
@@ -166,6 +169,7 @@ func (c *Client) CreateBatch(
  Purpose: PurposeBatch,
  })
  if err != nil {
+ err = errors.Join(ErrUploadBatchFile, err)
  return
  }
 

batch_test.go

@@ -29,8 +29,17 @@
  },
  },
  })
+ ctx := context.Background()
  _, err := client.CreateBatch(context.Background(), req)
  checks.NoError(t, err, "CreateBatch error")
+
+ server.RegisterHandler("/v1/files", func(w http.ResponseWriter, r *http.Request) {
+ w.WriteHeader(http.StatusGatewayTimeout)
+ })
+ _, err = client.CreateBatch(ctx, req)
+ msg := fmt.Sprintf("CreateBatch should return ErrUploadBatchFile, returned: %s", err)
+ fmt.Println(msg)
+ checks.ErrorIs(t, err, openai.ErrUploadBatchFile, msg)
 }
 
 func TestRetrieveBatch(t *testing.T) {