Logger package for gitea repositories
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

87 lines
2.4 KiB

  1. // Copyright 2019 The Gitea Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package log
  5. import (
  6. "fmt"
  7. "net/smtp"
  8. "strings"
  9. "testing"
  10. "time"
  11. "github.com/stretchr/testify/assert"
  12. )
  13. func TestSMTPLogger(t *testing.T) {
  14. prefix := "TestPrefix "
  15. level := INFO
  16. flags := LstdFlags | LUTC | Lfuncname
  17. username := "testuser"
  18. password := "testpassword"
  19. host := "testhost"
  20. subject := "testsubject"
  21. sendTos := []string{"testto1", "testto2"}
  22. logger := NewSMTPLogger()
  23. smtpLogger, ok := logger.(*SMTPLogger)
  24. assert.Equal(t, true, ok)
  25. err := logger.Init(fmt.Sprintf("{\"prefix\":\"%s\",\"level\":\"%s\",\"flags\":%d,\"username\":\"%s\",\"password\":\"%s\",\"host\":\"%s\",\"subject\":\"%s\",\"sendTos\":[\"%s\",\"%s\"]}", prefix, level.String(), flags, username, password, host, subject, sendTos[0], sendTos[1]))
  26. assert.NoError(t, err)
  27. assert.Equal(t, flags, smtpLogger.Flags)
  28. assert.Equal(t, level, smtpLogger.Level)
  29. assert.Equal(t, level, logger.GetLevel())
  30. location, _ := time.LoadLocation("EST")
  31. date := time.Date(2019, time.January, 13, 22, 3, 30, 15, location)
  32. dateString := date.UTC().Format("2006/01/02 15:04:05")
  33. event := Event{
  34. level: INFO,
  35. msg: "TEST MSG",
  36. caller: "CALLER",
  37. filename: "FULL/FILENAME",
  38. line: 1,
  39. time: date,
  40. }
  41. expected := fmt.Sprintf("%s%s %s:%d:%s [%c] %s\n", prefix, dateString, event.filename, event.line, event.caller, strings.ToUpper(event.level.String())[0], event.msg)
  42. var envToHost string
  43. var envFrom string
  44. var envTo []string
  45. var envMsg []byte
  46. smtpLogger.sendMailFn = func(addr string, a smtp.Auth, from string, to []string, msg []byte) error {
  47. envToHost = addr
  48. envFrom = from
  49. envTo = to
  50. envMsg = msg
  51. return nil
  52. }
  53. err = logger.LogEvent(&event)
  54. assert.NoError(t, err)
  55. assert.Equal(t, host, envToHost)
  56. assert.Equal(t, username, envFrom)
  57. assert.Equal(t, sendTos, envTo)
  58. assert.Contains(t, string(envMsg), expected)
  59. logger.Flush()
  60. event.level = WARN
  61. expected = fmt.Sprintf("%s%s %s:%d:%s [%c] %s\n", prefix, dateString, event.filename, event.line, event.caller, strings.ToUpper(event.level.String())[0], event.msg)
  62. err = logger.LogEvent(&event)
  63. assert.NoError(t, err)
  64. assert.Equal(t, host, envToHost)
  65. assert.Equal(t, username, envFrom)
  66. assert.Equal(t, sendTos, envTo)
  67. assert.Contains(t, string(envMsg), expected)
  68. logger.Close()
  69. }