package debug import ( "os" "testing" "github.com/sirupsen/logrus" ) func TestEnable(t *testing.T) { defer func() { os.Setenv("DEBUG", "") logrus.SetLevel(logrus.InfoLevel) }() Enable() if os.Getenv("DEBUG") != "1" { t.Fatalf("expected DEBUG=1, got %s\n", os.Getenv("DEBUG")) } if logrus.GetLevel() != logrus.DebugLevel { t.Fatalf("expected log level %v, got %v\n", logrus.DebugLevel, logrus.GetLevel()) } } func TestDisable(t *testing.T) { Disable() if os.Getenv("DEBUG") != "" { t.Fatalf("expected DEBUG=\"\", got %s\n", os.Getenv("DEBUG")) } if logrus.GetLevel() != logrus.InfoLevel { t.Fatalf("expected log level %v, got %v\n", logrus.InfoLevel, logrus.GetLevel()) } } func TestEnabled(t *testing.T) { Enable() if !IsEnabled() { t.Fatal("expected debug enabled, got false") } Disable() if IsEnabled() { t.Fatal("expected debug disabled, got true") } }