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  }