github.com/Prakhar-Agarwal-byte/moby@v0.0.0-20231027092010-a14e3e8ab87e/cli/debug/debug_test.go (about) 1 package debug // import "github.com/Prakhar-Agarwal-byte/moby/cli/debug" 2 3 import ( 4 "os" 5 "testing" 6 7 "github.com/containerd/log" 8 ) 9 10 func TestEnable(t *testing.T) { 11 t.Cleanup(func() { 12 _ = os.Setenv("DEBUG", "") 13 _ = log.SetLevel("info") 14 }) 15 Enable() 16 if debug := os.Getenv("DEBUG"); debug != "1" { 17 t.Fatalf("expected DEBUG=1, got %s", debug) 18 } 19 if lvl := log.GetLevel(); lvl != log.DebugLevel { 20 t.Fatalf("expected log level %v, got %v", log.DebugLevel, lvl) 21 } 22 } 23 24 func TestDisable(t *testing.T) { 25 Disable() 26 if debug := os.Getenv("DEBUG"); debug != "" { 27 t.Fatalf(`expected DEBUG="", got %s`, debug) 28 } 29 if lvl := log.GetLevel(); lvl != log.InfoLevel { 30 t.Fatalf("expected log level %v, got %v", log.InfoLevel, lvl) 31 } 32 } 33 34 func TestEnabled(t *testing.T) { 35 Enable() 36 if !IsEnabled() { 37 t.Fatal("expected debug enabled, got false") 38 } 39 Disable() 40 if IsEnabled() { 41 t.Fatal("expected debug disabled, got true") 42 } 43 }