diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/conftest.py | 10 | ||||
-rw-r--r-- | tests/test_datastores.py | 3 | ||||
-rw-r--r-- | tests/test_events.py | 1 | ||||
-rw-r--r-- | tests/test_marshalling.py | 1 | ||||
-rw-r--r-- | tests/test_schedulers.py | 1 | ||||
-rw-r--r-- | tests/test_workers.py | 1 | ||||
-rw-r--r-- | tests/triggers/test_calendarinterval.py | 1 | ||||
-rw-r--r-- | tests/triggers/test_combining.py | 1 | ||||
-rw-r--r-- | tests/triggers/test_cron.py | 1 | ||||
-rw-r--r-- | tests/triggers/test_interval.py | 1 |
10 files changed, 17 insertions, 4 deletions
diff --git a/tests/conftest.py b/tests/conftest.py index 1de47ed..62af791 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -3,6 +3,7 @@ from contextlib import asynccontextmanager, contextmanager from typing import AsyncContextManager, AsyncGenerator, ContextManager, Generator, Optional import pytest + from apscheduler.abc import AsyncDataStore, DataStore, Serializer from apscheduler.adapters import AsyncDataStoreAdapter from apscheduler.datastores.sync.memory import MemoryDataStore @@ -43,18 +44,20 @@ def setup_memory_store() -> Generator[DataStore, None, None]: @contextmanager def setup_mongodb_store() -> Generator[DataStore, None, None]: - from apscheduler.datastores.sync.mongodb import MongoDBDataStore from pymongo import MongoClient + from apscheduler.datastores.sync.mongodb import MongoDBDataStore + with MongoClient(tz_aware=True, serverSelectionTimeoutMS=1000) as client: yield MongoDBDataStore(client, start_from_scratch=True) @contextmanager def setup_sqlalchemy_store() -> Generator[DataStore, None, None]: - from apscheduler.datastores.sync.sqlalchemy import SQLAlchemyDataStore from sqlalchemy.future import create_engine + from apscheduler.datastores.sync.sqlalchemy import SQLAlchemyDataStore + engine = create_engine('postgresql+psycopg2://postgres:secret@localhost/testdb') try: yield SQLAlchemyDataStore(engine, start_from_scratch=True) @@ -64,9 +67,10 @@ def setup_sqlalchemy_store() -> Generator[DataStore, None, None]: @asynccontextmanager async def setup_async_sqlalchemy_store() -> AsyncGenerator[AsyncDataStore, None]: - from apscheduler.datastores.async_.sqlalchemy import SQLAlchemyDataStore from sqlalchemy.ext.asyncio import create_async_engine + from apscheduler.datastores.async_.sqlalchemy import SQLAlchemyDataStore + engine = create_async_engine('postgresql+asyncpg://postgres:secret@localhost/testdb', future=True) try: diff --git a/tests/test_datastores.py b/tests/test_datastores.py index 6f387ac..651a66d 100644 --- a/tests/test_datastores.py +++ b/tests/test_datastores.py @@ -6,11 +6,12 @@ from typing import AsyncContextManager, AsyncGenerator, List, Optional, Set, Typ import anyio import pytest +from freezegun.api import FrozenDateTimeFactory + from apscheduler.abc import AsyncDataStore, Job, Schedule from apscheduler.events import Event, JobAdded, ScheduleAdded, ScheduleRemoved, ScheduleUpdated from apscheduler.policies import CoalescePolicy, ConflictPolicy from apscheduler.triggers.date import DateTrigger -from freezegun.api import FrozenDateTimeFactory @pytest.fixture diff --git a/tests/test_events.py b/tests/test_events.py index 3c9ebd6..bbe344f 100644 --- a/tests/test_events.py +++ b/tests/test_events.py @@ -5,6 +5,7 @@ from typing import List, Optional import pytest from _pytest.logging import LogCaptureFixture + from apscheduler.events import AsyncEventHub, Event, EventHub diff --git a/tests/test_marshalling.py b/tests/test_marshalling.py index 541ea3e..61fc783 100644 --- a/tests/test_marshalling.py +++ b/tests/test_marshalling.py @@ -4,6 +4,7 @@ from functools import partial from types import ModuleType import pytest + from apscheduler.exceptions import SerializationError from apscheduler.marshalling import callable_from_ref, callable_to_ref diff --git a/tests/test_schedulers.py b/tests/test_schedulers.py index 5829f30..3c0911d 100644 --- a/tests/test_schedulers.py +++ b/tests/test_schedulers.py @@ -5,6 +5,7 @@ from typing import List import anyio import pytest from anyio import fail_after + from apscheduler.events import ( Event, JobAdded, ScheduleAdded, ScheduleRemoved, SchedulerStarted, SchedulerStopped) from apscheduler.schedulers.async_ import AsyncScheduler diff --git a/tests/test_workers.py b/tests/test_workers.py index 9f098bc..18f9fb7 100644 --- a/tests/test_workers.py +++ b/tests/test_workers.py @@ -5,6 +5,7 @@ from typing import Callable, List import anyio import pytest from anyio import fail_after + from apscheduler.abc import Job from apscheduler.datastores.sync.memory import MemoryDataStore from apscheduler.events import ( diff --git a/tests/triggers/test_calendarinterval.py b/tests/triggers/test_calendarinterval.py index 96f9b7e..4074ff8 100644 --- a/tests/triggers/test_calendarinterval.py +++ b/tests/triggers/test_calendarinterval.py @@ -1,6 +1,7 @@ from datetime import date, datetime import pytest + from apscheduler.triggers.calendarinterval import CalendarIntervalTrigger diff --git a/tests/triggers/test_combining.py b/tests/triggers/test_combining.py index c7c409e..99e7614 100644 --- a/tests/triggers/test_combining.py +++ b/tests/triggers/test_combining.py @@ -1,6 +1,7 @@ from datetime import datetime, timedelta import pytest + from apscheduler.exceptions import MaxIterationsReached from apscheduler.triggers.combining import AndTrigger, OrTrigger from apscheduler.triggers.date import DateTrigger diff --git a/tests/triggers/test_cron.py b/tests/triggers/test_cron.py index c378d8d..d21b916 100644 --- a/tests/triggers/test_cron.py +++ b/tests/triggers/test_cron.py @@ -2,6 +2,7 @@ import sys from datetime import datetime import pytest + from apscheduler.triggers.cron import CronTrigger if sys.version_info >= (3, 9): diff --git a/tests/triggers/test_interval.py b/tests/triggers/test_interval.py index 292659f..811af3a 100644 --- a/tests/triggers/test_interval.py +++ b/tests/triggers/test_interval.py @@ -1,6 +1,7 @@ from datetime import datetime, timedelta import pytest + from apscheduler.triggers.interval import IntervalTrigger |