51 lines
845 B
Go
51 lines
845 B
Go
|
package qcode
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
func TestGetQType(t *testing.T) {
|
||
|
|
type args struct {
|
||
|
|
gql string
|
||
|
|
}
|
||
|
|
type ts struct {
|
||
|
|
name string
|
||
|
|
args args
|
||
|
|
want QType
|
||
|
|
}
|
||
|
|
tests := []ts{
|
||
|
|
ts{
|
||
|
|
name: "query",
|
||
|
|
args: args{gql: " query {"},
|
||
|
|
want: QTQuery,
|
||
|
|
},
|
||
|
|
ts{
|
||
|
|
name: "mutation",
|
||
|
|
args: args{gql: " mutation {"},
|
||
|
|
want: QTMutation,
|
||
|
|
},
|
||
|
|
ts{
|
||
|
|
name: "default query",
|
||
|
|
args: args{gql: " {"},
|
||
|
|
want: QTQuery,
|
||
|
|
},
|
||
|
|
ts{
|
||
|
|
name: "default query with comment",
|
||
|
|
args: args{gql: `# query is good
|
||
|
|
{`},
|
||
|
|
want: QTQuery,
|
||
|
|
},
|
||
|
|
ts{
|
||
|
|
name: "failed query with comment",
|
||
|
|
args: args{gql: `# query is good query {`},
|
||
|
|
want: -1,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, tt := range tests {
|
||
|
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
|
if got := GetQType(tt.args.gql); got != tt.want {
|
||
|
|
t.Errorf("GetQType() = %v, want %v", got, tt.want)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|