mirror of
https://github.com/hanzo-go/openai.git
synced 2026-09-24 00:31:39 +00:00
zeekayandHanzo Dev
2e39f0248d
Modules declare go 1.27.1 and the build images follow. go fix applied the modernizers, and the stream request helper is now a generic method on Client; omitzero is not applied, since it changes what the JSON says. Co-authored-by: Hanzo Dev <[email protected]>
212 lines
4.3 KiB
Go
212 lines
4.3 KiB
Go
package jsonschema_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/hanzoai/go-openai/jsonschema"
|
|
)
|
|
|
|
func TestDefinition_MarshalJSON(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
def jsonschema.Definition
|
|
want string
|
|
}{
|
|
{
|
|
name: "Test with empty Definition",
|
|
def: jsonschema.Definition{},
|
|
want: `{"properties":{}}`,
|
|
},
|
|
{
|
|
name: "Test with Definition properties set",
|
|
def: jsonschema.Definition{
|
|
Type: jsonschema.String,
|
|
Description: "A string type",
|
|
Properties: map[string]jsonschema.Definition{
|
|
"name": {
|
|
Type: jsonschema.String,
|
|
},
|
|
},
|
|
},
|
|
want: `{
|
|
"type":"string",
|
|
"description":"A string type",
|
|
"properties":{
|
|
"name":{
|
|
"type":"string",
|
|
"properties":{}
|
|
}
|
|
}
|
|
}`,
|
|
},
|
|
{
|
|
name: "Test with nested Definition properties",
|
|
def: jsonschema.Definition{
|
|
Type: jsonschema.Object,
|
|
Properties: map[string]jsonschema.Definition{
|
|
"user": {
|
|
Type: jsonschema.Object,
|
|
Properties: map[string]jsonschema.Definition{
|
|
"name": {
|
|
Type: jsonschema.String,
|
|
},
|
|
"age": {
|
|
Type: jsonschema.Integer,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
want: `{
|
|
"type":"object",
|
|
"properties":{
|
|
"user":{
|
|
"type":"object",
|
|
"properties":{
|
|
"name":{
|
|
"type":"string",
|
|
"properties":{}
|
|
},
|
|
"age":{
|
|
"type":"integer",
|
|
"properties":{}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}`,
|
|
},
|
|
{
|
|
name: "Test with complex nested Definition",
|
|
def: jsonschema.Definition{
|
|
Type: jsonschema.Object,
|
|
Properties: map[string]jsonschema.Definition{
|
|
"user": {
|
|
Type: jsonschema.Object,
|
|
Properties: map[string]jsonschema.Definition{
|
|
"name": {
|
|
Type: jsonschema.String,
|
|
},
|
|
"age": {
|
|
Type: jsonschema.Integer,
|
|
},
|
|
"address": {
|
|
Type: jsonschema.Object,
|
|
Properties: map[string]jsonschema.Definition{
|
|
"city": {
|
|
Type: jsonschema.String,
|
|
},
|
|
"country": {
|
|
Type: jsonschema.String,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
want: `{
|
|
"type":"object",
|
|
"properties":{
|
|
"user":{
|
|
"type":"object",
|
|
"properties":{
|
|
"name":{
|
|
"type":"string",
|
|
"properties":{}
|
|
},
|
|
"age":{
|
|
"type":"integer",
|
|
"properties":{}
|
|
},
|
|
"address":{
|
|
"type":"object",
|
|
"properties":{
|
|
"city":{
|
|
"type":"string",
|
|
"properties":{}
|
|
},
|
|
"country":{
|
|
"type":"string",
|
|
"properties":{}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}`,
|
|
},
|
|
{
|
|
name: "Test with Array type Definition",
|
|
def: jsonschema.Definition{
|
|
Type: jsonschema.Array,
|
|
Items: &jsonschema.Definition{
|
|
Type: jsonschema.String,
|
|
},
|
|
Properties: map[string]jsonschema.Definition{
|
|
"name": {
|
|
Type: jsonschema.String,
|
|
},
|
|
},
|
|
},
|
|
want: `{
|
|
"type":"array",
|
|
"items":{
|
|
"type":"string",
|
|
"properties":{
|
|
|
|
}
|
|
},
|
|
"properties":{
|
|
"name":{
|
|
"type":"string",
|
|
"properties":{}
|
|
}
|
|
}
|
|
}`,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
wantBytes := []byte(tt.want)
|
|
var want map[string]any
|
|
err := json.Unmarshal(wantBytes, &want)
|
|
if err != nil {
|
|
t.Errorf("Failed to Unmarshal JSON: error = %v", err)
|
|
return
|
|
}
|
|
|
|
got := structToMap(t, tt.def)
|
|
gotPtr := structToMap(t, &tt.def)
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("MarshalJSON() got = %v, want %v", got, want)
|
|
}
|
|
if !reflect.DeepEqual(gotPtr, want) {
|
|
t.Errorf("MarshalJSON() gotPtr = %v, want %v", gotPtr, want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func structToMap(t *testing.T, v any) map[string]any {
|
|
t.Helper()
|
|
gotBytes, err := json.Marshal(v)
|
|
if err != nil {
|
|
t.Errorf("Failed to Marshal JSON: error = %v", err)
|
|
return nil
|
|
}
|
|
|
|
var got map[string]any
|
|
err = json.Unmarshal(gotBytes, &got)
|
|
if err != nil {
|
|
t.Errorf("Failed to Unmarshal JSON: error = %v", err)
|
|
return nil
|
|
}
|
|
return got
|
|
}
|