1 files changed, 18 insertions, 13 deletions

diff --git a/tests/cogs/test_vclogs.py b/tests/cogs/test_vclogs.py
index 2003478..da41482 100644
--- a/tests/cogs/test_vclogs.py
+++ b/tests/cogs/test_vclogs.py
@@ -1,6 +1,5 @@
 import pytest
 from unittest.mock import AsyncMock, patch, MagicMock
-import json
 from tests.conftest import MockApplicationContext
 
 
@@ -13,20 +12,26 @@ def test_format_duration():
 
 
 def test_load_activity_data():
-    with patch('builtins.open', MagicMock()):
-        with patch('os.path.exists', return_value=True):
-            with patch('json.load', return_value={"123": {"active": True}}):
-                from cogs.vclogs import load_activity_data
-                data = load_activity_data()
-                assert isinstance(data, dict)
+    with patch("cogs.vclogs.get_connection") as mock_conn:
+        mock_cursor = MagicMock()
+        mock_conn.return_value.cursor.return_value = mock_cursor
+        mock_conn.return_value.close.return_value = None
+        mock_cursor.fetchall.return_value = []
+
+        from cogs.vclogs import load_activity_data
+        data = load_activity_data()
+        assert isinstance(data, dict)
 
 
 def test_save_activity_data():
-    with patch('builtins.open', MagicMock()) as mock_open:
-        with patch('json.dump'):
-            from cogs.vclogs import save_activity_data
-            save_activity_data({"123": {"active": True}})
-            mock_open.assert_called_once()
+    with patch("cogs.vclogs.get_connection") as mock_conn:
+        mock_cursor = MagicMock()
+        mock_conn.return_value.cursor.return_value = mock_cursor
+        mock_conn.return_value.commit.return_value = None
+        mock_conn.return_value.close.return_value = None
+
+        from cogs.vclogs import save_activity_data
+        save_activity_data({"123": {"active": True}})
 
 
 @pytest.fixture
@@ -80,4 +85,4 @@ async def test_handle_state_change_mute_false(vclogs_cog):
 
     tts_text, channel = await vclogs_cog.handle_state_change(mock_member, mock_channel, 'mute', False, None)
 
-    assert "zrobił jajecznice" in tts_text
+    assert "zrobił jajecznice" in tts_text
\ No newline at end of file