lazy_test.go

package multicsv

import (
	"encoding/csv"
	"os"
	"testing"
)

func TestLazyReader(t *testing.T) {
	r := &LazyReader{
		Init: func() (*csv.Reader, error) {
			f, err := os.Open("testdata/custom.csv")
			ok(t, err)

			reader := csv.NewReader(f)
			reader.LazyQuotes = true

			_, err = reader.Read() // skip header
			ok(t, err)

			return reader, nil
		},
	}

	var cnt int

	for {
		fields, err := r.Read()
		if err != nil {
			break
		}

		cnt++

		equals(t, 2, len(fields))
	}

	equals(t, 10, cnt)
}

func TestLazyFileReader(t *testing.T) {
	r := LazyFileReader("testdata/basic.csv", false)

	var cnt int

	for {
		fields, err := r.Read()
		if err != nil {
			break
		}

		cnt++

		equals(t, 6, len(fields))
	}

	equals(t, 6, cnt)
}

func TestLazyFileReaderSkipHeader(t *testing.T) {
	r := LazyFileReader("testdata/basic.csv", true)

	var cnt int

	for {
		fields, err := r.Read()
		if err != nil {
			break
		}

		cnt++

		equals(t, 6, len(fields))
	}

	equals(t, 5, cnt)
}

func TestLazyFileReaderError(t *testing.T) {
	r := LazyFileReader("testdata/nonexists.csv", true)

	_, err := r.Read()
	assert(t, err != nil, "err is nil")
}