43 lines
1012 B
Go
43 lines
1012 B
Go
package csv2
|
|
|
|
import (
|
|
"encoding/csv"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
)
|
|
|
|
func TestLoad(t *testing.T) {
|
|
type specie struct {
|
|
Name string `csv:"name"`
|
|
FavFood string `csv:"fav_food"`
|
|
Age uint16 `csv:"age"`
|
|
Weight float32 `csv:"weight"`
|
|
Available bool `csv:"available"`
|
|
}
|
|
body := `name,available,fav_food,weight,age
|
|
Alice,false,Bananas,19.22,5
|
|
Frank,true,Burrito,14,9
|
|
Joel,true,Pesto,32.5,21`
|
|
exp := []specie{
|
|
{Name: "Alice", FavFood: "Bananas", Age: 5, Weight: 19.22, Available: false},
|
|
{Name: "Frank", FavFood: "Burrito", Age: 9, Weight: 14, Available: true},
|
|
{Name: "Joel", FavFood: "Pesto", Age: 21, Weight: 32.5, Available: true},
|
|
}
|
|
|
|
csvReader := csv.NewReader(strings.NewReader(body))
|
|
r := NewReader(csvReader)
|
|
r.ColumnNamesInFirstRow = true
|
|
|
|
var out []specie
|
|
err := r.Load(&out)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
|
|
if !cmp.Equal(exp, out) {
|
|
t.Errorf("Result value is different: %s", cmp.Diff(exp, out))
|
|
}
|
|
}
|