Spaces:
Running
Running
File size: 3,455 Bytes
a8c9ee8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 | import json
import pytest
from unittest.mock import MagicMock, patch
from fastapi.testclient import TestClient
from fastapi import FastAPI
import sys
import os
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../..')))
from backend.data_sources.api import router, get_redis_client, get_tenant_config
from backend.core.auth import AuthUser, get_current_user
app = FastAPI()
app.include_router(router)
def mock_get_current_user():
return AuthUser({
"id": "test_user",
"user_metadata": {"tenant_id": "test_tenant", "role": "user"}
})
app.dependency_overrides[get_current_user] = mock_get_current_user
client = TestClient(app)
class TestPatchAndDeleteEndpoints:
@patch('backend.data_sources.api.get_redis_client')
def test_patch_tenant_source(self, mock_redis):
mock_redis_instance = MagicMock()
mock_redis.return_value = mock_redis_instance
# Existing config
existing_config = [
{
"source_name": "old_db",
"source_type": "ibis",
"config": {"uri": "postgres://user:pass@localhost:5432/db1", "other": "val"}
}
]
mock_redis_instance.get.return_value = json.dumps(existing_config)
# Override the dependency for the duration of the request
app.dependency_overrides[get_redis_client] = lambda: mock_redis_instance
patch_request = {
"new_source_name": "new_db",
"config_updates": {"uri": "postgres://user:newpass@localhost:5432/db1"}
}
response = client.patch("/api/v1/data-sources/my-tenant/sources/old_db?validate_connection=false", json=patch_request)
# Clean up
app.dependency_overrides.pop(get_redis_client, None)
assert response.status_code == 200
# Check that set was called with updated config
mock_redis_instance.set.assert_called_once()
args, _ = mock_redis_instance.set.call_args
key, value = args
assert key == "tenant:test_tenant:config"
updated_sources = json.loads(value)
assert len(updated_sources) == 1
assert updated_sources[0]["source_name"] == "new_db"
assert updated_sources[0]["config"]["uri"] == "postgres://user:newpass@localhost:5432/db1"
assert updated_sources[0]["config"]["other"] == "val" # Ensure config was merged, not replaced
@patch('backend.data_sources.api.get_redis_client')
def test_delete_tenant_source_cascades(self, mock_redis):
mock_redis_instance = MagicMock()
mock_redis.return_value = mock_redis_instance
existing_config = [{"source_name": "db_to_delete", "config": {}}]
mock_redis_instance.get.return_value = json.dumps(existing_config)
app.dependency_overrides[get_redis_client] = lambda: mock_redis_instance
response = client.delete("/api/v1/data-sources/my-tenant/sources/db_to_delete")
app.dependency_overrides.pop(get_redis_client, None)
assert response.status_code == 200
# Redis delete should be called for caches
delete_calls = mock_redis_instance.delete.call_args_list
assert any(call[0][0] == "schema:test_tenant:db_to_delete" for call in delete_calls)
assert any(call[0][0] == "instructions:test_tenant:db_to_delete" for call in delete_calls)
|