config_test.go 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package sdk
  2. import (
  3. "net/http"
  4. "testing"
  5. "time"
  6. "github.com/stretchr/testify/assert"
  7. )
  8. func Test_Config(t *testing.T) {
  9. config := NewConfig()
  10. assert.NotNil(t, config, "NewConfig failed")
  11. assert.Equal(t, true, config.AutoRetry, "Default AutoRetry should be true")
  12. assert.Equal(t, 3, config.MaxRetryTime, "Default MaxRetryTime should be 3")
  13. assert.Equal(t, "", config.UserAgent, "Default UserAgent should be empty")
  14. assert.Equal(t, false, config.Debug, "Default AutoRetry should be false")
  15. assert.Equal(t, time.Duration(0), config.Timeout, "Default Timeout should be 10000000000")
  16. assert.Equal(t, (*http.Transport)(nil), config.HttpTransport, "Default HttpTransport should be nil")
  17. assert.Equal(t, false, config.EnableAsync, "Default EnableAsync should be false")
  18. assert.Equal(t, 1000, config.MaxTaskQueueSize, "Default MaxTaskQueueSize should be 1000")
  19. assert.Equal(t, 5, config.GoRoutinePoolSize, "Default GoRoutinePoolSize should be 5")
  20. assert.Equal(t, "HTTP", config.Scheme, "Default Scheme should be HTTP")
  21. transport := &http.Transport{
  22. MaxIdleConns: 10,
  23. IdleConnTimeout: 30 * time.Second,
  24. DisableCompression: true,
  25. }
  26. config.
  27. WithAutoRetry(false).
  28. WithMaxRetryTime(0).
  29. WithUserAgent("new user agent").
  30. WithDebug(true).
  31. WithTimeout(time.Duration(500000)).
  32. WithHttpTransport(transport).
  33. WithEnableAsync(true).
  34. WithMaxTaskQueueSize(1).
  35. WithGoRoutinePoolSize(10).
  36. WithScheme("HTTPS")
  37. assert.Equal(t, 0, config.MaxRetryTime)
  38. assert.Equal(t, false, config.AutoRetry)
  39. assert.Equal(t, "new user agent", config.UserAgent)
  40. assert.Equal(t, true, config.Debug)
  41. assert.Equal(t, time.Duration(500000), config.Timeout)
  42. assert.Equal(t, transport, config.HttpTransport)
  43. assert.Equal(t, true, config.EnableAsync)
  44. assert.Equal(t, 1, config.MaxTaskQueueSize)
  45. assert.Equal(t, 10, config.GoRoutinePoolSize)
  46. assert.Equal(t, "HTTPS", config.Scheme)
  47. }