42 lines
1.8 KiB
Python
42 lines
1.8 KiB
Python
import unittest
|
|
import json
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
from localpilot.server import create_app
|
|
from localpilot.server import _content_to_text, _sse
|
|
|
|
|
|
class ServerHelperTests(unittest.TestCase):
|
|
def test_content_to_text(self) -> None:
|
|
self.assertEqual(_content_to_text("hello"), "hello")
|
|
self.assertEqual(_content_to_text([{"type": "text", "text": "a"}, {"type": "text", "text": "b"}]), "ab")
|
|
|
|
def test_sse(self) -> None:
|
|
value = _sse({"ok": True})
|
|
self.assertTrue(value.startswith("data: "))
|
|
self.assertTrue(value.endswith("\n\n"))
|
|
|
|
def test_ollama_surface_and_registry(self) -> None:
|
|
with tempfile.TemporaryDirectory() as folder:
|
|
path = Path(folder) / "config.json"
|
|
path.write_text(json.dumps({"models": []}), encoding="utf-8")
|
|
client = TestClient(create_app(path))
|
|
self.assertEqual(client.get("/api/version").status_code, 200)
|
|
self.assertEqual(client.get("/api/tags").json(), {"models": []})
|
|
registered = client.post("/api/models/register", json={"id": "demo", "kind": "gguf", "path": "E:/demo.gguf"})
|
|
self.assertEqual(registered.status_code, 200)
|
|
self.assertEqual(client.get("/api/tags").json()["models"][0]["name"], "demo")
|
|
created = client.post("/api/create", json={"model": "created", "modelfile": "FROM E:/models/created.gguf\nPARAMETER temperature 0.2"})
|
|
self.assertEqual(created.status_code, 200)
|
|
copied = client.post("/api/copy", json={"source": "created", "destination": "created-copy"})
|
|
self.assertEqual(copied.status_code, 200)
|
|
removed = client.delete("/api/models/demo")
|
|
self.assertEqual(removed.status_code, 200)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|