3
0
Prechádzať zdrojové kódy

fix ftp mock params
add file.close

Jens Keim 6 rokov pred
rodič
commit
f00ff21b15
2 zmenil súbory, kde vykonal 2 pridanie a 1 odobranie
  1. 1 0
      code/Test/Lib.py
  2. 1 1
      code/Test/test_FTPWinaXeExploit.py

+ 1 - 0
code/Test/Lib.py

@@ -26,6 +26,7 @@ def get_sha256(file):
             if not data:
                 break
             sha.update(data)
+    f.close()
     return sha.hexdigest()
 
 

+ 1 - 1
code/Test/test_FTPWinaXeExploit.py

@@ -17,7 +17,7 @@ class UnitTestFTPWinaXeExploit(GenericTest):
 
     @mock.patch('ID2TLib.Utility.get_rnd_bytes', side_effect=get_bytes)
     @mock.patch('ID2TLib.Utility.get_rnd_x86_nop', side_effect=get_x86_nop)
-    def test_one_attacker_ftp(self):
+    def test_one_attacker_ftp(self, mock_get_rnd_bytes, mock_get_rnd_x86_nop):
         self.generic_test([['FTPWinaXeExploit']], sha_one_attacker_ftp)