mirror of
https://github.com/mgechev/revive.git
synced 2025-06-08 23:26:29 +02:00
Add rule
This commit is contained in:
parent
419da3b95d
commit
1858f4f7c2
@ -30,7 +30,7 @@ func (r *BlankImportsRule) Apply(file *file.File, arguments rule.Arguments) []ru
|
|||||||
|
|
||||||
// Name returns the rule name.
|
// Name returns the rule name.
|
||||||
func (r *BlankImportsRule) Name() string {
|
func (r *BlankImportsRule) Name() string {
|
||||||
return "argument-limit"
|
return "blank-imports"
|
||||||
}
|
}
|
||||||
|
|
||||||
type lintBlankImports struct {
|
type lintBlankImports struct {
|
||||||
|
54
defaultrule/imports.go
Normal file
54
defaultrule/imports.go
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
package defaultrule
|
||||||
|
|
||||||
|
import (
|
||||||
|
"go/ast"
|
||||||
|
|
||||||
|
"github.com/mgechev/revive/file"
|
||||||
|
"github.com/mgechev/revive/rule"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ImportsRule lints given else constructs.
|
||||||
|
type ImportsRule struct{}
|
||||||
|
|
||||||
|
// Apply applies the rule to given file.
|
||||||
|
func (r *ImportsRule) Apply(file *file.File, arguments rule.Arguments) []rule.Failure {
|
||||||
|
var failures []rule.Failure
|
||||||
|
|
||||||
|
fileAst := file.GetAST()
|
||||||
|
walker := lintImports{
|
||||||
|
file: file,
|
||||||
|
fileAst: fileAst,
|
||||||
|
onFailure: func(failure rule.Failure) {
|
||||||
|
failures = append(failures, failure)
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
ast.Walk(walker, fileAst)
|
||||||
|
|
||||||
|
return failures
|
||||||
|
}
|
||||||
|
|
||||||
|
// Name returns the rule name.
|
||||||
|
func (r *ImportsRule) Name() string {
|
||||||
|
return "imports"
|
||||||
|
}
|
||||||
|
|
||||||
|
type lintImports struct {
|
||||||
|
file *file.File
|
||||||
|
fileAst *ast.File
|
||||||
|
onFailure func(rule.Failure)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w lintImports) Visit(n ast.Node) ast.Visitor {
|
||||||
|
for _, is := range w.fileAst.Imports {
|
||||||
|
if is.Name != nil && is.Name.Name == "." && !isTest(w.file) {
|
||||||
|
w.onFailure(rule.Failure{
|
||||||
|
Confidence: 1,
|
||||||
|
Failure: "should not use dot imports",
|
||||||
|
Node: is,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
61
defaultrule/imports_test.go
Normal file
61
defaultrule/imports_test.go
Normal file
@ -0,0 +1,61 @@
|
|||||||
|
package defaultrule
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/mgechev/revive/rule"
|
||||||
|
"github.com/mgechev/revive/testutil"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestImports_Failure(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
program := `
|
||||||
|
package foo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
[@f1]. "path"[/@f1]
|
||||||
|
)
|
||||||
|
|
||||||
|
`
|
||||||
|
testutil.AssertFailures(t, program, &ImportsRule{}, rule.Arguments{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestImports(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
program := `
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
/* MATCH /blank import/ */ _ "os"
|
||||||
|
|
||||||
|
/* MATCH /blank import/ */ _ "net/http"
|
||||||
|
_ "path"
|
||||||
|
)
|
||||||
|
|
||||||
|
`
|
||||||
|
testutil.AssertSuccess(t, program, &ImportsRule{}, rule.Arguments{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestImports_SkipTesting(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
program := `
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
/* MATCH /blank import/ */ _ "os"
|
||||||
|
|
||||||
|
/* MATCH /blank import/ */ _ "net/http"
|
||||||
|
. "path"
|
||||||
|
)
|
||||||
|
|
||||||
|
`
|
||||||
|
testutil.AssertSuccessWithName(t, program, "foo_test.go", &ImportsRule{}, rule.Arguments{})
|
||||||
|
}
|
@ -53,14 +53,14 @@ func stripAnnotations(code string) string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// AssertSuccess checks if given rule runs correctly with no failures.
|
// AssertSuccess checks if given rule runs correctly with no failures.
|
||||||
func AssertSuccess(t *testing.T, code string, testingRule rule.Rule, args rule.Arguments) {
|
func AssertSuccessWithName(t *testing.T, code, name string, testingRule rule.Rule, args rule.Arguments) {
|
||||||
annotations := extractFailures(code)
|
annotations := extractFailures(code)
|
||||||
if annotations != nil {
|
if annotations != nil {
|
||||||
t.Errorf("There should be no failure annotations when verifying successful rule analysis")
|
t.Errorf("There should be no failure annotations when verifying successful rule analysis")
|
||||||
}
|
}
|
||||||
|
|
||||||
var fileSet token.FileSet
|
var fileSet token.FileSet
|
||||||
file, err := file.New("testing.go", []byte(stripAnnotations(code)), &fileSet)
|
file, err := file.New(name, []byte(stripAnnotations(code)), &fileSet)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Errorf("Cannot parse testing file: %s", err.Error())
|
t.Errorf("Cannot parse testing file: %s", err.Error())
|
||||||
}
|
}
|
||||||
@ -78,6 +78,11 @@ func AssertSuccess(t *testing.T, code string, testingRule rule.Rule, args rule.A
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// AssertSuccess checks if given rule runs correctly with no failures.
|
||||||
|
func AssertSuccess(t *testing.T, code string, testingRule rule.Rule, args rule.Arguments) {
|
||||||
|
AssertSuccessWithName(t, code, "testing.go", testingRule, args)
|
||||||
|
}
|
||||||
|
|
||||||
// AssertFailures checks if given rule runs correctly with failures.
|
// AssertFailures checks if given rule runs correctly with failures.
|
||||||
func AssertFailures(t *testing.T, code string, testingRule rule.Rule, args rule.Arguments) {
|
func AssertFailures(t *testing.T, code string, testingRule rule.Rule, args rule.Arguments) {
|
||||||
annotations := extractFailures(code)
|
annotations := extractFailures(code)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user