first init

tests/test_db.py

@@ -0,0 +1,71 @@
+from pathlib import Path
+
+from vpn_bot.db import Database
+
+
+def test_invite_is_stored_before_activation_and_can_be_loaded(tmp_path: Path):
+    db = Database(tmp_path / "bot.sqlite3")
+    db.init()
+
+    db.create_invite(token="abc", admin_id=42, created_at=1000)
+    user = db.get_by_token("abc")
+
+    assert user is not None
+    assert user["invite_token"] == "abc"
+    assert user["invite_created_by_admin_id"] == 42
+    assert user["tg_user_id"] is None
+
+
+def test_activation_stores_telegram_identity_and_marzban_username(tmp_path: Path):
+    db = Database(tmp_path / "bot.sqlite3")
+    db.init()
+    db.create_invite(token="abc", admin_id=42, created_at=1000)
+
+    db.activate_invite(
+        token="abc",
+        tg_user_id=856850518,
+        tg_username="krosh",
+        tg_first_name="Dmitry",
+        tg_last_name="Gudov",
+        tg_phone=None,
+        marzban_username="tg856850518",
+        activated_at=1100,
+    )
+
+    user = db.get_by_tg_user_id(856850518)
+    assert user is not None
+    assert user["invite_token"] == "abc"
+    assert user["marzban_username"] == "tg856850518"
+    assert user["tg_first_name"] == "Dmitry"
+
+
+def test_used_invite_cannot_be_activated_twice(tmp_path: Path):
+    db = Database(tmp_path / "bot.sqlite3")
+    db.init()
+    db.create_invite(token="abc", admin_id=42, created_at=1000)
+    db.activate_invite(
+        token="abc",
+        tg_user_id=1,
+        tg_username=None,
+        tg_first_name="One",
+        tg_last_name=None,
+        tg_phone=None,
+        marzban_username="tg1",
+        activated_at=1100,
+    )
+
+    try:
+        db.activate_invite(
+            token="abc",
+            tg_user_id=2,
+            tg_username=None,
+            tg_first_name="Two",
+            tg_last_name=None,
+            tg_phone=None,
+            marzban_username="tg2",
+            activated_at=1200,
+        )
+    except ValueError as exc:
+        assert "already used" in str(exc)
+    else:
+        raise AssertionError("second activation should fail")