config_test.go 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. package config
  2. import (
  3. "reflect"
  4. "testing"
  5. )
  6. func TestGetLogParam(t *testing.T) {
  7. log := Log{
  8. Compress: true,
  9. DisableColors: false,
  10. DisableTimestamp: false,
  11. Caller: true,
  12. Filename: "log.txt",
  13. Level: "trace",
  14. MaxAge: 30,
  15. MaxBackups: 3,
  16. MaxSize: 10,
  17. Output: "stdout",
  18. TimestampFormat: "2006-01-02 15:04:05",
  19. }
  20. tests := []struct {
  21. name string
  22. wantC Log
  23. wantErr bool
  24. }{
  25. // TODO: Add test cases.
  26. {
  27. name: "log test",
  28. wantC: log,
  29. },
  30. }
  31. for _, tt := range tests {
  32. t.Run(tt.name, func(t *testing.T) {
  33. gotC, err := GetLogParam()
  34. if (err != nil) != tt.wantErr {
  35. t.Errorf("GetLogParam() error = %v, wantErr %v", err, tt.wantErr)
  36. return
  37. }
  38. if !reflect.DeepEqual(gotC, tt.wantC) {
  39. t.Errorf("GetLogParam() gotC = %v, want %v", gotC, tt.wantC)
  40. }
  41. })
  42. }
  43. }
  44. func TestGetViperParam(t *testing.T) {
  45. v := viper{
  46. Remote: true,
  47. Provider: "etcd3",
  48. Endpoint: "http://etcd.familybaby.top:2379",
  49. Path: "/sys/config/upgrade",
  50. }
  51. tests := []struct {
  52. name string
  53. wantC viper
  54. wantErr bool
  55. }{
  56. // TODO: Add test cases.
  57. {
  58. name: "Viper test",
  59. wantC: v,
  60. },
  61. }
  62. for _, tt := range tests {
  63. t.Run(tt.name, func(t *testing.T) {
  64. gotC, err := GetViperParam()
  65. if (err != nil) != tt.wantErr {
  66. t.Errorf("GetViperParam() error = %v, wantErr %v", err, tt.wantErr)
  67. return
  68. }
  69. if !reflect.DeepEqual(gotC, tt.wantC) {
  70. t.Errorf("GetViperParam() gotC = %v, want %v", gotC, tt.wantC)
  71. }
  72. })
  73. }
  74. }