summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorAlex Grönholm <alex.gronholm@nextday.fi>2021-08-29 15:58:06 +0300
committerAlex Grönholm <alex.gronholm@nextday.fi>2021-08-29 16:08:49 +0300
commita0b7da22086b1704f52a3a34a3edb25f2bcd5d7e (patch)
treed7b25e7b1f29f968dd0968d2cb0ff511b3c8b56f /tests
parentc0d757cb8357c71a1187078bb18b246195cb8911 (diff)
downloadapscheduler-a0b7da22086b1704f52a3a34a3edb25f2bcd5d7e.tar.gz
Switched to the src/ layout
Diffstat (limited to 'tests')
-rw-r--r--tests/conftest.py10
-rw-r--r--tests/test_datastores.py3
-rw-r--r--tests/test_events.py1
-rw-r--r--tests/test_marshalling.py1
-rw-r--r--tests/test_schedulers.py1
-rw-r--r--tests/test_workers.py1
-rw-r--r--tests/triggers/test_calendarinterval.py1
-rw-r--r--tests/triggers/test_combining.py1
-rw-r--r--tests/triggers/test_cron.py1
-rw-r--r--tests/triggers/test_interval.py1
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