reader_test.go

package ion

import (
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"testing"
)

var blacklist = map[string]bool{
	"ion-tests/iontestdata/good/emptyAnnotatedInt.10n":    true,
	"ion-tests/iontestdata/good/subfieldVarUInt32bit.ion": true,
	"ion-tests/iontestdata/good/utf16.ion":                true,
	"ion-tests/iontestdata/good/utf32.ion":                true,
	"ion-tests/iontestdata/good/whitespace.ion":           true,
	"ion-tests/iontestdata/good/item1.10n":                true,
}

type drainfunc func(t *testing.T, r Reader, f string)

func TestReadFiles(t *testing.T) {
	testReadDir(t, "ion-tests/iontestdata/good", func(t *testing.T, r Reader, f string) {
		drain(t, r, 0)
	})
}

func drain(t *testing.T, r Reader, level int) {
	for r.Next() {
		// print(level, r.Type())

		if !r.IsNull() {
			switch r.Type() {
			case StructType, ListType, SexpType:
				if err := r.StepIn(); err != nil {
					t.Fatal(err)
				}

				drain(t, r, level+1)

				if err := r.StepOut(); err != nil {
					t.Fatal(err)
				}
			}
		}
	}

	if r.Err() != nil {
		t.Fatal(r.Err())
	}
}

func print(level int, obj interface{}) {
	fmt.Print(" > ")
	for i := 0; i < level; i++ {
		fmt.Print("  ")
	}
	fmt.Println(obj)
}

func TestDecodeFiles(t *testing.T) {
	testReadDir(t, "ion-tests/iontestdata/good", func(t *testing.T, r Reader, f string) {
		// fmt.Println(f)
		d := NewDecoder(r)
		for {
			v, err := d.Decode()
			if err == ErrNoInput {
				break
			}
			if err != nil {
				t.Fatal(err)
			}
			// fmt.Println(v)
			_ = v
		}
	})
}

var emptyFiles = []string{
	"ion-tests/iontestdata/good/blank.ion",
	"ion-tests/iontestdata/good/empty.ion",
}

func isEmptyFile(f string) bool {
	for _, s := range emptyFiles {
		if f == s {
			return true
		}
	}
	return false
}

func testReadDir(t *testing.T, path string, d drainfunc) {
	files, err := ioutil.ReadDir(path)
	if err != nil {
		t.Fatal(err)
	}

	for _, file := range files {
		fp := filepath.Join(path, file.Name())
		if file.IsDir() {
			testReadDir(t, fp, d)
		} else {
			t.Run(fp, func(t *testing.T) {
				testReadFile(t, fp, d)
			})
		}
	}
}

func testReadFile(t *testing.T, path string, d drainfunc) {
	if _, ok := blacklist[path]; ok {
		return
	}

	// fmt.Println(path)

	file, err := os.Open(path)
	if err != nil {
		t.Fatal(err)
	}
	defer file.Close()

	r := NewReader(file)

	d(t, r, path)
}