main_test.go

package main

import (
	"bytes"
	"testing"
)

// TestCountWords tests the count function set to count words.
func TestCountWords(t *testing.T) {
	b := bytes.NewBufferString("word1 word2 word3 word4\n")
	exp := 4
	res := count(b, "", false, false)

	if res != exp {
		t.Errorf("Expected %d, got %d instead.\n", exp, res)
	}
}

// TestCountLines tests the count function set to count lines.
func TestCountLines(t *testing.T) {
	b := bytes.NewBufferString("word1 word2 word3\nline2\nline3 word1")
	exp := 3
	res := count(b, "", true, false)

	if res != exp {
		t.Errorf("Expected %d, got %d instead.", exp, res)
	}
}

// TestCountBytes tests the count function set to count bytes.
func TestCountBytes(t *testing.T) {
	b := bytes.NewBufferString("word1 word2")
	exp := 11
	res := count(b, "", false, true)

	if res != exp {
		t.Errorf("Expected %d, got %d instead.", exp, res)
	}
}

// TestCountWords tests the count function set to count words.
func TestReadFile(t *testing.T) {
	b := bytes.NewBufferString("")
	exp := 5
	res := count(b, "testdata/testfile.txt", false, false)

	if res != exp {
		t.Errorf("Expected %d, got %d instead.\n", exp, res)
	}
}