mirror of
https://github.com/Ichinga-Samuel/aiomql.git
synced 2026-08-16 21:38:06 +00:00
Ichinga Samuel
06061ec917
rename unittests to unit change TradeRecords.get_csv_records and TradeRecord.get_json_records methods to non-async.
69 lines
2.9 KiB
Python
69 lines
2.9 KiB
Python
from datetime import datetime, time
|
|
|
|
import pytest
|
|
import pytz
|
|
|
|
from aiomql.lib.sessions import Session, Sessions, delta
|
|
|
|
|
|
class TestSessions:
|
|
@pytest.fixture(scope='class')
|
|
def make_sessions(self, make_session):
|
|
london, all_day, over_night = make_session
|
|
return Sessions(sessions=[london, all_day, over_night])
|
|
|
|
@pytest.fixture(scope='class')
|
|
def make_session(self):
|
|
end = time(hour=16, minute=59, second=59, microsecond=999_999, tzinfo=pytz.UTC)
|
|
london = Session(start=8, end=end, name='London', on_end='close_all')
|
|
start, end = time(hour=0, tzinfo=pytz.UTC), time(hour=23, minute=59, second=59, tzinfo=pytz.UTC)
|
|
all_day = Session(start=start, end=end, name='AllDay', on_end='close_all')
|
|
end = time(hour=6, minute=59, second=59, microsecond=999_999, tzinfo=pytz.UTC)
|
|
over_night = Session(start=18, end=end, name='OverNight', on_end='close_all')
|
|
return london, all_day, over_night
|
|
|
|
def test_session_attributes(self, make_session):
|
|
london, all_day, over_night = make_session
|
|
period = over_night.duration()
|
|
assert london.name == 'London'
|
|
assert london.start == time(hour=8, tzinfo=pytz.UTC)
|
|
assert london.end.hour == 16
|
|
assert period.hours == 12
|
|
assert period.minutes == period.seconds == 59
|
|
|
|
def test_session_intervals(self, make_session):
|
|
london, all_day, over_night = make_session
|
|
two_am = time(hour=2, tzinfo=pytz.UTC)
|
|
noon = time(hour=12, tzinfo=pytz.UTC)
|
|
now = datetime.now(pytz.UTC).time()
|
|
hours_till_london_starts = (delta(london.start) - delta(now)).seconds // 3600
|
|
assert hours_till_london_starts == london.until() // 3600
|
|
assert two_am in over_night
|
|
assert noon in london
|
|
assert two_am not in london
|
|
assert noon not in over_night
|
|
# all_day session is always open
|
|
assert all_day.in_session()
|
|
|
|
async def test_sessions(self, make_session):
|
|
london, all_day, over_night = make_session
|
|
sessions = Sessions(sessions=[london, over_night])
|
|
now = time(hour=21, tzinfo=pytz.UTC)
|
|
noon = time(hour=12, tzinfo=pytz.UTC)
|
|
mid_nite = time(hour=0, tzinfo=pytz.UTC)
|
|
next_sess = sessions.find_next(moment=now)
|
|
noon_sess = sessions.find(moment=noon)
|
|
no_sess = sessions.find(moment=time(hour=17, tzinfo=pytz.UTC))
|
|
mid_nite_sess = sessions.find(moment=mid_nite)
|
|
current_sess = sessions.find(moment=now)
|
|
assert current_sess.name == 'OverNight'
|
|
assert noon_sess.name == 'London'
|
|
assert no_sess is None
|
|
assert next_sess.name == 'London'
|
|
assert mid_nite_sess.name == 'OverNight'
|
|
current = datetime.now(pytz.UTC).time()
|
|
if current.hour not in (7, 17):
|
|
await sessions.check()
|
|
assert sessions.current_session is not None
|
|
assert sessions.current_session.name in ('London', 'OverNight')
|