mirror of
https://github.com/securego/gosec.git
synced 2024-11-06 03:55:50 +00:00
29 lines
647 B
Go
29 lines
647 B
Go
|
package testutils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"go/ast"
|
||
|
|
|
||
|
|
"github.com/GoASTScanner/gas"
|
||
|
|
)
|
||
|
|
|
||
|
|
// MockVisitor is useful for stubbing out ast.Visitor with callback
|
||
|
|
// and looking for specific conditions to exist.
|
||
|
|
type MockVisitor struct {
|
||
|
|
Context *gas.Context
|
||
|
|
Callback func(n ast.Node, ctx *gas.Context) bool
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewMockVisitor creates a new empty struct, the Context and
|
||
|
|
// Callback must be set manually. See call_list_test.go for an example.
|
||
|
|
func NewMockVisitor() *MockVisitor {
|
||
|
|
return &MockVisitor{}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Visit satisfies the ast.Visitor interface
|
||
|
|
func (v *MockVisitor) Visit(n ast.Node) ast.Visitor {
|
||
|
|
if v.Callback(n, v.Context) {
|
||
|
|
return v
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|