From dba3388567818cb5c60be9d406d3e1cccc84b3ea Mon Sep 17 00:00:00 2001
From: salmonik <raisersoft@proton.me>
Date: Wed, 29 Apr 2026 14:22:03 +0200
Subject: Refactor database connections and add integration tests

- Use context managers for sqlite3 connections in cron, vclogs, and database utils
- Add URL helper utility for yt-dlp with clean_url_for_ytdlp function
- Add slow integration test markers and new integration test files
- Fix test imports to use conftest directly instead of tests.conftest
- Improve downloader error handling and use COPYPARTY_URL_UPLOAD env var
- Update pytest.ini with slow test markers and warning filters
- Document slow test usage in AGENTS.md
---
 tests/cogs/test_vclogs_integration.py | 86 +++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100644 tests/cogs/test_vclogs_integration.py

(limited to 'tests/cogs/test_vclogs_integration.py')

diff --git a/tests/cogs/test_vclogs_integration.py b/tests/cogs/test_vclogs_integration.py
new file mode 100644
index 0000000..9391876
--- /dev/null
+++ b/tests/cogs/test_vclogs_integration.py
@@ -0,0 +1,86 @@
+import pytest
+import asyncio
+from unittest.mock import MagicMock, AsyncMock, patch
+
+
+@pytest.fixture
+def vclogs_cog(bot):
+    from cogs.vclogs import VcLogs
+    cog = VcLogs(bot)
+    return cog
+
+
+@pytest.mark.slow
+@pytest.mark.asyncio
+async def test_full_voice_session_tracking(vclogs_cog, monkeypatch):
+    """Test complete voice session with AFK transitions using real time sleeps."""
+    cog = vclogs_cog
+
+    # Set up environment
+    monkeypatch.setenv("STATUS_CHANNEL_ID", "123456")
+
+    member = MagicMock()
+    member.id = 123
+    member.guild.id = 456
+    member.bot = False
+    member.display_name = "TestUser"
+    member.guild.get_member.return_value = member
+
+    # Create a mock log channel
+    log_channel = MagicMock()
+    log_channel.send = AsyncMock(return_value=MagicMock())
+
+    # Mock bot.get_channel to return our log channel
+    with patch.object(vclogs_cog.bot, 'get_channel', return_value=log_channel):
+        # Join voice
+        before = MagicMock()
+        before.channel = None
+        after_join = MagicMock()
+        after_join.channel = MagicMock()
+        after_join.channel.id = 789
+        after_join.channel.name = "Test Voice"
+        after_join.self_deaf = False
+        after_join.self_mute = False
+        after_join.deaf = False
+        after_join.mute = False
+        await cog.on_voice_state_update(member, before, after_join)
+
+        await asyncio.sleep(2)  # Simulate 2s active
+
+        # Go AFK (self_deaf)
+        before_afk = after_join
+        after_afk = MagicMock()
+        after_afk.channel = before_afk.channel
+        after_afk.self_deaf = True
+        after_afk.self_mute = False
+        after_afk.deaf = True
+        after_afk.mute = False
+        await cog.on_voice_state_update(member, before_afk, after_afk)
+
+        await asyncio.sleep(3)  # Simulate 3s AFK
+
+        # Return from AFK
+        before_back = after_afk
+        after_back = MagicMock()
+        after_back.channel = before_back.channel
+        after_back.self_deaf = False
+        after_back.self_mute = False
+        after_back.deaf = False
+        after_back.mute = False
+        await cog.on_voice_state_update(member, before_back, after_back)
+
+        await asyncio.sleep(1)  # Simulate 1s active
+
+        # Leave voice
+        before_leave = after_back
+        after_leave = MagicMock()
+        after_leave.channel = None
+        after_leave.self_deaf = False
+        after_leave.self_mute = False
+        await cog.on_voice_state_update(member, before_leave, after_leave)
+
+    # Verify activity was saved (check DB)
+    from cogs.vclogs import load_activity_data
+    activity_data = load_activity_data()
+    # After leaving, data should be deleted, so check the log channel was called
+    assert log_channel.send.called
-- 
cgit v1.2.3