debug_test.go 943 B

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. package utils
  2. import (
  3. "os"
  4. "testing"
  5. "github.com/Sirupsen/logrus"
  6. )
  7. func TestEnableDebug(t *testing.T) {
  8. defer func() {
  9. os.Setenv("DEBUG", "")
  10. logrus.SetLevel(logrus.InfoLevel)
  11. }()
  12. EnableDebug()
  13. if os.Getenv("DEBUG") != "1" {
  14. t.Fatalf("expected DEBUG=1, got %s\n", os.Getenv("DEBUG"))
  15. }
  16. if logrus.GetLevel() != logrus.DebugLevel {
  17. t.Fatalf("expected log level %v, got %v\n", logrus.DebugLevel, logrus.GetLevel())
  18. }
  19. }
  20. func TestDisableDebug(t *testing.T) {
  21. DisableDebug()
  22. if os.Getenv("DEBUG") != "" {
  23. t.Fatalf("expected DEBUG=\"\", got %s\n", os.Getenv("DEBUG"))
  24. }
  25. if logrus.GetLevel() != logrus.InfoLevel {
  26. t.Fatalf("expected log level %v, got %v\n", logrus.InfoLevel, logrus.GetLevel())
  27. }
  28. }
  29. func TestDebugEnabled(t *testing.T) {
  30. EnableDebug()
  31. if !IsDebugEnabled() {
  32. t.Fatal("expected debug enabled, got false")
  33. }
  34. DisableDebug()
  35. if IsDebugEnabled() {
  36. t.Fatal("expected debug disabled, got true")
  37. }
  38. }