84 lines
3.0 KiB
Python
84 lines
3.0 KiB
Python
import unittest
|
|
from contextlib import contextmanager
|
|
from unittest.mock import patch
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
from ca_api.app import app
|
|
|
|
|
|
class _FakeCursor:
|
|
pass
|
|
|
|
|
|
@contextmanager
|
|
def _fake_db_cursor():
|
|
# Mimics ca_api.db.db_cursor() which yields a cursor
|
|
yield _FakeCursor()
|
|
|
|
|
|
class TestApiUnit(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.client = TestClient(app)
|
|
|
|
def test_create_creator_201(self):
|
|
with patch("ca_api.app.db_cursor", _fake_db_cursor), \
|
|
patch("ca_api.app.entity_core.insert_creator", return_value=123) as m_insert:
|
|
payload = {"name": "Alice", "public_key": "pk-alice"}
|
|
r = self.client.post("/creators", json=payload)
|
|
|
|
self.assertEqual(r.status_code, 201)
|
|
self.assertEqual(r.json(), {"id": 123})
|
|
|
|
# Ensure core called with (cursor, name, public_key)
|
|
args, kwargs = m_insert.call_args
|
|
self.assertIsInstance(args[0], _FakeCursor)
|
|
self.assertEqual(args[1], "Alice")
|
|
self.assertEqual(args[2], "pk-alice")
|
|
self.assertEqual(kwargs, {})
|
|
|
|
def test_get_entity_404(self):
|
|
with patch("ca_api.app.db_cursor", _fake_db_cursor), \
|
|
patch("ca_api.app.entity_core.get_entity", return_value=None):
|
|
r = self.client.get("/entities/99999")
|
|
|
|
self.assertEqual(r.status_code, 404)
|
|
self.assertEqual(r.json()["detail"], "Entity not found")
|
|
|
|
def test_get_entity_200(self):
|
|
# FastAPI returns dict(row), so return any mapping-like object here
|
|
with patch("ca_api.app.db_cursor", _fake_db_cursor), \
|
|
patch("ca_api.app.entity_core.get_entity", return_value={"id": 7, "name": "Bob"}):
|
|
r = self.client.get("/entities/7")
|
|
|
|
self.assertEqual(r.status_code, 200)
|
|
self.assertEqual(r.json()["id"], 7)
|
|
self.assertEqual(r.json()["name"], "Bob")
|
|
|
|
def test_set_property_400_on_value_error(self):
|
|
def _raise(*args, **kwargs):
|
|
raise ValueError("bad property")
|
|
|
|
with patch("ca_api.app.db_cursor", _fake_db_cursor), \
|
|
patch("ca_api.app.property_core.set_property", side_effect=_raise):
|
|
payload = {"entity_id": 1, "property_name": "email", "validation_policy": "default"}
|
|
r = self.client.post("/properties", json=payload)
|
|
|
|
self.assertEqual(r.status_code, 400)
|
|
self.assertEqual(r.json()["detail"], "bad property")
|
|
|
|
def test_add_member_201(self):
|
|
with patch("ca_api.app.db_cursor", _fake_db_cursor), \
|
|
patch("ca_api.app.group_core.add_group_member", return_value=None) as m_add:
|
|
payload = {"group_id": 10, "member_id": 20, "role": "admin"}
|
|
r = self.client.post("/groups/members", json=payload)
|
|
|
|
self.assertEqual(r.status_code, 201)
|
|
self.assertEqual(r.json(), {"ok": True})
|
|
|
|
args, kwargs = m_add.call_args
|
|
self.assertIsInstance(args[0], _FakeCursor)
|
|
self.assertEqual(args[1:], (10, 20, "admin"))
|
|
self.assertEqual(kwargs, {})
|