def test_merge_observable(self):
merged_obs = {
'id': self.draft_obs_id0,
'title': 'abc.txt 123123123MD5',
'file_name': "abc.txt",
'objectType': 'File',
'hashes': [{
'hash_type': 'MD5',
'hash_value': '123123123'
}]
}
self.mock_draft_load.return_value = self.draft_ind_with_draft_obs
self.mock_draft_list.return_value = [self.draft_ind_with_draft_obs]
self.mock_request.body = json.dumps({
"id":
self.draft_ind_id,
"ids": [self.draft_obs_id0, self.draft_obs_id1]
})
extract_visualiser_merge_observables(self.mock_request)
self.mock_draft_upsert.assert_called_with('ind', {
'observables': [merged_obs],
'id': 'indicator:123'
}, self.mock_request.user)
def test_try_merge_non_draft_observable(self):
self.mock_draft_load.return_value = self.draft_ind_with_obs
self.mock_draft_list.return_value = [self.draft_ind_with_obs]
self.mock_request.body = json.dumps({"id": self.draft_ind_id, "ids": [self.obs_id0]})
extract_visualiser_merge_observables(self.mock_request)
self.mock_draft_upsert.assert_not_called()
def test_try_merge_non_file_observable(self):
draft_ind = {'draft': {'id': self.draft_ind_id}, 'observables': [self.obs0_NotFile, self.obs1]}
self.mock_draft_load.return_value = draft_ind
self.mock_draft_list.return_value = [draft_ind]
self.mock_request.body = json.dumps({"id": self.draft_ind_id, "ids": [self.obs_id0]})
extract_visualiser_merge_observables(self.mock_request)
self.mock_draft_upsert.assert_not_called()