|
1 | 1 | package main |
2 | 2 |
|
3 | | - import ( |
4 | | - "bufio" |
5 | | - "flag" |
6 | | - "fmt" |
7 | | - "log" |
8 | | - "os" |
9 | | - "strings" |
| 3 | +import ( |
| 4 | + "bufio" |
| 5 | + "flag" |
| 6 | + "fmt" |
| 7 | + "log" |
| 8 | + "os" |
| 9 | + "strings" |
10 | 10 |
|
11 | | - "github.com/logrusorgru/aurora" |
12 | | - "github.com/pkg/errors" |
| 11 | + "github.com/logrusorgru/aurora" |
| 12 | + "github.com/pkg/errors" |
13 | 13 |
|
14 | | - "github.com/projectdiscovery/naabu/v2/cmd/internal/testutils" |
15 | | - ) |
| 14 | + "github.com/projectdiscovery/naabu/v2/internal/testutils" |
| 15 | +) |
16 | 16 |
|
17 | | - var ( |
18 | | - debug = os.Getenv("DEBUG") == "true" |
19 | | - success = aurora.Green("[✓]").String() |
20 | | - failed = aurora.Red("[✘]").String() |
21 | | - errored = false |
| 17 | +var ( |
| 18 | + debug = os.Getenv("DEBUG") == "true" |
| 19 | + success = aurora.Green("[✓]").String() |
| 20 | + failed = aurora.Red("[✘]").String() |
| 21 | + errored = false |
22 | 22 |
|
23 | | - mainNaabuBinary = flag.String("main", "", "Main Branch Naabu Binary") |
24 | | - devNaabuBinary = flag.String("dev", "", "Dev Branch Naabu Binary") |
25 | | - testcases = flag.String("testcases", "", "Test cases file for Naabu functional tests") |
26 | | - ) |
| 23 | + mainNaabuBinary = flag.String("main", "", "Main Branch Naabu Binary") |
| 24 | + devNaabuBinary = flag.String("dev", "", "Dev Branch Naabu Binary") |
| 25 | + testcases = flag.String("testcases", "", "Test cases file for Naabu functional tests") |
| 26 | +) |
27 | 27 |
|
28 | | - func main() { |
29 | | - flag.Parse() |
| 28 | +func main() { |
| 29 | + flag.Parse() |
30 | 30 |
|
31 | | - if err := runFunctionalTests(); err != nil { |
32 | | - log.Fatalf("Could not run functional tests: %s\n", err) |
33 | | - } |
34 | | - if errored { |
35 | | - os.Exit(1) |
36 | | - } |
37 | | - } |
| 31 | + if err := runFunctionalTests(); err != nil { |
| 32 | + log.Fatalf("Could not run functional tests: %s\n", err) |
| 33 | + } |
| 34 | + if errored { |
| 35 | + os.Exit(1) |
| 36 | + } |
| 37 | +} |
38 | 38 |
|
39 | | - func runFunctionalTests() error { |
40 | | - file, err := os.Open(*testcases) |
41 | | - if err != nil { |
42 | | - return errors.Wrap(err, "could not open test cases") |
43 | | - } |
44 | | - defer file.Close() |
| 39 | +func runFunctionalTests() error { |
| 40 | + file, err := os.Open(*testcases) |
| 41 | + if err != nil { |
| 42 | + return errors.Wrap(err, "could not open test cases") |
| 43 | + } |
| 44 | + defer file.Close() |
45 | 45 |
|
46 | | - scanner := bufio.NewScanner(file) |
47 | | - for scanner.Scan() { |
48 | | - text := strings.TrimSpace(scanner.Text()) |
49 | | - if text == "" { |
50 | | - continue |
51 | | - } |
52 | | - if err := runIndividualTestCase(text); err != nil { |
53 | | - errored = true |
54 | | - fmt.Fprintf(os.Stderr, "%s Test \"%s\" failed: %s\n", failed, text, err) |
55 | | - } else { |
56 | | - fmt.Printf("%s Test \"%s\" passed!\n", success, text) |
57 | | - } |
58 | | - } |
59 | | - return nil |
60 | | - } |
| 46 | + scanner := bufio.NewScanner(file) |
| 47 | + for scanner.Scan() { |
| 48 | + text := strings.TrimSpace(scanner.Text()) |
| 49 | + if text == "" { |
| 50 | + continue |
| 51 | + } |
| 52 | + if err := runIndividualTestCase(text); err != nil { |
| 53 | + errored = true |
| 54 | + fmt.Fprintf(os.Stderr, "%s Test \"%s\" failed: %s\n", failed, text, err) |
| 55 | + } else { |
| 56 | + fmt.Printf("%s Test \"%s\" passed!\n", success, text) |
| 57 | + } |
| 58 | + } |
| 59 | + return nil |
| 60 | +} |
61 | 61 |
|
62 | | - func runIndividualTestCase(testcase string) error { |
63 | | - parts := strings.Fields(testcase) |
| 62 | +func runIndividualTestCase(testcase string) error { |
| 63 | + parts := strings.Fields(testcase) |
64 | 64 |
|
65 | | - var finalArgs []string |
66 | | - var target string |
67 | | - if len(parts) > 1 { |
68 | | - finalArgs = parts[2:] |
69 | | - target = parts[0] |
70 | | - } |
71 | | - mainOutput, err := testutils.RunNaabuBinaryAndGetResults(target, *mainNaabuBinary, debug, finalArgs) |
72 | | - if err != nil { |
73 | | - return errors.Wrap(err, "could not run naabu main test") |
74 | | - } |
75 | | - devOutput, err := testutils.RunNaabuBinaryAndGetResults(target, *devNaabuBinary, debug, finalArgs) |
76 | | - if err != nil { |
77 | | - return errors.Wrap(err, "could not run naabu dev test") |
78 | | - } |
79 | | - if len(mainOutput) == len(devOutput) { |
80 | | - return nil |
81 | | - } |
82 | | - return fmt.Errorf("%s main is not equal to %s dev", mainOutput, devOutput) |
83 | | - } |
| 65 | + var finalArgs []string |
| 66 | + var target string |
| 67 | + if len(parts) > 1 { |
| 68 | + finalArgs = parts[2:] |
| 69 | + target = parts[0] |
| 70 | + } |
| 71 | + mainOutput, err := testutils.RunNaabuBinaryAndGetResults(target, *mainNaabuBinary, debug, finalArgs) |
| 72 | + if err != nil { |
| 73 | + return errors.Wrap(err, "could not run naabu main test") |
| 74 | + } |
| 75 | + devOutput, err := testutils.RunNaabuBinaryAndGetResults(target, *devNaabuBinary, debug, finalArgs) |
| 76 | + if err != nil { |
| 77 | + return errors.Wrap(err, "could not run naabu dev test") |
| 78 | + } |
| 79 | + if len(mainOutput) == len(devOutput) { |
| 80 | + return nil |
| 81 | + } |
| 82 | + return fmt.Errorf("%s main is not equal to %s dev", mainOutput, devOutput) |
| 83 | +} |
0 commit comments