Merge pull request #5825 from nyaruka/simplify-test-login

Simplify test login
nyaruka · Jan 24, 2025 · 5a16393 · 5a16393
2 parents 04785fc + 1a1406b
commit 5a16393
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/temba/tests/base.py b/temba/tests/base.py
@@ -155,10 +155,7 @@ def login(self, user, *, update_last_auth_on: bool = True, choose_org=None):
 
         # infer our org if we weren't handed one
         if not choose_org:
-            orgs = user.orgs.filter(is_active=True).order_by("-created_on")
-            if orgs.count() > 0:
-                # take the newest one
-                choose_org = orgs[0]
+            choose_org = user.orgs.filter(is_active=True).order_by("-created_on").first()
 
         if update_last_auth_on:
             user.record_auth()