mirror of
https://github.com/hanzo-go/openai.git
synced 2026-09-24 00:31:39 +00:00
zeekayandHanzo Dev
88762272c3
The fork kept the upstream module path (module github.com/sashabaranov/ go-openai), so every consumer could only reach it through a "replace github.com/sashabaranov/go-openai => github.com/hanzoai/go-openai" directive — and Go replace directives do NOT propagate to consumers, so each downstream repo had to copy the same replace or fail to compile against fork-only fields (e.g. ReasoningContent). That is the opposite of one-and- only-one-way. Declare the fork's OWN module path so it can be required directly and inherited transitively. Package name stays 'openai' (import identifier unchanged); only the module path + internal self-imports (/internal, /jsonschema) are renamed. Upstream issue-URL comments are left intact as attribution. Builds + vets clean. Co-authored-by: Hanzo Dev <[email protected]>
159 lines
4.6 KiB
Go
159 lines
4.6 KiB
Go
package openai //nolint:testpackage // testing private field
|
|
|
|
import (
|
|
utils "github.com/hanzoai/go-openai/internal"
|
|
"github.com/hanzoai/go-openai/internal/test/checks"
|
|
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
type mockFormBuilder struct {
|
|
mockCreateFormFile func(string, *os.File) error
|
|
mockCreateFormFileReader func(string, io.Reader, string) error
|
|
mockWriteField func(string, string) error
|
|
mockClose func() error
|
|
}
|
|
|
|
func (fb *mockFormBuilder) CreateFormFile(fieldname string, file *os.File) error {
|
|
return fb.mockCreateFormFile(fieldname, file)
|
|
}
|
|
|
|
func (fb *mockFormBuilder) CreateFormFileReader(fieldname string, r io.Reader, filename string) error {
|
|
return fb.mockCreateFormFileReader(fieldname, r, filename)
|
|
}
|
|
|
|
func (fb *mockFormBuilder) WriteField(fieldname, value string) error {
|
|
return fb.mockWriteField(fieldname, value)
|
|
}
|
|
|
|
func (fb *mockFormBuilder) Close() error {
|
|
return fb.mockClose()
|
|
}
|
|
|
|
func (fb *mockFormBuilder) FormDataContentType() string {
|
|
return ""
|
|
}
|
|
|
|
func TestImageFormBuilderFailures(t *testing.T) {
|
|
config := DefaultConfig("")
|
|
config.BaseURL = ""
|
|
client := NewClientWithConfig(config)
|
|
|
|
mockBuilder := &mockFormBuilder{}
|
|
client.createFormBuilder = func(io.Writer) utils.FormBuilder {
|
|
return mockBuilder
|
|
}
|
|
ctx := context.Background()
|
|
|
|
req := ImageEditRequest{
|
|
Mask: &os.File{},
|
|
}
|
|
|
|
mockFailedErr := fmt.Errorf("mock form builder fail")
|
|
mockBuilder.mockCreateFormFile = func(string, *os.File) error {
|
|
return mockFailedErr
|
|
}
|
|
_, err := client.CreateEditImage(ctx, req)
|
|
checks.ErrorIs(t, err, mockFailedErr, "CreateImage should return error if form builder fails")
|
|
|
|
mockBuilder.mockCreateFormFile = func(name string, _ *os.File) error {
|
|
if name == "mask" {
|
|
return mockFailedErr
|
|
}
|
|
return nil
|
|
}
|
|
_, err = client.CreateEditImage(ctx, req)
|
|
checks.ErrorIs(t, err, mockFailedErr, "CreateImage should return error if form builder fails")
|
|
|
|
mockBuilder.mockCreateFormFile = func(string, *os.File) error {
|
|
return nil
|
|
}
|
|
|
|
var failForField string
|
|
mockBuilder.mockWriteField = func(fieldname, _ string) error {
|
|
if fieldname == failForField {
|
|
return mockFailedErr
|
|
}
|
|
return nil
|
|
}
|
|
|
|
failForField = "prompt"
|
|
_, err = client.CreateEditImage(ctx, req)
|
|
checks.ErrorIs(t, err, mockFailedErr, "CreateImage should return error if form builder fails")
|
|
|
|
failForField = "n"
|
|
_, err = client.CreateEditImage(ctx, req)
|
|
checks.ErrorIs(t, err, mockFailedErr, "CreateImage should return error if form builder fails")
|
|
|
|
failForField = "size"
|
|
_, err = client.CreateEditImage(ctx, req)
|
|
checks.ErrorIs(t, err, mockFailedErr, "CreateImage should return error if form builder fails")
|
|
|
|
failForField = "response_format"
|
|
_, err = client.CreateEditImage(ctx, req)
|
|
checks.ErrorIs(t, err, mockFailedErr, "CreateImage should return error if form builder fails")
|
|
|
|
failForField = ""
|
|
mockBuilder.mockClose = func() error {
|
|
return mockFailedErr
|
|
}
|
|
_, err = client.CreateEditImage(ctx, req)
|
|
checks.ErrorIs(t, err, mockFailedErr, "CreateImage should return error if form builder fails")
|
|
}
|
|
|
|
func TestVariImageFormBuilderFailures(t *testing.T) {
|
|
config := DefaultConfig("")
|
|
config.BaseURL = ""
|
|
client := NewClientWithConfig(config)
|
|
|
|
mockBuilder := &mockFormBuilder{}
|
|
client.createFormBuilder = func(io.Writer) utils.FormBuilder {
|
|
return mockBuilder
|
|
}
|
|
ctx := context.Background()
|
|
|
|
req := ImageVariRequest{}
|
|
|
|
mockFailedErr := fmt.Errorf("mock form builder fail")
|
|
mockBuilder.mockCreateFormFile = func(string, *os.File) error {
|
|
return mockFailedErr
|
|
}
|
|
_, err := client.CreateVariImage(ctx, req)
|
|
checks.ErrorIs(t, err, mockFailedErr, "CreateVariImage should return error if form builder fails")
|
|
|
|
mockBuilder.mockCreateFormFile = func(string, *os.File) error {
|
|
return nil
|
|
}
|
|
|
|
var failForField string
|
|
mockBuilder.mockWriteField = func(fieldname, _ string) error {
|
|
if fieldname == failForField {
|
|
return mockFailedErr
|
|
}
|
|
return nil
|
|
}
|
|
|
|
failForField = "n"
|
|
_, err = client.CreateVariImage(ctx, req)
|
|
checks.ErrorIs(t, err, mockFailedErr, "CreateVariImage should return error if form builder fails")
|
|
|
|
failForField = "size"
|
|
_, err = client.CreateVariImage(ctx, req)
|
|
checks.ErrorIs(t, err, mockFailedErr, "CreateVariImage should return error if form builder fails")
|
|
|
|
failForField = "response_format"
|
|
_, err = client.CreateVariImage(ctx, req)
|
|
checks.ErrorIs(t, err, mockFailedErr, "CreateVariImage should return error if form builder fails")
|
|
|
|
failForField = ""
|
|
mockBuilder.mockClose = func() error {
|
|
return mockFailedErr
|
|
}
|
|
_, err = client.CreateVariImage(ctx, req)
|
|
checks.ErrorIs(t, err, mockFailedErr, "CreateImage should return error if form builder fails")
|
|
}
|