Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

add timeouts to the download tests

Browse Source
This commit is contained in:
Lincoln Stein 2024-03-20 22:40:55 -04:00
parent ce687a2869
commit b65eff1c65
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/app/services/download/test_download_queue.py
View File

@ -51,6 +51,7 @@ def session() -> Session:
return sess return sess
@pytest.mark.timeout(timeout=20, method="thread")
def test_basic_queue_download(tmp_path: Path, session: Session) -> None: def test_basic_queue_download(tmp_path: Path, session: Session) -> None:
events = set() events = set()
@ -80,6 +81,7 @@ def test_basic_queue_download(tmp_path: Path, session: Session) -> None:
queue.stop() queue.stop()
@pytest.mark.timeout(timeout=20, method="thread")
def test_errors(tmp_path: Path, session: Session) -> None: def test_errors(tmp_path: Path, session: Session) -> None:
queue = DownloadQueueService( queue = DownloadQueueService(
requests_session=session, requests_session=session,
@ -101,6 +103,7 @@ def test_errors(tmp_path: Path, session: Session) -> None:
queue.stop() queue.stop()
@pytest.mark.timeout(timeout=20, method="thread")
def test_event_bus(tmp_path: Path, session: Session) -> None: def test_event_bus(tmp_path: Path, session: Session) -> None:
event_bus = TestEventService() event_bus = TestEventService()
@ -136,6 +139,7 @@ def test_event_bus(tmp_path: Path, session: Session) -> None:
queue.stop() queue.stop()
@pytest.mark.timeout(timeout=20, method="thread")
def test_broken_callbacks(tmp_path: Path, session: Session, capsys) -> None: def test_broken_callbacks(tmp_path: Path, session: Session, capsys) -> None:
queue = DownloadQueueService( queue = DownloadQueueService(
requests_session=session, requests_session=session,