frederik
170 lines
6.7 KiB
Python
170 lines
6.7 KiB
Python
import os
|
|
import sqlite3
|
|
from datetime import datetime
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from scripts.utils import notifications
|
|
from scripts.utils.notifications import sendAppriseNotifications
|
|
|
|
|
|
class TestAppriseNotifications(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.db_file = "test.db"
|
|
cls.apprise_body_file = "test_apprise_body"
|
|
cls.apprise_config_file = "test_apprise_config"
|
|
|
|
def create_test_db(self):
|
|
""" create a database connection to a SQLite database """
|
|
conn = None
|
|
try:
|
|
conn = sqlite3.connect(self.db_file)
|
|
sql_create_detections_table = """ CREATE TABLE IF NOT EXISTS detections (
|
|
id integer PRIMARY KEY,
|
|
Sci_Name text NOT NULL,
|
|
Com_Name text NOT NULL,
|
|
Date date NOT NULL,
|
|
Time time NULL
|
|
); """
|
|
cur = conn.cursor()
|
|
cur.execute(sql_create_detections_table)
|
|
sql = ''' INSERT INTO detections(Sci_Name, Com_Name, Date)
|
|
VALUES(?,?,?) '''
|
|
|
|
today = datetime.now().strftime("%Y-%m-%d") # SQLite stores date as YYYY-MM-DD
|
|
cur.execute(sql, ["Myiarchus crinitus", "Great Crested Flycatcher", today])
|
|
conn.commit()
|
|
|
|
except Exception as e:
|
|
print(e)
|
|
finally:
|
|
if conn:
|
|
conn.close()
|
|
|
|
def create_apprise_config(self):
|
|
with open(self.apprise_body_file, 'w') as f:
|
|
f.write('A $comname ($sciname) was just detected with a confidence of $confidencepct ($reason)')
|
|
with open(self.apprise_config_file, 'w') as f:
|
|
f.write('a dummy config')
|
|
notifications.APPRISE_BODY = self.apprise_body_file
|
|
notifications.APPRISE_CONFIG = self.apprise_config_file
|
|
|
|
def tearDown(self):
|
|
if os.path.exists(self.db_file):
|
|
os.remove(self.db_file)
|
|
if os.path.exists(self.apprise_body_file):
|
|
os.remove(self.apprise_body_file)
|
|
if os.path.exists(self.apprise_config_file):
|
|
os.remove(self.apprise_config_file)
|
|
|
|
@patch('scripts.utils.notifications.notify')
|
|
def test_notifications(self, mock_notify):
|
|
self.create_test_db()
|
|
self.create_apprise_config()
|
|
settings_dict = {
|
|
"APPRISE_NOTIFICATION_TITLE": "New backyard bird!",
|
|
"APPRISE_NOTIFY_EACH_DETECTION": "0",
|
|
"APPRISE_NOTIFY_NEW_SPECIES": "0",
|
|
"APPRISE_NOTIFY_NEW_SPECIES_EACH_DAY": "0",
|
|
"APPRISE_MINIMUM_SECONDS_BETWEEN_NOTIFICATIONS_PER_SPECIES": "0"
|
|
}
|
|
sendAppriseNotifications("Myiarchus crinitus",
|
|
"Great Crested Flycatcher",
|
|
"0.91",
|
|
"91",
|
|
"filename",
|
|
"1666-06-06",
|
|
"06:06:06",
|
|
"06",
|
|
"-1",
|
|
"-1",
|
|
"0.7",
|
|
"1.25",
|
|
"0.0",
|
|
settings_dict,
|
|
self.db_file)
|
|
|
|
# No active apprise notifications configured. Confirm no notifications.
|
|
self.assertEqual(mock_notify.call_count, 0)
|
|
|
|
# Add daily notification.
|
|
mock_notify.reset_mock()
|
|
settings_dict["APPRISE_NOTIFY_NEW_SPECIES_EACH_DAY"] = "1"
|
|
sendAppriseNotifications("Myiarchus crinitus",
|
|
"Great Crested Flycatcher",
|
|
"0.91",
|
|
"91",
|
|
"filename",
|
|
"1666-06-06",
|
|
"06:06:06",
|
|
"06",
|
|
"-1",
|
|
"-1",
|
|
"0.7",
|
|
"1.25",
|
|
"0.0",
|
|
settings_dict,
|
|
self.db_file)
|
|
|
|
self.assertEqual(mock_notify.call_count, 1)
|
|
self.assertEqual(
|
|
mock_notify.call_args_list[0][0][0],
|
|
"A Great Crested Flycatcher (Myiarchus crinitus) was just detected with a confidence of 91 (first time today)"
|
|
)
|
|
|
|
# Add new species notification.
|
|
mock_notify.reset_mock()
|
|
settings_dict["APPRISE_NOTIFY_NEW_SPECIES"] = "1"
|
|
sendAppriseNotifications("Myiarchus crinitus",
|
|
"Great Crested Flycatcher",
|
|
"0.91",
|
|
"91",
|
|
"filename",
|
|
"1666-06-06",
|
|
"06:06:06",
|
|
"06",
|
|
"-1",
|
|
"-1",
|
|
"0.7",
|
|
"1.25",
|
|
"0.0",
|
|
settings_dict,
|
|
self.db_file)
|
|
|
|
self.assertEqual(mock_notify.call_count, 2)
|
|
self.assertEqual(
|
|
mock_notify.call_args_list[0][0][0],
|
|
"A Great Crested Flycatcher (Myiarchus crinitus) was just detected with a confidence of 91 (first time today)"
|
|
)
|
|
self.assertEqual(
|
|
mock_notify.call_args_list[1][0][0],
|
|
"A Great Crested Flycatcher (Myiarchus crinitus) was just detected with a confidence of 91 (only seen 1 times in last 7d)"
|
|
)
|
|
|
|
# Add each species notification.
|
|
mock_notify.reset_mock()
|
|
settings_dict["APPRISE_NOTIFY_EACH_DETECTION"] = "1"
|
|
sendAppriseNotifications("Myiarchus crinitus",
|
|
"Great Crested Flycatcher",
|
|
"0.91",
|
|
"91",
|
|
"filename",
|
|
"1666-06-06",
|
|
"06:06:06",
|
|
"06",
|
|
"-1",
|
|
"-1",
|
|
"0.7",
|
|
"1.25",
|
|
"0.0",
|
|
settings_dict,
|
|
self.db_file)
|
|
|
|
self.assertEqual(mock_notify.call_count, 3)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|