marshaler_test.go

package goptions

import (
	"fmt"
	"reflect"
	"strings"
	"testing"
)

type Name struct {
	FirstName string
	LastName  string
}

func (n *Name) MarshalGoption(val string) error {
	f := strings.SplitN(val, " ", 2)
	if len(f) != 2 {
		return fmt.Errorf("Incomplete name")
	}
	n.FirstName = f[0]
	n.LastName = f[1]
	return nil
}

func TestMarshaler(t *testing.T) {
	var args []string
	var err error
	var fs *FlagSet
	var options struct {
		Name *Name `goptions:"--name"`
	}
	args = []string{"--name", "Alexander Surma"}
	fs = NewFlagSet("goptions", &options)
	err = fs.Parse(args)
	if err != nil {
		t.Fatalf("Parsing failed: %s", err)
	}
	expected := &Name{
		FirstName: "Alexander",
		LastName:  "Surma",
	}
	if !reflect.DeepEqual(options.Name, expected) {
		t.Fatalf("Unexpected value: %#v", options)
	}
}

func TestArrayOfMarshaler(t *testing.T) {
	var args []string
	var err error
	var fs *FlagSet
	var options struct {
		Names []*Name `goptions:"--name"`
	}
	args = []string{"--name", "Alexander Surma", "--name", "Yo Mama"}
	fs = NewFlagSet("goptions", &options)
	err = fs.Parse(args)
	if err != nil {
		t.Fatalf("Parsing failed: %s", err)
	}
	expected := []*Name{
		&Name{
			FirstName: "Alexander",
			LastName:  "Surma",
		},
		&Name{
			FirstName: "Yo",
			LastName:  "Mama",
		},
	}
	if !reflect.DeepEqual(options.Names, expected) {
		t.Fatalf("Unexpected value: %#v", options)
	}
}