package dbg import ( "io/ioutil" "os" "testing" "git.sr.ht/~kisom/goutils/testio" "github.com/stretchr/testify/require" ) func TestNew(t *testing.T) { buf := testio.NewBufCloser(nil) dbg := New() dbg.out = buf dbg.Print("hello") dbg.Println("hello") dbg.Printf("hello %s", "world") require.Equal(t, 0, buf.Len()) dbg.Enabled = true dbg.Print("hello") // +5 dbg.Println("hello") // +6 dbg.Printf("hello %s", "world") // +11 require.Equal(t, 22, buf.Len()) err := dbg.Close() require.NoError(t, err) } func TestTo(t *testing.T) { buf := testio.NewBufCloser(nil) dbg := To(buf) dbg.Print("hello") dbg.Println("hello") dbg.Printf("hello %s", "world") require.Equal(t, 0, buf.Len()) dbg.Enabled = true dbg.Print("hello") // +5 dbg.Println("hello") // +6 dbg.Printf("hello %s", "world") // +11 require.Equal(t, 22, buf.Len()) err := dbg.Close() require.NoError(t, err) } func TestToFile(t *testing.T) { testFile, err := ioutil.TempFile("", "dbg") require.NoError(t, err) err = testFile.Close() require.NoError(t, err) testFileName := testFile.Name() defer os.Remove(testFileName) dbg, err := ToFile(testFileName) require.NoError(t, err) dbg.Print("hello") dbg.Println("hello") dbg.Printf("hello %s", "world") stat, err := os.Stat(testFileName) require.NoError(t, err) require.EqualValues(t, 0, stat.Size()) dbg.Enabled = true dbg.Print("hello") // +5 dbg.Println("hello") // +6 dbg.Printf("hello %s", "world") // +11 stat, err = os.Stat(testFileName) require.NoError(t, err) require.EqualValues(t, 22, stat.Size()) err = dbg.Close() require.NoError(t, err) } func TestWriting(t *testing.T) { data := []byte("hello, world") buf := testio.NewBufCloser(nil) dbg := To(buf) n, err := dbg.Write(data) require.NoError(t, err) require.EqualValues(t, 0, n) dbg.Enabled = true n, err = dbg.Write(data) require.NoError(t, err) require.EqualValues(t, 12, n) err = dbg.Close() require.NoError(t, err) } func TestToFileError(t *testing.T) { testFile, err := ioutil.TempFile("", "dbg") require.NoError(t, err) err = testFile.Chmod(0400) require.NoError(t, err) err = testFile.Close() require.NoError(t, err) testFileName := testFile.Name() _, err = ToFile(testFileName) require.Error(t, err) err = os.Remove(testFileName) require.NoError(t, err) }