fix cmd_exec tests on python/windows

rapid7 · Jun 2, 2019 · cd460aa · cd460aa
1 parent d335415
commit cd460aa
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/test/modules/post/test/cmd_exec.rb b/test/modules/post/test/cmd_exec.rb
@@ -71,7 +71,10 @@ def test_cmd_exec_quotes
 
     it "should return the result of echo with single quotes" do
       test_string = Rex::Text.rand_text_alpha(4)
-      if session.platform.eql? 'windows'
+      if session.platform.eql? 'windows' and session.arch == ARCH_PYTHON
+        output = cmd_exec("cmd.exe", "/c echo \"#{test_string}\"")
+        output == test_string
+      elsif session.platform.eql? 'windows'
         output = cmd_exec("cmd.exe", "/c echo '#{test_string}'")
         output == "'" + test_string + "'"
       else
@@ -82,7 +85,10 @@ def test_cmd_exec_quotes
 
     it "should return the result of echo with double quotes" do
       test_string = Rex::Text.rand_text_alpha(4)
-      if session.platform.eql? 'windows'
+      if session.platform.eql? 'windows' and session.arch == ARCH_PYTHON
+        output = cmd_exec("cmd.exe", "/c echo \"#{test_string}\"")
+        output == test_string
+      elsif session.platform.eql? 'windows'
         output = cmd_exec("cmd.exe", "/c echo \"#{test_string}\"")
         output == "\"" + test_string + "\""
       else