File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed
spec/lib/msf/core/exploit Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -78,6 +78,7 @@ def decompress(code)
78
78
79
79
context 'when strip_whitespace is true' do
80
80
before do
81
+ subject . datastore [ 'Powershell::strip_comments' ] = false
81
82
subject . datastore [ 'Powershell::strip_whitespace' ] = true
82
83
subject . options . validate ( subject . datastore )
83
84
end
@@ -90,13 +91,16 @@ def decompress(code)
90
91
91
92
context 'when strip_whitespace is false' do
92
93
before do
94
+ subject . datastore [ 'Powershell::strip_comments' ] = false
93
95
subject . datastore [ 'Powershell::strip_whitespace' ] = false
94
96
subject . options . validate ( subject . datastore )
95
97
end
96
98
it 'shouldnt strip whitespace' do
97
99
script = File . read ( example_script )
100
+ File . open ( '/tmp/1' , 'w' ) { |f | f . write script }
98
101
compressed = subject . compress_script ( script )
99
- decompress ( compressed ) . length . should be script . length
102
+ File . open ( '/tmp/2' , 'w' ) { |f | f . write decompress ( compressed ) }
103
+ expect ( decompress ( compressed ) . length ) . to eq ( script . length )
100
104
end
101
105
end
102
106
You can’t perform that action at this time.
0 commit comments