textfile_test.go 382 B

1234567891011121314151617181920212223242526
  1. package iox
  2. import (
  3. "os"
  4. "testing"
  5. "github.com/stretchr/testify/assert"
  6. )
  7. func TestCountLines(t *testing.T) {
  8. const val = `1
  9. 2
  10. 3
  11. 4`
  12. file, err := os.CreateTemp(os.TempDir(), "test-")
  13. if err != nil {
  14. t.Fatal(err)
  15. }
  16. defer os.Remove(file.Name())
  17. file.WriteString(val)
  18. file.Close()
  19. lines, err := CountLines(file.Name())
  20. assert.Nil(t, err)
  21. assert.Equal(t, 4, lines)
  22. }