48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from unittest.mock import MagicMock
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from fastapi import FastAPI
|
||
|
|
|
||
|
|
from yuxi.channel.container import setup_channel
|
||
|
|
|
||
|
|
|
||
|
|
class TestSetupChannel:
|
||
|
|
def test_setup_channel_with_fastapi(self):
|
||
|
|
app = FastAPI()
|
||
|
|
container = MagicMock()
|
||
|
|
setup_channel(app, container)
|
||
|
|
assert hasattr(app.state, "channel")
|
||
|
|
assert app.state.channel is container
|
||
|
|
|
||
|
|
def test_setup_channel_with_non_fastapi(self):
|
||
|
|
app = MagicMock()
|
||
|
|
container = MagicMock()
|
||
|
|
setup_channel(app, container)
|
||
|
|
assert not hasattr(app.state, "channel")
|
||
|
|
|
||
|
|
def test_get_channel(self):
|
||
|
|
from yuxi.channel.container import get_channel
|
||
|
|
from fastapi import Request
|
||
|
|
|
||
|
|
app = FastAPI()
|
||
|
|
container = MagicMock()
|
||
|
|
app.state.channel = container
|
||
|
|
|
||
|
|
request = MagicMock(spec=Request)
|
||
|
|
request.app = app
|
||
|
|
result = get_channel(request)
|
||
|
|
assert result is container
|
||
|
|
|
||
|
|
def test_get_channel_not_initialized(self):
|
||
|
|
from yuxi.channel.container import get_channel
|
||
|
|
from fastapi import Request, HTTPException
|
||
|
|
|
||
|
|
app = FastAPI()
|
||
|
|
request = MagicMock(spec=Request)
|
||
|
|
request.app = app
|
||
|
|
with pytest.raises(HTTPException) as exc_info:
|
||
|
|
get_channel(request)
|
||
|
|
assert exc_info.value.status_code == 503
|