group_handler_openai_fast_test.go 905 B

123456789101112131415161718192021222324252627
  1. package admin
  2. import (
  3. "encoding/json"
  4. "testing"
  5. "github.com/stretchr/testify/require"
  6. )
  7. func TestGroupRequestsDecodeForceOpenAIFast(t *testing.T) {
  8. var createReq CreateGroupRequest
  9. require.NoError(t, json.Unmarshal([]byte(`{"name":"fast","force_openai_fast":true,"free_openai_fast":true}`), &createReq))
  10. require.True(t, createReq.ForceOpenAIFast)
  11. require.True(t, createReq.FreeOpenAIFast)
  12. var updateReq UpdateGroupRequest
  13. require.NoError(t, json.Unmarshal([]byte(`{"force_openai_fast":false,"free_openai_fast":false}`), &updateReq))
  14. require.NotNil(t, updateReq.ForceOpenAIFast)
  15. require.False(t, *updateReq.ForceOpenAIFast)
  16. require.NotNil(t, updateReq.FreeOpenAIFast)
  17. require.False(t, *updateReq.FreeOpenAIFast)
  18. var omitted UpdateGroupRequest
  19. require.NoError(t, json.Unmarshal([]byte(`{}`), &omitted))
  20. require.Nil(t, omitted.ForceOpenAIFast)
  21. require.Nil(t, omitted.FreeOpenAIFast)
  22. }