Fix: Getting next ASN when no documents have an ASN (#5431)

* Fixes the next ASN logic to account for no ASNs yet being assigned

* Updates so the ASN will start at 1

* Do the same calculation without the branch
This commit is contained in:
Trenton H
2024-01-16 15:08:37 -08:00
committed by GitHub
parent e16645b146
commit 51dd95be3d
2 changed files with 34 additions and 9 deletions

View File

@@ -2051,6 +2051,35 @@ class TestDocumentApi(DirectoriesMixin, DocumentConsumeDelayMixin, APITestCase):
self.assertEqual(resp.status_code, status.HTTP_200_OK)
self.assertEqual(resp.content, b"1000")
def test_next_asn_no_documents_with_asn(self):
"""
GIVEN:
- Existing document, but with no ASN assugned
WHEN:
- API request to get next ASN
THEN:
- ASN 1 is returned
"""
user1 = User.objects.create_user(username="test1")
user1.user_permissions.add(*Permission.objects.all())
user1.save()
doc1 = Document.objects.create(
title="test",
mime_type="application/pdf",
content="this is a document 1",
checksum="1",
)
doc1.save()
self.client.force_authenticate(user1)
resp = self.client.get(
"/api/documents/next_asn/",
)
self.assertEqual(resp.status_code, status.HTTP_200_OK)
self.assertEqual(resp.content, b"1")
class TestDocumentApiV2(DirectoriesMixin, APITestCase):
def setUp(self):