def test_write_exploit_source(self): test_exploit = LinuxExploit() test_exploit.write_exploit_source() print("randomstring {}".format(test_exploit.source_c_path)) self.assertTrue(os.path.exists(test_exploit.source_c_path)) # remove it so it's not in the project if os.path.exists(test_exploit.source_c_path): os.remove(test_exploit.source_c_path)
def test_compile_exploit_source(self): test_exploit = LinuxExploit() test_exploit.write_exploit_source() test_exploit.exploit_compile() self.assertTrue(os.path.exists(test_exploit.compilation_path)) # remove files so they're not in the project if os.path.exists(test_exploit.source_c_path): os.remove(test_exploit.source_c_path) if os.path.exists(test_exploit.compilation_path): os.remove(test_exploit.compilation_path)