Skip to content

Commit

Permalink
fix formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
wolyslager committed Dec 14, 2023
1 parent 848a5f8 commit 486382f
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -750,7 +750,7 @@ def generate_abuse_campaign_incidents(client, campaigns):
for campaign in campaigns:
campaign_details = client.get_details_of_an_abuse_mailbox_campaign_request(campaign["campaignId"])
incident = {"dbotMirrorId": str(campaign["campaignId"]), "name": "Abuse Campaign",
"occurred": campaign_details["firstReported"],'details': "Abuse Campaign",
"occurred": campaign_details["firstReported"], 'details': "Abuse Campaign",
"rawJSON": json.dumps(campaign_details) if campaign_details else {}}
incidents.append(incident)
return incidents
Expand All @@ -761,7 +761,7 @@ def generate_account_takeover_cases_incidents(client, cases):
for case in cases:
case_details = client.get_details_of_an_abnormal_case_request(case["caseId"])
incident = {"dbotMirrorId": str(case["caseId"]), "name": "Account Takeover Case",
"occurred": case_details["firstObserved"],'details': case['description'],
"occurred": case_details["firstObserved"], 'details': case['description'],
"rawJSON": json.dumps(case_details) if case_details else {}}
incidents.append(incident)
return incidents
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -501,6 +501,7 @@ def test_fetch_cases_incidents(mocker, mock_get_a_list_of_abnormal_cases_identif
fetch_threats=False
)
assert len(incidents) == 1


def test_fetch_abuse_campaign_incidents(mocker, mock_get_a_list_of_campaigns_submitted_to_abuse_mailbox_request):
client = mock_client(mocker, util_load_json('test_data/test_get_details_of_abuse_campaign.json'))
Expand All @@ -514,4 +515,5 @@ def test_fetch_abuse_campaign_incidents(mocker, mock_get_a_list_of_campaigns_sub
fetch_abuse_campaigns=True,
fetch_threats=False
)
assert len(incidents) == 1
assert len(incidents) == 1

0 comments on commit 486382f

Please sign in to comment.