From dd856b350e88d5095e9f2815c9e09c1cd0b21eb6 Mon Sep 17 00:00:00 2001
From: spacemanspiff2007 <10754716+spacemanspiff2007@users.noreply.github.com>
Date: Wed, 14 Apr 2021 17:55:25 +0200
Subject: [PATCH] added tests

---
 tests/jobs/test_job_onetime.py | 15 ++++++++++++++-
 tests/test_view.py             |  4 ++++
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/tests/jobs/test_job_onetime.py b/tests/jobs/test_job_onetime.py
index 088437b..d6938a1 100644
--- a/tests/jobs/test_job_onetime.py
+++ b/tests/jobs/test_job_onetime.py
@@ -1,9 +1,10 @@
 import asyncio
+import time
 
 import pytest
 
 from eascheduler.const import SKIP_EXECUTION
-from eascheduler.errors import OneTimeJobCanNotBeSkipped, JobAlreadyCanceledException
+from eascheduler.errors import JobAlreadyCanceledException, OneTimeJobCanNotBeSkipped
 from eascheduler.jobs.job_one_time import OneTimeJob
 from eascheduler.schedulers import AsyncScheduler
 from tests.helper import mocked_executor, set_now
@@ -43,3 +44,15 @@ async def test_remove():
 
     with pytest.raises(JobAlreadyCanceledException):
         j.cancel()
+
+
+@pytest.mark.asyncio
+async def test_init():
+    s = AsyncScheduler()
+    j = OneTimeJob(s, lambda x: x)
+
+    j._initialize_base_time(None)
+    assert j._next_base <= round(time.time(), 3)
+
+    j._initialize_base_time(3)
+    assert j._next_base <= round(time.time() + 3, 3)
diff --git a/tests/test_view.py b/tests/test_view.py
index 5b62d60..2c66829 100644
--- a/tests/test_view.py
+++ b/tests/test_view.py
@@ -27,9 +27,13 @@ async def test_at(view: SchedulerView):
 
     view.at(time(11), lambda x: x)
     assert s.jobs[0].get_next_run() == datetime(2001, 1, 2, 11)
+
     view.at(time(13), lambda x: x)
     assert s.jobs[0].get_next_run() == datetime(2001, 1, 1, 13)
 
+    view.at(5, lambda x: x)
+    assert s.jobs[0].get_next_run() == datetime(2001, 1, 1, 12, 0, 5)
+
 
 @pytest.mark.asyncio
 async def test_every(view: SchedulerView):