def _random_data(self):
self.authkey = {
'ident': random_string(),
'secret': random_string(),
'pubchans': random_string(),
'subchans': random_string(),
'owner': self.user
}
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 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)
def setUp(self):
super(MalwareTestCase, self).setUp()
self.file_path = random_file()
self.file_name = random_string(8)
