124 lines
3.5 KiB
Go
124 lines
3.5 KiB
Go
package openai //nolint:testpackage // testing private field
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"testing"
|
|
|
|
utils "git.vaala.cloud/VaalaCat/go-openai/internal"
|
|
"git.vaala.cloud/VaalaCat/go-openai/internal/test/checks"
|
|
)
|
|
|
|
func TestFileBytesUploadWithFailingFormBuilder(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 := FileBytesRequest{
|
|
Name: "foo",
|
|
Bytes: []byte("foo"),
|
|
Purpose: PurposeAssistants,
|
|
}
|
|
|
|
mockError := fmt.Errorf("mockWriteField error")
|
|
mockBuilder.mockWriteField = func(string, string) error {
|
|
return mockError
|
|
}
|
|
_, err := client.CreateFileBytes(ctx, req)
|
|
checks.ErrorIs(t, err, mockError, "CreateFile should return error if form builder fails")
|
|
|
|
mockError = fmt.Errorf("mockCreateFormFile error")
|
|
mockBuilder.mockWriteField = func(string, string) error {
|
|
return nil
|
|
}
|
|
mockBuilder.mockCreateFormFileReader = func(string, io.Reader, string) error {
|
|
return mockError
|
|
}
|
|
_, err = client.CreateFileBytes(ctx, req)
|
|
checks.ErrorIs(t, err, mockError, "CreateFile should return error if form builder fails")
|
|
|
|
mockError = fmt.Errorf("mockClose error")
|
|
mockBuilder.mockWriteField = func(string, string) error {
|
|
return nil
|
|
}
|
|
mockBuilder.mockCreateFormFileReader = func(string, io.Reader, string) error {
|
|
return nil
|
|
}
|
|
mockBuilder.mockClose = func() error {
|
|
return mockError
|
|
}
|
|
_, err = client.CreateFileBytes(ctx, req)
|
|
checks.ErrorIs(t, err, mockError, "CreateFile should return error if form builder fails")
|
|
}
|
|
|
|
func TestFileUploadWithFailingFormBuilder(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 := FileRequest{
|
|
FileName: "test.go",
|
|
FilePath: "client.go",
|
|
Purpose: "fine-tune",
|
|
}
|
|
|
|
mockError := fmt.Errorf("mockWriteField error")
|
|
mockBuilder.mockWriteField = func(string, string) error {
|
|
return mockError
|
|
}
|
|
_, err := client.CreateFile(ctx, req)
|
|
checks.ErrorIs(t, err, mockError, "CreateFile should return error if form builder fails")
|
|
|
|
mockError = fmt.Errorf("mockCreateFormFile error")
|
|
mockBuilder.mockWriteField = func(string, string) error {
|
|
return nil
|
|
}
|
|
mockBuilder.mockCreateFormFile = func(string, *os.File) error {
|
|
return mockError
|
|
}
|
|
_, err = client.CreateFile(ctx, req)
|
|
checks.ErrorIs(t, err, mockError, "CreateFile should return error if form builder fails")
|
|
|
|
mockError = fmt.Errorf("mockClose error")
|
|
mockBuilder.mockWriteField = func(string, string) error {
|
|
return nil
|
|
}
|
|
mockBuilder.mockCreateFormFile = func(string, *os.File) error {
|
|
return nil
|
|
}
|
|
mockBuilder.mockClose = func() error {
|
|
return mockError
|
|
}
|
|
_, err = client.CreateFile(ctx, req)
|
|
if err == nil {
|
|
t.Fatal("CreateFile should return error if form builder fails")
|
|
}
|
|
checks.ErrorIs(t, err, mockError, "CreateFile should return error if form builder fails")
|
|
}
|
|
|
|
func TestFileUploadWithNonExistentPath(t *testing.T) {
|
|
config := DefaultConfig("")
|
|
config.BaseURL = ""
|
|
client := NewClientWithConfig(config)
|
|
|
|
ctx := context.Background()
|
|
req := FileRequest{
|
|
FilePath: "some non existent file path/F616FD18-589E-44A8-BF0C-891EAE69C455",
|
|
}
|
|
|
|
_, err := client.CreateFile(ctx, req)
|
|
checks.ErrorIs(t, err, os.ErrNotExist, "CreateFile should return error if file does not exist")
|
|
}
|