Skip to content

Refactor TestInitialize assertions for clarity and correctness #661

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
hahwul merged 5 commits into from
Mar 14, 2025
Merged

Refactor TestInitialize assertions for clarity and correctness #661

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
b904ca6
Refactor TestInitialize assertions for clarity and correctness
hahwul Mar 14, 2025
6b1bd26
Update lib/func_test.go
hahwul Mar 14, 2025
776970c
Fix indentation in TestInitialize assertions for consistency
hahwul Mar 14, 2025
eb33ad3
Enhance TestInitialize options with additional parameters and assertions
hahwul Mar 14, 2025
667fa87
Add TestNewScan function to validate scanning results and error handling
hahwul Mar 14, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
145 changes: 96 additions & 49 deletions lib/func_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,32 +9,42 @@ import (

func TestInitialize(t *testing.T) {
opt := dalfox.Options{
Cookie: "ABCD=1234",
UniqParam: []string{"q"},
BlindURL: "your-callback-url",
CustomAlertValue: "1",
CustomAlertType: "none",
Header: []string{"Cookie: 1234", "ABCD: 1234"},
Data: "b=123",
UserAgent: "Test-UA",
ProxyAddress: "http://127.0.0.1",
Grep: "Test",
IgnoreReturn: "301",
IgnoreParams: []string{"qqq"},
OnlyDiscovery: true,
FollowRedirect: true,
Trigger: "https://google.com",
Timeout: 5,
Mining: true,
FindingDOM: true,
Concurrence: 10,
Delay: 2,
NoBAV: true,
NoGrep: true,
RemotePayloads: "portswigger",
RemoteWordlists: "burp",
PoCType: "curl",
UseBAV: false,
Cookie: "ABCD=1234",
UniqParam: []string{"q"},
BlindURL: "your-callback-url",
CustomAlertValue: "1",
CustomAlertType: "none",
Header: []string{"Cookie: 1234", "ABCD: 1234"},
Data: "b=123",
UserAgent: "Test-UA",
ProxyAddress: "http://127.0.0.1",
Grep: "Test",
IgnoreReturn: "301",
IgnoreParams: []string{"qqq"},
OnlyDiscovery: true,
FollowRedirect: true,
Trigger: "https://google.com",
TriggerMethod: "GET",
Timeout: 5,
Mining: true,
FindingDOM: true,
Concurrence: 10,
Delay: 2,
NoBAV: true,
NoGrep: true,
RemotePayloads: "portswigger",
RemoteWordlists: "burp",
PoCType: "curl",
UseBAV: false,
CustomPayloadFile: "payloads.txt",
OutputFile: "output.txt",
FoundAction: "notify",
FoundActionShell: "bash",
OnlyCustomPayload: true,
UseHeadless: false,
UseDeepDXSS: true,
WAFEvasion: true,
Sequence: 1,
}
target := dalfox.Target{
URL: "https://www.hahwul.com",
Expand All @@ -43,27 +53,64 @@ func TestInitialize(t *testing.T) {
}

newOptions := dalfox.Initialize(target, opt)
assert.NotEqual(t, newOptions.Cookie, "", "they should not bee equal")
assert.NotEqual(t, newOptions.UniqParam, "", "they should not bee equal")
assert.NotEqual(t, newOptions.BlindURL, "", "they should not bee equal")
assert.NotEqual(t, newOptions.CustomAlertValue, "", "they should not bee equal")
assert.NotEqual(t, newOptions.CustomAlertType, "", "they should not bee equal")
assert.NotEqual(t, newOptions.Data, "", "they should not bee equal")
assert.NotEqual(t, newOptions.UserAgent, "", "they should not bee equal")
assert.NotEqual(t, newOptions.ProxyAddress, "", "they should not bee equal")
assert.NotEqual(t, newOptions.Grep, "", "they should not bee equal")
assert.NotEqual(t, newOptions.IgnoreReturn, "", "they should not bee equal")
assert.NotEqual(t, newOptions.IgnoreParams, []string{""}, "they should not bee equal")
assert.NotEqual(t, newOptions.Trigger, "", "they should not bee equal")
assert.NotEqual(t, newOptions.Timeout, 10, "they should not bee equal")
assert.NotEqual(t, newOptions.Concurrence, 100, "they should not bee equal")
assert.NotEqual(t, newOptions.Delay, 0, "they should not bee equal")
assert.NotEqual(t, newOptions.NoBAV, false, "they should not bee equal")
assert.NotEqual(t, newOptions.NoGrep, false, "they should not bee equal")
assert.NotEqual(t, newOptions.RemotePayloads, "", "they should not bee equal")
assert.NotEqual(t, newOptions.RemoteWordlists, "", "they should not bee equal")
assert.NotEqual(t, newOptions.OnlyDiscovery, false, "they should not bee equal")
assert.NotEqual(t, newOptions.FollowRedirect, false, "they should not bee equal")
assert.NotEqual(t, newOptions.Mining, false, "they should not bee equal")
assert.NotEqual(t, newOptions.FindingDOM, false, "they should not bee equal")
assert.NotEqual(t, newOptions.Cookie, "", "they should not be equal")
assert.NotEqual(t, newOptions.UniqParam, []string{}, "they should not be equal")
assert.NotEqual(t, newOptions.BlindURL, "", "they should not be equal")
assert.NotEqual(t, newOptions.CustomAlertValue, "", "they should not be equal")
assert.NotEqual(t, newOptions.CustomAlertType, "", "they should not be equal")
assert.NotEqual(t, newOptions.Data, "", "they should not be equal")
assert.NotEqual(t, newOptions.UserAgent, "", "they should not be equal")
assert.NotEqual(t, newOptions.ProxyAddress, "", "they should not be equal")
assert.NotEqual(t, newOptions.Grep, "", "they should not be equal")
assert.NotEqual(t, newOptions.IgnoreReturn, "", "they should not be equal")
assert.NotEqual(t, newOptions.IgnoreParams, []string{}, "they should not be equal")
assert.NotEqual(t, newOptions.Trigger, "", "they should not be equal")
assert.NotEqual(t, newOptions.Timeout, 10, "they should not be equal")
assert.NotEqual(t, newOptions.Concurrence, 100, "they should not be equal")
assert.NotEqual(t, newOptions.Delay, 0, "they should not be equal")
assert.NotEqual(t, newOptions.NoBAV, false, "they should not be equal")
assert.NotEqual(t, newOptions.NoGrep, false, "they should not be equal")
assert.NotEqual(t, newOptions.RemotePayloads, "", "they should not be equal")
assert.NotEqual(t, newOptions.RemoteWordlists, "", "they should not be equal")
assert.NotEqual(t, newOptions.OnlyDiscovery, false, "they should not be equal")
assert.NotEqual(t, newOptions.FollowRedirect, false, "they should not be equal")
assert.NotEqual(t, newOptions.Mining, false, "they should not be equal")
assert.NotEqual(t, newOptions.FindingDOM, false, "they should not be equal")
assert.Equal(t, newOptions.Method, "GET", "Method should be GET")
assert.Equal(t, newOptions.Format, "plain", "Format should be plain")
assert.Equal(t, newOptions.FoundActionShell, "bash", "FoundActionShell should be bash")
assert.Equal(t, newOptions.Timeout, 5, "Timeout should be 5")
assert.Equal(t, newOptions.Concurrence, 10, "Concurrence should be 10")
assert.Equal(t, newOptions.Delay, 2, "Delay should be 2")
assert.Equal(t, newOptions.NoBAV, true, "NoBAV should be true")
assert.Equal(t, newOptions.NoGrep, true, "NoGrep should be true")
assert.Equal(t, newOptions.RemotePayloads, "portswigger", "RemotePayloads should be portswigger")
assert.Equal(t, newOptions.RemoteWordlists, "burp", "RemoteWordlists should be burp")
assert.Equal(t, newOptions.PoCType, "curl", "PoCType should be curl")
assert.Equal(t, newOptions.UseBAV, false, "UseBAV should be false")
assert.Equal(t, newOptions.CustomPayloadFile, "payloads.txt", "CustomPayloadFile should be payloads.txt")
assert.Equal(t, newOptions.OutputFile, "output.txt", "OutputFile should be output.txt")
assert.Equal(t, newOptions.FoundAction, "notify", "FoundAction should be notify")
assert.Equal(t, newOptions.OnlyCustomPayload, true, "OnlyCustomPayload should be true")
assert.Equal(t, newOptions.UseHeadless, false, "UseHeadless should be false")
assert.Equal(t, newOptions.UseDeepDXSS, true, "UseDeepDXSS should be true")
assert.Equal(t, newOptions.WAFEvasion, true, "WAFEvasion should be true")
assert.Equal(t, newOptions.Sequence, 1, "Sequence should be 1")
assert.Equal(t, newOptions.UseBAV, false, "UseBAV should be true")
}

func TestNewScan(t *testing.T) {
opt := dalfox.Options{}
target := dalfox.Target{
URL: "https://www.hahwul.com",
Method: "GET",
Options: opt,
}

result, err := dalfox.NewScan(target)
assert.NoError(t, err, "Error should be nil")
assert.NotNil(t, result, "Result should not be nil")
assert.NotZero(t, result.Duration, "Duration should not be zero")
assert.NotZero(t, result.StartTime, "StartTime should not be zero")
assert.NotZero(t, result.EndTime, "EndTime should not be zero")
}
Loading