Skip to content

Commit

Permalink
Merge pull request #83 from uktrade/LTD-1480-Ignore-licenceUsage-when…
Browse files Browse the repository at this point in the history
…-completion

LTD-1480: Don't send licenceUsage transactions with completion date to LITE-API
  • Loading branch information
saruniitr authored Nov 22, 2021
2 parents 08ef8cd + 507fd26 commit f5cb5f9
Show file tree
Hide file tree
Showing 2 changed files with 105 additions and 1 deletion.
7 changes: 7 additions & 0 deletions mail/libraries/usage_data_decomposition.py
Original file line number Diff line number Diff line change
Expand Up @@ -112,6 +112,13 @@ def build_json_payload_from_data_blocks(data_blocks: list) -> dict:

payload.append(licence_payload)

"""
Filter blocks that has a completion date because this is an additional transaction
that is sent when all the allowed quantity of all items on that licence are used.
Sending it again results in double counting of the recorded usage
"""
payload = [item for item in payload if not item["completion_date"]]

return {"licences": payload}


Expand Down
99 changes: 98 additions & 1 deletion mail/tests/test_send_licence_usage_figures_to_lite_api.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
from unittest import mock
from uuid import uuid4

from rest_framework.status import HTTP_400_BAD_REQUEST, HTTP_207_MULTI_STATUS, HTTP_208_ALREADY_REPORTED

from conf.settings import LITE_API_URL, HAWK_LITE_HMRC_INTEGRATION_CREDENTIALS, LITE_API_REQUEST_TIMEOUT, MAX_ATTEMPTS
from mail.models import Mail, UsageData, LicencePayload
from mail.models import GoodIdMapping, LicenceIdMapping, Mail, UsageData, LicencePayload
from mail.tasks import send_licence_usage_figures_to_lite_api, schedule_max_tried_task_as_new_task
from mail.tests.libraries.client import LiteHMRCTestClient

Expand Down Expand Up @@ -175,3 +176,99 @@ def test_schedule_new_task(self, send_licence_usage_figures):
schedule_max_tried_task_as_new_task(str(self.usage_data.id))

send_licence_usage_figures.assert_called_with(str(self.usage_data.id), schedule=mock.ANY)

@mock.patch("mail.tasks.put")
def test_licence_usage_ignore_licence_completion(self, put_request):
"""
Test that ensures that licenceUsage transaction that has a completion date
value are not reported to LITE-API because this is an additional transaction
that is sent when all the allowed quantity of all items on that licence are used.
Sending it again results in double counting of the recorded usage.
"""
licence = LicenceIdMapping.objects.create(
lite_id="5678d9b2-09a9-4e86-840f-236d186e1234", reference="GBSIEL/2020/0000025/P"
)
good_mappings = [
GoodIdMapping.objects.create(
line_number=(i + 1), lite_id=str(uuid4()), licence_reference="GBSIEL/2020/0000025/P"
)
for i in range(3)
]

mail = Mail.objects.create(
edi_data=(
"1\\fileHeader\\CHIEF\\SPIRE\\usageData\\201901130300\\49543\\\n"
"2\\licenceUsage\\LU04148/00001\\insert\\GBSIEL/2020/0000025/P\\O\\\n"
"3\\line\\1\\2\\0\\\n"
"4\\usage\\O\\9GB000001328000-PE112345\\R\\20190112\\2\\0\\\\000262\\\\\\\\\n"
"5\\end\\line\\5\n"
"6\\line\\2\\5\\0\\\n"
"7\\usage\\O\\9GB000001328000-PE112345\\R\\20190112\\5\\0\\\\000262\\\\\\\\\n"
"8\\end\\line\\5\n"
"9\\line\\3\\10\\0\\\n"
"10\\usage\\O\\9GB000001328000-PE112345\\R\\20190112\\10\\0\\\\000262\\\\\\\\\n"
"11\\end\\line\\5\n"
"12\\end\\licenceUsage\\10\n"
"13\\licenceUsage\\LU04148/00002\\insert\\GBSIEL/2020/0000025/P\\C\\20211110\n"
"14\\line\\1\\2\\0\\\n"
"15\\end\\line\\2\n"
"16\\line\\2\\5\\0\\\n"
"17\\end\\line\\2\n"
"18\\line\\3\\10\\0\\\n"
"19\\end\\line\\2\n"
"20\\end\\licenceUsage\\8\n"
"21\\fileTrailer\\2"
)
)
usage_data = UsageData.objects.create(
mail=mail, licence_ids='["GBSIEL/2020/0000025/P"]', hmrc_run_number=0, spire_run_number=0,
)

expected_payload = {
"usage_data_id": str(usage_data.id),
"licences": [
{
"id": str(licence.lite_id),
"goods": [
{"id": str(good_mappings[0].lite_id), "usage": "2", "value": "0", "currency": ""},
{"id": str(good_mappings[1].lite_id), "usage": "5", "value": "0", "currency": ""},
{"id": str(good_mappings[2].lite_id), "usage": "10", "value": "0", "currency": ""},
],
"action": "open",
"completion_date": "",
}
],
}

put_request.return_value = MockResponse(
json={
"licences": {
"accepted": [
{
"id": str(licence.lite_id),
"goods": [
{"id": str(good_mappings[0].lite_id), "usage": "2"},
{"id": str(good_mappings[1].lite_id), "usage": "5"},
{"id": str(good_mappings[2].lite_id), "usage": "10"},
],
}
],
"rejected": [],
}
},
status_code=HTTP_207_MULTI_STATUS,
)

send_licence_usage_figures_to_lite_api.now(str(usage_data.id))

usage_data.refresh_from_db()
put_request.assert_called_with(
f"{LITE_API_URL}/licences/hmrc-integration/",
expected_payload,
hawk_credentials=HAWK_LITE_HMRC_INTEGRATION_CREDENTIALS,
timeout=LITE_API_REQUEST_TIMEOUT,
)
usage_data.refresh_from_db()
self.assertIsNotNone(usage_data.lite_sent_at)
self.assertEqual(usage_data.lite_accepted_licences, ["GBSIEL/2020/0000025/P"])
self.assertEqual(usage_data.lite_rejected_licences, [])

0 comments on commit f5cb5f9

Please sign in to comment.