mirror of
https://github.com/valitydev/thriftlint.git
synced 2024-11-06 00:05:20 +00:00
44 lines
1.3 KiB
Go
44 lines
1.3 KiB
Go
|
package thriftlint
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/alecthomas/go-thrift/parser"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCallCheckerValidation(t *testing.T) {
|
||
|
|
failf := func(*parser.Thrift) {}
|
||
|
|
require.Panics(t, func() { callChecker(failf, []interface{}{}) })
|
||
|
|
okf := func(*parser.Thrift) Messages { return nil }
|
||
|
|
require.NotPanics(t, func() { callChecker(okf, []interface{}{&parser.Thrift{}}) })
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCallChecker(t *testing.T) {
|
||
|
|
okfuncs := []interface{}{
|
||
|
|
func(*parser.Thrift, *parser.Struct, *parser.Field) Messages {
|
||
|
|
return Messages{}
|
||
|
|
},
|
||
|
|
func(*parser.Struct, *parser.Field) Messages { return Messages{} },
|
||
|
|
func(*parser.Thrift, *parser.Field) Messages { return Messages{} },
|
||
|
|
func(*parser.Field) Messages { return Messages{} },
|
||
|
|
func(self interface{}) Messages { return Messages{} },
|
||
|
|
func(parent, self interface{}) Messages { return Messages{} },
|
||
|
|
}
|
||
|
|
ancestors := []interface{}{&parser.Thrift{}, &parser.Struct{}, &parser.Field{}}
|
||
|
|
for _, okf := range okfuncs {
|
||
|
|
out := callChecker(okf, ancestors)
|
||
|
|
require.NotNil(t, out)
|
||
|
|
}
|
||
|
|
|
||
|
|
badfuncs := []interface{}{
|
||
|
|
func(*parser.Thrift) Messages { return Messages{} },
|
||
|
|
func(*parser.Struct) Messages { return Messages{} },
|
||
|
|
func(*parser.Field, *parser.Struct) Messages { return Messages{} },
|
||
|
|
}
|
||
|
|
for _, badf := range badfuncs {
|
||
|
|
out := callChecker(badf, ancestors)
|
||
|
|
require.Nil(t, out)
|
||
|
|
}
|
||
|
|
}
|