mirror of https://github.com/daffainfo/nuclei.git
80 lines
1.8 KiB
Go
80 lines
1.8 KiB
Go
package main
|
||
|
||
import (
|
||
"bufio"
|
||
"flag"
|
||
"fmt"
|
||
"log"
|
||
"os"
|
||
"strings"
|
||
|
||
"github.com/logrusorgru/aurora"
|
||
"github.com/pkg/errors"
|
||
"github.com/projectdiscovery/nuclei/v2/internal/testutils"
|
||
)
|
||
|
||
var (
|
||
success = aurora.Green("[✓]").String()
|
||
failed = aurora.Red("[✘]").String()
|
||
errored = false
|
||
|
||
mainNucleiBinary = flag.String("main", "", "Main Branch Nuclei Binary")
|
||
devNucleiBinary = flag.String("dev", "", "Dev Branch Nuclei Binary")
|
||
testcases = flag.String("testcases", "", "Test cases file for nuclei functional tests")
|
||
)
|
||
|
||
func main() {
|
||
flag.Parse()
|
||
|
||
if err := runFunctionalTests(); err != nil {
|
||
log.Fatalf("Could not run functional tests: %s\n", err)
|
||
}
|
||
if errored {
|
||
os.Exit(1)
|
||
}
|
||
}
|
||
|
||
func runFunctionalTests() error {
|
||
file, err := os.Open(*testcases)
|
||
if err != nil {
|
||
return errors.Wrap(err, "could not open test cases")
|
||
}
|
||
defer file.Close()
|
||
|
||
scanner := bufio.NewScanner(file)
|
||
for scanner.Scan() {
|
||
text := strings.TrimSpace(scanner.Text())
|
||
if text == "" {
|
||
continue
|
||
}
|
||
if err := runIndividualTestCase(text); err != nil {
|
||
errored = true
|
||
fmt.Fprintf(os.Stderr, "%s Test \"%s\" failed: %s\n", failed, text, err)
|
||
} else {
|
||
fmt.Printf("%s Test \"%s\" passed!\n", success, text)
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func runIndividualTestCase(testcase string) error {
|
||
parts := strings.Fields(testcase)
|
||
|
||
var finalArgs []string
|
||
if len(parts) > 1 {
|
||
finalArgs = parts[1:]
|
||
}
|
||
mainOutput, err := testutils.RunNucleiBinaryAndGetLoadedTemplates(*mainNucleiBinary, finalArgs)
|
||
if err != nil {
|
||
return errors.Wrap(err, "could not run nuclei main test")
|
||
}
|
||
devOutput, err := testutils.RunNucleiBinaryAndGetLoadedTemplates(*devNucleiBinary, finalArgs)
|
||
if err != nil {
|
||
return errors.Wrap(err, "could not run nuclei dev test")
|
||
}
|
||
if mainOutput == devOutput {
|
||
return nil
|
||
}
|
||
return fmt.Errorf("%s main is not equal to %s dev", mainOutput, devOutput)
|
||
}
|