moar/m/reader_test.go

package m

import (
	"testing"
	"path"
	"runtime"
	"io/ioutil"
)

func _TestGetLines(t *testing.T, reader *_Reader) {
	t.Logf("Testing file: %s...", reader.name)

	lines := reader.GetLines(1, 10)
	if len(lines.lines) > 10 {
		t.Errorf("Asked for 10 lines, got too many: %d", len(lines.lines))
	}

	if len(lines.lines) < 10 {
		// No good plan for how to test short files more
		return
	}

	// Test clipping at the end
	lineCount := reader.LineCount()
	lines = reader.GetLines(lineCount, 10)
	if len(lines.lines) != 10 {
		t.Errorf("Asked for 10 lines but got %d", len(lines.lines))
		return
	}
	if lines.firstLineOneBased != lineCount - 9 {
		t.Errorf("Expected first line to be %d, was %d", lineCount - 9, lines.firstLineOneBased)
		return
	}
}

func _GetTestFiles() []string {
	// From: https://coderwall.com/p/_fmbug/go-get-path-to-current-file
	_, filename, _, ok := runtime.Caller(0)
	if !ok {
		panic("Getting current filename failed")
	}

	samplesDir := path.Join(path.Dir(filename), "../sample-files")

	files, err := ioutil.ReadDir(samplesDir)
    if err != nil {
		panic(err)
    }

	var filenames []string
    for _, file := range files {
		filenames = append(filenames, "../sample-files/" + file.Name())
	}

	return filenames
}

func TestGetLines(t *testing.T) {
	for _, file := range _GetTestFiles() {
		reader, e := NewReaderFromFilename(file)
		if e != nil {
			t.Errorf("Error opening file <%s>: %s", file, e.Error())
			continue
		}

		_TestGetLines(t, reader)
	}
}