module5_test.go

package module5

import (
	"bufio"
	"log"
	"os"
	"path"
	"strings"
	"testing"
)

func TestModule5goimportsContent(t *testing.T) {
	expected := "	\"fmt\""
	found := OpenFileAndFindNthString("./module5_code.go", 0, expected)
	if found != true {
		t.Errorf("the fmt package is not found")
	}

	expected = "	\"net/http\""
	found = OpenFileAndFindNthString("./module5_code.go", 0, expected)
	if found != true {
		t.Errorf("the net/http package is not found")
	}
}

// OpenFileAndFindNthString opens a file, look for Nth string splitted by a space, and return if given expected string is found or not
func OpenFileAndFindNthString(filename string, nth int, expected string) bool {
	f, err := os.Open(filename)
	if err != nil {
		log.Fatal(err)
	}
	defer f.Close()

	scanner := bufio.NewScanner(f)
	scanner.Split(bufio.ScanLines)

	for scanner.Scan() {
		t := scanner.Text()
		trimmed := strings.Trim(t, " ")
		if trimmed == "" {
			continue
		}

		// matching logic
		ss := strings.Split(trimmed, " ")
		if ss[nth] == expected {
			return true
		}
	}

	return false
}

// FindFileAtPath returns if theFilename is found at thePath
func FindFileAtPath(thePath string, theFilename string) bool {
	if _, err := os.Stat(path.Join(thePath, theFilename)); os.IsNotExist(err) {
		return false
	}
	return true
}

// OpenFileAndCountLines opens a file and returns number of lines in the file
func OpenFileAndCountLines(filename string) int {
	f, err := os.Open(filename)
	if err != nil {
		log.Fatal(err)
	}
	defer f.Close()

	scanner := bufio.NewScanner(f)
	number := 0
	for scanner.Scan() {
		number++
	}

	return number
}