first implementation

backend/tests/test_counter.py

@@ -1,7 +1,11 @@
+"""Tests for counter endpoints.
+
+Note: Registration now requires an invite code.
+"""
 import pytest
 
 from auth import COOKIE_NAME
-from tests.helpers import unique_email
+from tests.helpers import unique_email, create_invite_for_registration
 
 
 # Protected endpoint tests - without auth
@@ -34,9 +38,16 @@ async def test_increment_counter_invalid_cookie(client_factory):
 # Authenticated counter tests
 @pytest.mark.asyncio
 async def test_get_counter_authenticated(client_factory):
+    async with client_factory.get_db_session() as db:
+        invite_code = await create_invite_for_registration(db, unique_email("gf"))
+    
     reg = await client_factory.post(
         "/api/auth/register",
-        json={"email": unique_email(), "password": "testpass123"},
+        json={
+            "email": unique_email(),
+            "password": "testpass123",
+            "invite_identifier": invite_code,
+        },
     )
     cookies = dict(reg.cookies)
     
@@ -49,9 +60,16 @@ async def test_get_counter_authenticated(client_factory):
 
 @pytest.mark.asyncio
 async def test_increment_counter(client_factory):
+    async with client_factory.get_db_session() as db:
+        invite_code = await create_invite_for_registration(db, unique_email("gf"))
+    
     reg = await client_factory.post(
         "/api/auth/register",
-        json={"email": unique_email(), "password": "testpass123"},
+        json={
+            "email": unique_email(),
+            "password": "testpass123",
+            "invite_identifier": invite_code,
+        },
     )
     cookies = dict(reg.cookies)
     
@@ -68,9 +86,16 @@ async def test_increment_counter(client_factory):
 
 @pytest.mark.asyncio
 async def test_increment_counter_multiple(client_factory):
+    async with client_factory.get_db_session() as db:
+        invite_code = await create_invite_for_registration(db, unique_email("gf"))
+    
     reg = await client_factory.post(
         "/api/auth/register",
-        json={"email": unique_email(), "password": "testpass123"},
+        json={
+            "email": unique_email(),
+            "password": "testpass123",
+            "invite_identifier": invite_code,
+        },
     )
     cookies = dict(reg.cookies)
     
@@ -89,9 +114,16 @@ async def test_increment_counter_multiple(client_factory):
 
 @pytest.mark.asyncio
 async def test_get_counter_after_increment(client_factory):
+    async with client_factory.get_db_session() as db:
+        invite_code = await create_invite_for_registration(db, unique_email("gf"))
+    
     reg = await client_factory.post(
         "/api/auth/register",
-        json={"email": unique_email(), "password": "testpass123"},
+        json={
+            "email": unique_email(),
+            "password": "testpass123",
+            "invite_identifier": invite_code,
+        },
     )
     cookies = dict(reg.cookies)
     
@@ -109,10 +141,19 @@ async def test_get_counter_after_increment(client_factory):
 # Counter is shared between users
 @pytest.mark.asyncio
 async def test_counter_shared_between_users(client_factory):
+    # Create invites for two users
+    async with client_factory.get_db_session() as db:
+        invite1 = await create_invite_for_registration(db, unique_email("gf1"))
+        invite2 = await create_invite_for_registration(db, unique_email("gf2"))
+    
     # Create first user
     reg1 = await client_factory.post(
         "/api/auth/register",
-        json={"email": unique_email("share1"), "password": "testpass123"},
+        json={
+            "email": unique_email("share1"),
+            "password": "testpass123",
+            "invite_identifier": invite1,
+        },
     )
     cookies1 = dict(reg1.cookies)
     
@@ -127,7 +168,11 @@ async def test_counter_shared_between_users(client_factory):
     # Create second user - should see the increments
     reg2 = await client_factory.post(
         "/api/auth/register",
-        json={"email": unique_email("share2"), "password": "testpass123"},
+        json={
+            "email": unique_email("share2"),
+            "password": "testpass123",
+            "invite_identifier": invite2,
+        },
     )
     cookies2 = dict(reg2.cookies)