elvish/pkg/eval/vals/concat_test.go
Kurtis Rader 71cd3835bc Don't dot import pkg/tt
Qualified imports of pkg/tt outnumber unqualified (27 to 24). Improve
consistency, and clarity, by changing the dot (unqualified) imports of
that package symbols to qualified.
2022-06-04 23:39:19 +01:00

62 lines
1.7 KiB
Go

package vals
import (
"errors"
"math/big"
"testing"
"src.elv.sh/pkg/tt"
)
// An implementation for Concatter that accepts strings, returns a special
// error when rhs is a float64, and returns ErrConcatNotImplemented when rhs is
// of other types.
type concatter struct{}
var errBadFloat64 = errors.New("float64 is bad")
func (concatter) Concat(rhs any) (any, error) {
switch rhs := rhs.(type) {
case string:
return "concatter " + rhs, nil
case float64:
return nil, errBadFloat64
default:
return nil, ErrConcatNotImplemented
}
}
// An implementation of RConcatter that accepts all types.
type rconcatter struct{}
func (rconcatter) RConcat(lhs any) (any, error) {
return "rconcatter", nil
}
func TestConcat(t *testing.T) {
tt.Test(t, tt.Fn("Concat", Concat), tt.Table{
Args("foo", "bar").Rets("foobar", nil),
// string+number
Args("foo", 2).Rets("foo2", nil),
Args("foo", bigInt(z)).Rets("foo"+z, nil),
Args("foo", big.NewRat(1, 2)).Rets("foo1/2", nil),
Args("foo", 2.0).Rets("foo2.0", nil),
// number+string
Args(2, "foo").Rets("2foo", nil),
Args(bigInt(z), "foo").Rets(z+"foo", nil),
Args(big.NewRat(1, 2), "foo").Rets("1/2foo", nil),
Args(2.0, "foo").Rets("2.0foo", nil),
// LHS implements Concatter and succeeds
Args(concatter{}, "bar").Rets("concatter bar", nil),
// LHS implements Concatter but returns ErrConcatNotImplemented; RHS
// does not implement RConcatter
Args(concatter{}, 12).Rets(nil, cannotConcat{"!!vals.concatter", "number"}),
// LHS implements Concatter but returns another error
Args(concatter{}, 12.0).Rets(nil, errBadFloat64),
// LHS does not implement Concatter but RHS implements RConcatter
Args(12, rconcatter{}).Rets("rconcatter", nil),
})
}