redis_test.go 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. package repository
  2. import (
  3. "testing"
  4. "time"
  5. "github.com/Wei-Shaw/sub2api/internal/config"
  6. "github.com/stretchr/testify/require"
  7. )
  8. func TestBuildRedisOptions(t *testing.T) {
  9. cfg := &config.Config{
  10. Redis: config.RedisConfig{
  11. Host: "localhost",
  12. Port: 6379,
  13. Username: "app-user",
  14. Password: "secret",
  15. DB: 2,
  16. DialTimeoutSeconds: 5,
  17. ReadTimeoutSeconds: 3,
  18. WriteTimeoutSeconds: 4,
  19. PoolSize: 100,
  20. MinIdleConns: 10,
  21. },
  22. }
  23. opts := buildRedisOptions(cfg)
  24. require.Equal(t, "localhost:6379", opts.Addr)
  25. require.Equal(t, "app-user", opts.Username)
  26. require.Equal(t, "secret", opts.Password)
  27. require.Equal(t, 2, opts.DB)
  28. require.Equal(t, 5*time.Second, opts.DialTimeout)
  29. require.Equal(t, 3*time.Second, opts.ReadTimeout)
  30. require.Equal(t, 4*time.Second, opts.WriteTimeout)
  31. require.Equal(t, 100, opts.PoolSize)
  32. require.Equal(t, 10, opts.MinIdleConns)
  33. require.Nil(t, opts.TLSConfig)
  34. // Test case with TLS enabled
  35. cfgTLS := &config.Config{
  36. Redis: config.RedisConfig{
  37. Host: "localhost",
  38. EnableTLS: true,
  39. },
  40. }
  41. optsTLS := buildRedisOptions(cfgTLS)
  42. require.NotNil(t, optsTLS.TLSConfig)
  43. require.Equal(t, "localhost", optsTLS.TLSConfig.ServerName)
  44. }