typeset: more testing

Author: scott-cotton

typeset/node.go

@@ -116,10 +116,45 @@ func (n *node) PlainDecode(r io.Reader) error {
 	case Tuple:
 		err = wrapJoinDecode(r, "(", ", ", ")", &n.fields)
 	case Func:
+		err = n.decodeFunc(r)
 	}
 	return nil
 }
 
+func (n *node) decodeFunc(r io.Reader) error {
+	buf := make([]byte, 2)
+	_, err := io.ReadFull(r, buf)
+	if err != nil {
+		return err
+	}
+	n.key = NoType
+	switch buf[0] {
+	case 'f':
+	case 'm':
+		_, err = fmt.Fscanf(r, "%08x", &n.key)
+		if err != nil {
+			return err
+		}
+	default:
+		return fmt.Errorf("unexpected '%s'", string(buf))
+	}
+	switch buf[1] {
+	case '+':
+		n.variadic = true
+	case '-':
+		n.variadic = false
+	default:
+		return fmt.Errorf("unexpected '%s'", string(buf))
+	}
+	err = wrapJoinDecode(r, "(", ", ", ")", &n.params)
+	if err != nil {
+		return err
+	}
+	err = wrapJoinDecode(r, "(", ", ", ")", &n.results)
+
+	return nil
+}
+
 func wrapJoinEncode(w io.Writer, left, sep, right string, elts []named) error {
 	_, err := fmt.Fprintf(w, left)
 	if err != nil {

typeset/t.go

@@ -24,7 +24,7 @@ type T struct {
 }
 
 const (
-	initCap = 128
+	initCap = 64
 )
 
 func New() *T {

typeset/t_test.go

@@ -17,9 +17,20 @@ package typeset
 import (
 	"bytes"
 	"go/types"
+	"os"
 	"testing"
 )
 
+func TestTypeSetGrow(t *testing.T) {
+	ts := New()
+	var base types.Type = types.Typ[types.Int64]
+	var palBase Type = ts.FromGoType(base)
+	for i := 0; i < 2025; i++ {
+		palBase = ts.getPointer(palBase)
+	}
+	ts.PlainEncode(os.Stdout)
+}
+
 func TestTypeSet(t *testing.T) {
 	ts := New()
 	sli := types.NewSlice(types.Typ[types.Int])