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()