Initial Commit 🚀 🚀

api/tests/test_disposition_mapper.py

@@ -0,0 +1,122 @@
+from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
+
+from api.services.workflow.disposition_mapper import (
+    apply_disposition_mapping,
+    get_organization_id_from_workflow_run,
+)
+
+
+@pytest.mark.asyncio
+async def test_apply_disposition_mapping_with_valid_mapping():
+    """Test disposition mapping with valid configuration."""
+    with patch("api.services.workflow.disposition_mapper.db_client") as mock_db_client:
+        # Mock disposition mapping configuration
+        mock_db_client.get_configuration_value = AsyncMock(
+            return_value={
+                "XFER": "TRANSFERRED",
+                "ND": "NOT_QUALIFIED",
+                "user_hangup": "HANGUP",
+            }
+        )
+
+        # Test mapping exists
+        result = await apply_disposition_mapping("XFER", 1)
+        assert result == "TRANSFERRED"
+
+        # Test mapping doesn't exist
+        result = await apply_disposition_mapping("UNKNOWN", 1)
+        assert result == "UNKNOWN"
+
+        # Verify db_client was called correctly
+        mock_db_client.get_configuration_value.assert_called_with(
+            1, "DISPOSITION_CODE_MAPPING", default={}
+        )
+
+
+@pytest.mark.asyncio
+async def test_apply_disposition_mapping_no_organization_id():
+    """Test disposition mapping with no organization ID."""
+    # Should return original value
+    result = await apply_disposition_mapping("XFER", None)
+    assert result == "XFER"
+
+
+@pytest.mark.asyncio
+async def test_apply_disposition_mapping_empty_value():
+    """Test disposition mapping with empty value."""
+    # Should return original empty value
+    result = await apply_disposition_mapping("", 1)
+    assert result == ""
+
+
+@pytest.mark.asyncio
+async def test_apply_disposition_mapping_error_handling():
+    """Test disposition mapping handles errors gracefully."""
+    with patch("api.services.workflow.disposition_mapper.db_client") as mock_db_client:
+        # Mock database error
+        mock_db_client.get_configuration_value = AsyncMock(
+            side_effect=Exception("Database error")
+        )
+
+        # Should return original value on error
+        result = await apply_disposition_mapping("XFER", 1)
+        assert result == "XFER"
+
+
+@pytest.mark.asyncio
+async def test_get_organization_id_from_workflow_run():
+    """Test getting organization ID from workflow run ID."""
+    with patch("api.services.workflow.disposition_mapper.db_client") as mock_db_client:
+        # Mock workflow run with organization
+        mock_workflow_run = MagicMock()
+        mock_workflow_run.workflow.user.selected_organization_id = 123
+        mock_db_client.get_workflow_run_by_id = AsyncMock(
+            return_value=mock_workflow_run
+        )
+
+        result = await get_organization_id_from_workflow_run(1)
+        assert result == 123
+
+        # Verify db_client was called correctly
+        mock_db_client.get_workflow_run_by_id.assert_called_once_with(1)
+
+
+@pytest.mark.asyncio
+async def test_get_organization_id_no_workflow_run():
+    """Test getting organization ID when workflow run doesn't exist."""
+    with patch("api.services.workflow.disposition_mapper.db_client") as mock_db_client:
+        # Mock no workflow run found
+        mock_db_client.get_workflow_run_by_id = AsyncMock(return_value=None)
+
+        result = await get_organization_id_from_workflow_run(1)
+        assert result is None
+
+
+@pytest.mark.asyncio
+async def test_get_organization_id_no_user():
+    """Test getting organization ID when workflow has no user."""
+    with patch("api.services.workflow.disposition_mapper.db_client") as mock_db_client:
+        # Mock workflow run with no user
+        mock_workflow_run = MagicMock()
+        mock_workflow_run.workflow.user = None
+        mock_db_client.get_workflow_run_by_id = AsyncMock(
+            return_value=mock_workflow_run
+        )
+
+        result = await get_organization_id_from_workflow_run(1)
+        assert result is None
+
+
+@pytest.mark.asyncio
+async def test_get_organization_id_error_handling():
+    """Test getting organization ID handles errors gracefully."""
+    with patch("api.services.workflow.disposition_mapper.db_client") as mock_db_client:
+        # Mock database error
+        mock_db_client.get_workflow_run_by_id = AsyncMock(
+            side_effect=Exception("Database error")
+        )
+
+        result = await get_organization_id_from_workflow_run(1)
+        assert result is None