From fe186fec44e0cedfb7f95ffd82bf08b15ac1cb60 Mon Sep 17 00:00:00 2001
From: rasarkar <105652044+rasarkar@users.noreply.github.com>
Date: Thu, 28 Mar 2024 13:30:14 -0400
Subject: [PATCH] CONCD-749 don't hardcode user ID (#2328)

---
 concordia/tests/test_views.py | 20 +++++++++-----------
 concordia/utils.py            |  2 +-
 2 files changed, 10 insertions(+), 12 deletions(-)

diff --git a/concordia/tests/test_views.py b/concordia/tests/test_views.py
index 4f9b616de..484b4e99e 100644
--- a/concordia/tests/test_views.py
+++ b/concordia/tests/test_views.py
@@ -494,10 +494,8 @@ def _asset_reservation_test_payload(self, user_id, anonymous=False):
         expected_update_queries = 2
         if not anonymous and settings.SESSION_ENGINE.endswith("db"):
             expected_update_queries += 1
-        expected_acquire_queries = expected_update_queries
-        if not anonymous:
-            # + 1 get user ID from request
-            expected_acquire_queries += 1
+        # + 1 get user ID from request
+        expected_acquire_queries = expected_update_queries + 1
 
         with self.assertNumQueries(expected_acquire_queries):
             resp = self.client.post(reverse("reserve-asset", args=(asset.pk,)))
@@ -549,9 +547,9 @@ def test_asset_reservation_competition(self):
         # We'll reserve the test asset as the anonymous user and then attempt
         # to edit it after logging in
 
-        # 3 queries =
-        # 1 expiry + 1 acquire
-        with self.assertNumQueries(2):
+        # 4 queries =
+        # 1 expiry + 1 acquire + 2 get user ID from request
+        with self.assertNumQueries(4):
             resp = self.client.post(reverse("reserve-asset", args=(asset.pk,)))
         self.assertEqual(200, resp.status_code)
         self.assertEqual(1, AssetTranscriptionReservation.objects.count())
@@ -655,11 +653,11 @@ def test_asset_reservation_tombstone(self):
 
         self.client.logout()
 
-        # 1 session check + 1 reservation check + 1 acquire
+        # 1 reservation check + 1 acquire + 2 get user ID from request
+        expected_queries = 4
         if settings.SESSION_ENGINE.endswith("db"):
-            expected_queries = 3
-        else:
-            expected_queries = 2
+            # + 1 session check
+            expected_queries += 1
 
         with self.assertNumQueries(expected_queries):
             resp = self.client.post(reverse("reserve-asset", args=(asset.pk,)))
diff --git a/concordia/utils.py b/concordia/utils.py
index 9a11c3c9c..c2bcb968a 100644
--- a/concordia/utils.py
+++ b/concordia/utils.py
@@ -34,7 +34,7 @@ def get_or_create_reservation_token(request):
                 # We should probably call get_anonymous_user,
                 # but I'm going to avoid doing so since it would
                 # incur yet *another* database query
-                uid = 8
+                uid = get_anonymous_user().id
             request.session["reservation_token"] += str(uid).zfill(6)
     return request.session["reservation_token"]