MISP
/
PyMISP
mirror of https://github.com/MISP/PyMISP
2
0
Fork 0
Code Issues Releases Wiki Activity

t push origin masterMerge branch 'l3m0ntr33-master'

pull/366/head
Raphaël Vinot 2019-03-11 11:33:08 +01:00
parent c592bfcca9 ec6439ce34
commit c458a3df51
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pymisp/api.py
View File

@ -1032,6 +1032,14 @@ class PyMISP(object):
query = {"comment": comment}
return self.__query('edit/{}'.format(attribute_uuid), query, controller='attributes')
def change_disable_correlation(self, attribute_uuid, disable_correlation):
"""Change the disable_correlation flag"""
possible_values = [0, 1, False, True]
if disable_correlation not in possible_values:
raise Exception('disable_correlation can only be in {}'.format(', '.join(possible_values)))
query = {"disable_correlation": disable_correlation}
return self.__query('edit/{}'.format(attribute_uuid), query, controller='attributes')
# ##############################
# ###### Attribute update ######
# ##############################

4
tests/testlive_comprehensive.py
View File

@ -703,6 +703,10 @@ class TestComprehensive(unittest.TestCase):
self.assertEqual(attribute.comment, 'This is the modified comment')
attribute = self.user_misp_connector.change_comment(first.attributes[0].uuid, 'This is the modified comment, again')
self.assertEqual(attribute['Attribute']['comment'], 'This is the modified comment, again')
attribute = self.user_misp_connector.change_disable_correlation(first.attributes[0].uuid, True)
self.assertEqual(attribute['Attribute']['disable_correlation'], True)
attribute = self.user_misp_connector.change_disable_correlation(first.attributes[0].uuid, 0)
self.assertEqual(attribute['Attribute']['disable_correlation'], False)
finally:
# Delete event
self.admin_misp_connector.delete_event(first.id)