Download internal/config/options_request_timeout_test.go from SaylorTwift/crush: direct link, hf CLI and curl.
- Browser
- Download file 1.27 kB
-
https://huggingface.co/SaylorTwift/crush/resolve/main/internal/config/options_request_timeout_test.go
- Command line
-
hf download hf://SaylorTwift/crush/internal/config/options_request_timeout_test.go
-
curl -L -o options_request_timeout_test.go https://huggingface.co/SaylorTwift/crush/resolve/main/internal/config/options_request_timeout_test.go
1.27 kB
| package config | |
| import ( | |
| "encoding/json" | |
| "testing" | |
| "time" | |
| "github.com/stretchr/testify/require" | |
| ) | |
| func ptr[T any](v T) *T { return &v } | |
| func TestOptionsGetRequestTimeout(t *testing.T) { | |
| t.Parallel() | |
| tests := []struct { | |
| name string | |
| options *Options | |
| expected time.Duration | |
| }{ | |
| { | |
| name: "nil options", | |
| options: nil, | |
| expected: DefaultRequestTimeout, | |
| }, | |
| { | |
| name: "unset field", | |
| options: &Options{}, | |
| expected: DefaultRequestTimeout, | |
| }, | |
| { | |
| name: "disabled", | |
| options: &Options{RequestTimeout: ptr(0)}, | |
| expected: 0, | |
| }, | |
| { | |
| name: "negative disables too", | |
| options: &Options{RequestTimeout: ptr(-1)}, | |
| expected: 0, | |
| }, | |
| { | |
| name: "seconds to duration", | |
| options: &Options{RequestTimeout: ptr(300)}, | |
| expected: 5 * time.Minute, | |
| }, | |
| } | |
| for _, tt := range tests { | |
| t.Run(tt.name, func(t *testing.T) { | |
| t.Parallel() | |
| require.Equal(t, tt.expected, tt.options.GetRequestTimeout()) | |
| }) | |
| } | |
| } | |
| func TestOptionsRequestTimeoutFromJSON(t *testing.T) { | |
| t.Parallel() | |
| var cfg Config | |
| require.NoError(t, json.Unmarshal([]byte(`{"options":{"request_timeout":300}}`), &cfg)) | |
| require.Equal(t, 300, *cfg.Options.RequestTimeout) | |
| require.Equal(t, 5*time.Minute, cfg.Options.GetRequestTimeout()) | |
| } | |