main_test.go 634 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. package main
  2. import (
  3. "bufio"
  4. "bytes"
  5. "strings"
  6. "testing"
  7. )
  8. var testOk = `1
  9. 2
  10. 3
  11. 3
  12. 4
  13. 5`
  14. var testOkResult = `1
  15. 2
  16. 3
  17. 4
  18. 5
  19. `
  20. func TestOk(t *testing.T) {
  21. in := bufio.NewReader(strings.NewReader(testOk))
  22. out := new(bytes.Buffer)
  23. err := uniq(in, out)
  24. if err != nil {
  25. t.Errorf("test for OK Failed - error")
  26. }
  27. if out.String() != testOkResult {
  28. t.Errorf("test for OK Failed - results not match")
  29. }
  30. }
  31. var testFail = `1
  32. 2
  33. 1`
  34. func TestForError(t *testing.T) {
  35. in := bufio.NewReader(strings.NewReader(testFail))
  36. out := new(bytes.Buffer)
  37. err := uniq(in, out)
  38. if err == nil {
  39. t.Errorf("test for Error Failed - error")
  40. }
  41. }