Upload firebase_test.py

tests/firebase_test.py

@@ -0,0 +1,96 @@
+import pytest
+import sys
+from pathlib import Path
+from unittest.mock import patch, MagicMock
+import pandas as pd
+
+sys.path.append(str(Path(__file__).resolve().parent.parent))
+from src.configs.database.firebase import (
+    read_all_users,
+    write_user_data,
+    update_user_data,
+)
+
+
+# Mock data for tests
+MOCK_USER_DATA = {
+    "uuid7-1": {
+        "id": "uuid7-1",
+        "name": "Alice",
+        "score": 90,
+        "interview_question": "What are your strengths?",
+        "job_requirements": "Team player",
+        "feedback": "Excellent response.",
+        "created_at": "2024-01-01T10:00:00",
+        "updated_at": "2024-01-01T10:00:00",
+    },
+    "uuid7-2": {
+        "id": "uuid7-2",
+        "name": "Bob",
+        "score": 85,
+        "interview_question": "Describe your work ethic.",
+        "job_requirements": "Self-starter",
+        "feedback": "Good examples provided.",
+        "created_at": "2024-01-02T10:00:00",
+        "updated_at": "2024-01-02T10:00:00",
+    },
+}
+
+
+@patch("src.configs.database.firebase.users_ref")  # Mock Firebase reference
+def test_write_user_data(mock_users_ref):
+    mock_users_ref.child.return_value.set = MagicMock()
+
+    name = "Charlie"
+    score = 88
+    interview_question = "How do you handle challenges?"
+    job_requirements = "Problem solver"
+    feedback = "Well-articulated response."
+
+    entry_id = write_user_data(
+        name, score, interview_question, job_requirements, feedback
+    )
+
+    # Check that Firebase `set` was called with correct data
+    mock_users_ref.child.assert_called_with(entry_id)
+    mock_users_ref.child(entry_id).set.assert_called_once()
+
+
+@patch("src.configs.database.firebase.users_ref")
+def test_read_all_users(mock_users_ref):
+    # Mock the Firebase `get` method
+    mock_users_ref.get.return_value = MOCK_USER_DATA
+
+    df = read_all_users()
+
+    assert isinstance(df, pd.DataFrame)
+    assert not df.empty
+    assert "name" in df.columns
+
+
+@patch("src.configs.database.firebase.users_ref")
+def test_update_user_data(mock_users_ref):
+    mock_users_ref.child.return_value.get.return_value = MOCK_USER_DATA["uuid7-1"]
+    mock_users_ref.child.return_value.update = MagicMock()
+
+    uuid = "uuid7-1"
+    update_dict = {"score": 95, "feedback": "Updated feedback."}
+
+    result = update_user_data(uuid, update_dict)
+
+    assert result is True
+    mock_users_ref.child.assert_called_with(uuid)
+    mock_users_ref.child(uuid).update.assert_called_once_with(update_dict)
+
+
+@patch("src.configs.database.firebase.users_ref")
+def test_update_user_data_no_record(mock_users_ref):
+    mock_users_ref.child.return_value.get.return_value = None
+
+    uuid = "non-existent-uuid"
+    update_dict = {"score": 95, "feedback": "Updated feedback."}
+
+    result = update_user_data(uuid, update_dict)
+
+    assert result is False
+    mock_users_ref.child.assert_called_with(uuid)