Merge pull request #3 from posener/predicate-as-function-type

Predicate as function type

1 files changed, 8 insertions, 10 deletions

diff --git a/gocomplete/tests.go b/gocomplete/tests.go
index 0ab99ec..60218a5 100644
--- a/gocomplete/tests.go
+++ b/gocomplete/tests.go
@@ -11,16 +11,14 @@ import (
 	"github.com/posener/complete"
 )
 
-func predictTest(testType string) *complete.Predicate {
-	return &complete.Predicate{
-		Predictor: func() []complete.Option {
-			tests := testNames(testType)
-			options := make([]complete.Option, len(tests))
-			for i := range tests {
-				options[i] = complete.Arg(tests[i])
-			}
-			return options
-		},
+func predictTest(testType string) complete.Predicate {
+	return func(last string) []complete.Option {
+		tests := testNames(testType)
+		options := make([]complete.Option, len(tests))
+		for i := range tests {
+			options[i] = complete.Arg(tests[i])
+		}
+		return options
 	}
 }