args = dict( (arg['name'], arg['value']) for arg in row['arguments']) buf += args['Buffer'] self.assertEqual(buf, 'hello!!1\r\n') def test_file_write(self): report = self.run_analysis(""" open('a.txt', 'wb').write('Hello World') """) self.assertTrue('dropped' in report) self.assertEqual(len(report['dropped']), 1) self.assertEqual( open(report['dropped'][0]['path']).read(), 'Hello World') if __name__ == '__main__': parser = argparse.ArgumentParser() # parser.add_argument("vmname", type=str, required=False, help="Name of the Virtual Machine to check.") args = parser.parse_args() # Clean the database. subprocess.Popen([os.path.join(CUCKOO_ROOT, 'utils', 'clean.sh')], cwd=CUCKOO_ROOT).wait() cuckoo_init(quiet=True) db = Database() unittest.main()
args = dict((arg['name'], arg['value']) for arg in row['arguments']) buf += args['Buffer'] self.assertEqual(buf, 'hello!!1\r\n') def test_file_write(self): report = self.run_analysis(""" open('a.txt', 'wb').write('Hello World') """) self.assertTrue('dropped' in report) self.assertEqual(len(report['dropped']), 1) self.assertEqual(open(report['dropped'][0]['path']).read(), 'Hello World') if __name__ == '__main__': parser = argparse.ArgumentParser() # parser.add_argument("vmname", type=str, required=False, help="Name of the Virtual Machine to check.") args = parser.parse_args() # Clean the database. subprocess.Popen([os.path.join(CUCKOO_ROOT, 'utils', 'clean.sh')], cwd=CUCKOO_ROOT).wait() cuckoo_init(quiet=True) db = Database() unittest.main()