def _random_data(self): self.authkey = { 'ident': random_string(), 'secret': random_string(), 'pubchans': random_string(), 'subchans': random_string(), 'owner': self.user }
def random_file(): app_path = os.path.dirname(__file__) tests_folder = os.path.join(app_path, 'tests') if not os.path.exists(tests_folder): os.mkdir(tests_folder) path = os.path.join(app_path, 'tests', random_string(10)) fp = open(path, 'wb') fp.write(random_string(10)) fp.close() return path
def test_can_update(self): self._upload(self.file_path, self.file_name) malware = Malware.objects.get(name=self.file_name) new_file_name = random_string(8) response = self.client.post( reverse_lazy('malware.update', args=[malware.sha256]), { 'name': new_file_name, 'type': malware.type, 'size': malware.size, 'crc32': malware.crc32, 'md5': malware.md5, 'sha1': malware.sha1, 'sha256': malware.sha256, 'sha512': malware.sha512, 'ssdeep': malware.ssdeep, } ) try: Malware.objects.get(name=new_file_name) except Malware.DoesNotExist: updated = False else: updated = True self.assertTrue(updated)
def _upload(self, file_path, file_name): with open(file_path, 'rb') as fp: response = self.client.post(reverse_lazy('malware.upload'), { 'name': file_name, 'malware': fp, 'description': random_string(), }, follow=True) return response
def _upload(self, file_path, file_name): with open(file_path, 'rb') as fp: response = self.client.post( reverse_lazy('malware.upload'), { 'name': file_name, 'malware': fp, 'description': random_string(), }, follow=True ) return response
def _random_data(self): self.channel = { "default": True, "name": random_string(), "host": random_string(), "port": random.randint(1024, 65535), "ident": random_string(), "secret": random_string(), "pubchans": random_string(), "subchans": random_string(), "owner": self.user, }
def _random_data(self): self.channel = { 'default': True, 'name': random_string(), 'host': random_string(), 'port': random.randint(1024, 65535), 'ident': random_string(), 'secret': random_string(), 'pubchans': random_string(), 'subchans': random_string(), 'owner': self.user }
def test_can_update(self): self._upload(self.file_path, self.file_name) malware = Malware.objects.get(name=self.file_name) new_file_name = random_string(8) response = self.client.post( reverse_lazy('malware.update', args=[malware.sha256]), { 'name': new_file_name, 'type': malware.type, 'size': malware.size, 'crc32': malware.crc32, 'md5': malware.md5, 'sha1': malware.sha1, 'sha256': malware.sha256, 'sha512': malware.sha512, 'ssdeep': malware.ssdeep, }) try: Malware.objects.get(name=new_file_name) except Malware.DoesNotExist: updated = False else: updated = True self.assertTrue(updated)
def setUp(self): super(MalwareTestCase, self).setUp() self.file_path = random_file() self.file_name = random_string(8)