Skip to content

Commit 49823d4

Browse files
committed
Added/fixed unit tests
1 parent c9c89da commit 49823d4

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

spec/msf/ui/console/module_argument_parsing_spec.rb

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,18 @@
6565
expect(subject.send(opts[:method_name], ['-o', 'RHOSTS=192.168.172.1', '-o', 'RPORT=1337', '-o', 'rhosts=192.168.172.2'])).to include(expected_result)
6666
end
6767

68+
it 'allows setting action inline' do
69+
expected_result = {
70+
datastore_options: {
71+
'RHOSTS' => '192.168.172.1',
72+
'RPORT' => '1337',
73+
}
74+
}
75+
expected_result[:action] = 'action-name' unless opts[:method_name] == 'parse_check_opts'
76+
result = subject.send(opts[:method_name], ['RHOSTS=192.168.172.1', 'RPORT=1337', 'action=action-name'])
77+
expect(result).to include(expected_result)
78+
end
79+
6880
it 'parses the option str directly into its components' do
6981
expected_result = {
7082
datastore_options: {
@@ -344,6 +356,7 @@ def cmd_exploit_help
344356
it 'handles no arguments being supplied' do
345357
args = []
346358
expected_result = {
359+
action: nil,
347360
jobify: false,
348361
quiet: false,
349362
datastore_options: {}
@@ -377,6 +390,7 @@ def cmd_exploit_help
377390
'example.com'
378391
]
379392
expected_result = {
393+
action: nil,
380394
jobify: false,
381395
quiet: true,
382396
datastore_options: {

0 commit comments

Comments
 (0)