37 lines
591 B
Go
37 lines
591 B
Go
package ast
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestCallType(t *testing.T) {
|
|
c := &Call{Func: "foo"}
|
|
scope := &BasicScope{
|
|
FuncMap: map[string]Function{
|
|
"foo": Function{ReturnType: TypeString},
|
|
},
|
|
}
|
|
|
|
actual, err := c.Type(scope)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
if actual != TypeString {
|
|
t.Fatalf("bad: %s", actual)
|
|
}
|
|
}
|
|
|
|
func TestCallType_invalid(t *testing.T) {
|
|
c := &Call{Func: "bar"}
|
|
scope := &BasicScope{
|
|
FuncMap: map[string]Function{
|
|
"foo": Function{ReturnType: TypeString},
|
|
},
|
|
}
|
|
|
|
_, err := c.Type(scope)
|
|
if err == nil {
|
|
t.Fatal("should error")
|
|
}
|
|
}
|