package utils
 
 import (
 	"os"
 	"testing"
 
 	"github.com/Sirupsen/logrus"
 )
 
 func TestEnableDebug(t *testing.T) {
 	defer func() {
 		os.Setenv("DEBUG", "")
 		logrus.SetLevel(logrus.InfoLevel)
 	}()
 	EnableDebug()
 	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 TestDisableDebug(t *testing.T) {
 	DisableDebug()
 	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 TestDebugEnabled(t *testing.T) {
 	EnableDebug()
 	if !IsDebugEnabled() {
 		t.Fatal("expected debug enabled, got false")
 	}
 	DisableDebug()
 	if IsDebugEnabled() {
 		t.Fatal("expected debug disabled, got true")
 	}
 }