From c24cf7a838e684f9dcd2285866985c8aee3d9ca3 Mon Sep 17 00:00:00 2001 From: akhdanre Date: Fri, 23 May 2025 15:07:16 +0700 Subject: [PATCH] fix: minor --- app/controllers/socket_conroller.py | 2 -- test/repository/test_user_repository.py | 46 +++++++++++++++++++++++++ 2 files changed, 46 insertions(+), 2 deletions(-) diff --git a/app/controllers/socket_conroller.py b/app/controllers/socket_conroller.py index 88ffc48..db2d85c 100644 --- a/app/controllers/socket_conroller.py +++ b/app/controllers/socket_conroller.py @@ -2,8 +2,6 @@ from flask_socketio import SocketIO, emit, join_room, leave_room from flask import request from app.services import SessionService import threading -import json -from redis import Redis class SocketController: diff --git a/test/repository/test_user_repository.py b/test/repository/test_user_repository.py index 82f8414..03f0a2f 100644 --- a/test/repository/test_user_repository.py +++ b/test/repository/test_user_repository.py @@ -153,3 +153,49 @@ def test_delete_user(user_repository): result = user_repository.delete_user(user_id) assert result is True + + +def test_get_user_by_id_not_found(user_repository): + user_repository.collection.find_one.return_value = None + + result = user_repository.get_user_by_id(str(ObjectId())) + + assert result is None + + +def test_get_by_google_id_not_found(user_repository): + user_repository.collection.find_one.return_value = None + + result = user_repository.get_by_google_id("nonexistent-google-id") + + assert result is None + + +def test_update_user_not_found(user_repository): + user_id = str(ObjectId()) + mock_result = MagicMock(modified_count=0) + user_repository.collection.update_one.return_value = mock_result + + result = user_repository.update_user(user_id, {"name": "Nonexistent User"}) + + assert result is False + + +def test_update_user_field_not_found(user_repository): + user_id = str(ObjectId()) + mock_result = MagicMock(modified_count=0) + user_repository.collection.update_one.return_value = mock_result + + result = user_repository.update_user_field(user_id, "phone", "000000000") + + assert result is False + + +def test_delete_user_not_found(user_repository): + user_id = str(ObjectId()) + mock_result = MagicMock(deleted_count=0) + user_repository.collection.delete_one.return_value = mock_result + + result = user_repository.delete_user(user_id) + + assert result is False