mirror of
https://github.com/caronc/apprise.git
synced 2024-11-24 17:14:00 +01:00
467 lines
15 KiB
Python
467 lines
15 KiB
Python
# -*- coding: utf-8 -*-
|
|
# BSD 2-Clause License
|
|
#
|
|
# Apprise - Push Notification Library.
|
|
# Copyright (c) 2024, Chris Caron <lead2gold@gmail.com>
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are met:
|
|
#
|
|
# 1. Redistributions of source code must retain the above copyright notice,
|
|
# this list of conditions and the following disclaimer.
|
|
#
|
|
# 2. Redistributions in binary form must reproduce the above copyright notice,
|
|
# this list of conditions and the following disclaimer in the documentation
|
|
# and/or other materials provided with the distribution.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
|
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
# POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
from os.path import dirname
|
|
from os.path import join
|
|
from apprise.decorators import notify
|
|
from apprise.decorators.CustomNotifyPlugin import CustomNotifyPlugin
|
|
from apprise import Apprise
|
|
from apprise import AppriseConfig
|
|
from apprise import AppriseAsset
|
|
from apprise import AppriseAttachment
|
|
from apprise import common
|
|
from apprise.NotificationManager import NotificationManager
|
|
|
|
# Disable logging for a cleaner testing output
|
|
import logging
|
|
logging.disable(logging.CRITICAL)
|
|
|
|
# Grant access to our Notification Manager Singleton
|
|
N_MGR = NotificationManager()
|
|
|
|
TEST_VAR_DIR = join(dirname(__file__), 'var')
|
|
|
|
|
|
def test_notify_simple_decoration():
|
|
"""decorators: Test simple @notify
|
|
"""
|
|
|
|
# Verify our schema we're about to declare doesn't already exist
|
|
# in our schema map:
|
|
assert 'utiltest' not in N_MGR
|
|
|
|
verify_obj = {}
|
|
|
|
# Define a function here on the spot
|
|
@notify(on="utiltest", name="Apprise @notify Decorator Testing")
|
|
def my_inline_notify_wrapper(
|
|
body, title, notify_type, attach, *args, **kwargs):
|
|
|
|
# Populate our object we can use to validate
|
|
verify_obj.update({
|
|
'body': body,
|
|
'title': title,
|
|
'notify_type': notify_type,
|
|
'attach': attach,
|
|
'args': args,
|
|
'kwargs': kwargs,
|
|
})
|
|
|
|
# Now after our hook being inline... it's been loaded
|
|
assert 'utiltest' in N_MGR
|
|
|
|
# Create ourselves an apprise object
|
|
aobj = Apprise()
|
|
|
|
assert aobj.add("utiltest://") is True
|
|
|
|
assert len(verify_obj) == 0
|
|
|
|
assert aobj.notify(
|
|
"Hello World", title="My Title",
|
|
# add some attachments too
|
|
attach=(
|
|
join(TEST_VAR_DIR, 'apprise-test.gif'),
|
|
join(TEST_VAR_DIR, 'apprise-test.png'),
|
|
)
|
|
) is True
|
|
|
|
# Our content was populated after the notify() call
|
|
assert len(verify_obj) > 0
|
|
assert verify_obj['body'] == "Hello World"
|
|
assert verify_obj['title'] == "My Title"
|
|
assert verify_obj['notify_type'] == common.NotifyType.INFO
|
|
assert isinstance(verify_obj['attach'], AppriseAttachment)
|
|
assert len(verify_obj['attach']) == 2
|
|
|
|
# No format was defined
|
|
assert 'body_format' in verify_obj['kwargs']
|
|
assert verify_obj['kwargs']['body_format'] is None
|
|
|
|
# The meta argument allows us to further parse the URL parameters
|
|
# specified
|
|
assert isinstance(verify_obj['kwargs'], dict)
|
|
assert 'meta' in verify_obj['kwargs']
|
|
assert isinstance(verify_obj['kwargs']['meta'], dict)
|
|
assert len(verify_obj['kwargs']['meta']) == 4
|
|
assert 'tag' in verify_obj['kwargs']['meta']
|
|
|
|
assert 'asset' in verify_obj['kwargs']['meta']
|
|
assert isinstance(verify_obj['kwargs']['meta']['asset'], AppriseAsset)
|
|
|
|
assert verify_obj['kwargs']['meta']['schema'] == 'utiltest'
|
|
assert verify_obj['kwargs']['meta']['url'] == 'utiltest://'
|
|
|
|
# Reset our verify object (so it can be populated again)
|
|
verify_obj = {}
|
|
|
|
# We'll do another test now
|
|
assert aobj.notify(
|
|
"Hello Another World", title="My Other Title",
|
|
body_format=common.NotifyFormat.HTML,
|
|
notify_type=common.NotifyType.WARNING,
|
|
) is True
|
|
|
|
# Our content was populated after the notify() call
|
|
assert len(verify_obj) > 0
|
|
assert verify_obj['body'] == "Hello Another World"
|
|
assert verify_obj['title'] == "My Other Title"
|
|
assert verify_obj['notify_type'] == common.NotifyType.WARNING
|
|
# We have no attachments
|
|
assert verify_obj['attach'] is None
|
|
|
|
# No format was defined
|
|
assert 'body_format' in verify_obj['kwargs']
|
|
assert verify_obj['kwargs']['body_format'] == common.NotifyFormat.HTML
|
|
|
|
# The meta argument allows us to further parse the URL parameters
|
|
# specified
|
|
assert 'meta' in verify_obj['kwargs']
|
|
assert isinstance(verify_obj['kwargs'], dict)
|
|
assert len(verify_obj['kwargs']['meta']) == 4
|
|
assert 'asset' in verify_obj['kwargs']['meta']
|
|
assert isinstance(verify_obj['kwargs']['meta']['asset'], AppriseAsset)
|
|
assert 'tag' in verify_obj['kwargs']['meta']
|
|
assert isinstance(verify_obj['kwargs']['meta']['tag'], set)
|
|
assert verify_obj['kwargs']['meta']['schema'] == 'utiltest'
|
|
assert verify_obj['kwargs']['meta']['url'] == 'utiltest://'
|
|
|
|
assert 'notexc' not in N_MGR
|
|
|
|
# Define a function here on the spot
|
|
@notify(on="notexc", name="Apprise @notify Exception Handling")
|
|
def my_exception_inline_notify_wrapper(
|
|
body, title, notify_type, attach, *args, **kwargs):
|
|
raise ValueError("An exception was thrown!")
|
|
|
|
assert 'notexc' in N_MGR
|
|
|
|
# Create ourselves an apprise object
|
|
aobj = Apprise()
|
|
|
|
assert aobj.add("notexc://") is True
|
|
|
|
# Isn't handled
|
|
assert aobj.notify("Exceptions will be thrown!") is False
|
|
|
|
# Tidy
|
|
N_MGR.remove('utiltest', 'notexc')
|
|
|
|
|
|
def test_notify_complex_decoration():
|
|
"""decorators: Test complex @notify
|
|
"""
|
|
|
|
# Verify our schema we're about to declare doesn't already exist
|
|
# in our schema map:
|
|
assert 'utiltest' not in N_MGR
|
|
|
|
verify_obj = {}
|
|
|
|
# Define a function here on the spot
|
|
@notify(on="utiltest://user@myhost:23?key=value&NOT=CaseSensitive",
|
|
name="Apprise @notify Decorator Testing")
|
|
def my_inline_notify_wrapper(
|
|
body, title, notify_type, attach, *args, **kwargs):
|
|
|
|
# Populate our object we can use to validate
|
|
verify_obj.update({
|
|
'body': body,
|
|
'title': title,
|
|
'notify_type': notify_type,
|
|
'attach': attach,
|
|
'args': args,
|
|
'kwargs': kwargs,
|
|
})
|
|
|
|
# Now after our hook being inline... it's been loaded
|
|
assert 'utiltest' in N_MGR
|
|
|
|
# Create ourselves an apprise object
|
|
aobj = Apprise()
|
|
|
|
assert aobj.add("utiltest://") is True
|
|
|
|
assert len(verify_obj) == 0
|
|
|
|
assert aobj.notify(
|
|
"Hello World", title="My Title",
|
|
# add some attachments too
|
|
attach=(
|
|
join(TEST_VAR_DIR, 'apprise-test.gif'),
|
|
join(TEST_VAR_DIR, 'apprise-test.png'),
|
|
)
|
|
) is True
|
|
|
|
# Our content was populated after the notify() call
|
|
assert len(verify_obj) > 0
|
|
assert verify_obj['body'] == "Hello World"
|
|
assert verify_obj['title'] == "My Title"
|
|
assert verify_obj['notify_type'] == common.NotifyType.INFO
|
|
assert isinstance(verify_obj['attach'], AppriseAttachment)
|
|
assert len(verify_obj['attach']) == 2
|
|
|
|
# No format was defined
|
|
assert 'body_format' in verify_obj['kwargs']
|
|
assert verify_obj['kwargs']['body_format'] is None
|
|
|
|
# The meta argument allows us to further parse the URL parameters
|
|
# specified
|
|
assert isinstance(verify_obj['kwargs'], dict)
|
|
assert 'meta' in verify_obj['kwargs']
|
|
assert isinstance(verify_obj['kwargs']['meta'], dict)
|
|
|
|
assert 'asset' in verify_obj['kwargs']['meta']
|
|
assert isinstance(verify_obj['kwargs']['meta']['asset'], AppriseAsset)
|
|
|
|
assert 'tag' in verify_obj['kwargs']['meta']
|
|
assert isinstance(verify_obj['kwargs']['meta']['tag'], set)
|
|
|
|
assert len(verify_obj['kwargs']['meta']) == 8
|
|
# We carry all of our default arguments from the @notify's initialization
|
|
assert verify_obj['kwargs']['meta']['schema'] == 'utiltest'
|
|
|
|
# Case sensitivity is lost on key assignment and always made lowercase
|
|
# however value case sensitivity is preseved.
|
|
# this is the assembled URL based on the combined values of the default
|
|
# parameters with values provided in the URL (user's configuration)
|
|
assert verify_obj['kwargs']['meta']['url'].startswith(
|
|
'utiltest://user@myhost:23?')
|
|
|
|
# We don't know where they get placed, so just search for their match
|
|
assert 'key=value' in verify_obj['kwargs']['meta']['url']
|
|
assert 'not=CaseSensitive' in verify_obj['kwargs']['meta']['url']
|
|
|
|
# Reset our verify object (so it can be populated again)
|
|
verify_obj = {}
|
|
|
|
# We'll do another test now
|
|
aobj = Apprise()
|
|
|
|
assert aobj.add("utiltest://customhost?key=new&key2=another") is True
|
|
|
|
assert len(verify_obj) == 0
|
|
|
|
# Send our notification
|
|
assert aobj.notify("Hello World", title="My Title") is True
|
|
|
|
# Our content was populated after the notify() call
|
|
assert len(verify_obj) > 0
|
|
assert verify_obj['body'] == "Hello World"
|
|
assert verify_obj['title'] == "My Title"
|
|
assert verify_obj['notify_type'] == common.NotifyType.INFO
|
|
assert verify_obj['attach'] is None
|
|
|
|
# No format was defined
|
|
assert 'body_format' in verify_obj['kwargs']
|
|
assert verify_obj['kwargs']['body_format'] is None
|
|
|
|
# The meta argument allows us to further parse the URL parameters
|
|
# specified
|
|
assert 'meta' in verify_obj['kwargs']
|
|
assert isinstance(verify_obj['kwargs'], dict)
|
|
assert len(verify_obj['kwargs']['meta']) == 8
|
|
|
|
# We carry all of our default arguments from the @notify's initialization
|
|
assert verify_obj['kwargs']['meta']['schema'] == 'utiltest'
|
|
# Our host get's correctly over-ridden
|
|
assert verify_obj['kwargs']['meta']['host'] == 'customhost'
|
|
|
|
assert verify_obj['kwargs']['meta']['user'] == "user"
|
|
assert verify_obj['kwargs']['meta']['port'] == 23
|
|
assert isinstance(verify_obj['kwargs']['meta']['qsd'], dict)
|
|
assert len(verify_obj['kwargs']['meta']['qsd']) == 3
|
|
# our key is over-ridden
|
|
assert verify_obj['kwargs']['meta']['qsd']['key'] == 'new'
|
|
# Our other keys are preserved
|
|
assert verify_obj['kwargs']['meta']['qsd']['not'] == 'CaseSensitive'
|
|
# New keys are added
|
|
assert verify_obj['kwargs']['meta']['qsd']['key2'] == 'another'
|
|
|
|
# Case sensitivity is lost on key assignment and always made lowercase
|
|
# however value case sensitivity is preseved.
|
|
# this is the assembled URL based on the combined values of the default
|
|
# parameters with values provided in the URL (user's configuration)
|
|
assert verify_obj['kwargs']['meta']['url'].startswith(
|
|
'utiltest://user@customhost:23?')
|
|
|
|
# We don't know where they get placed, so just search for their match
|
|
assert 'key=new' in verify_obj['kwargs']['meta']['url']
|
|
assert 'not=CaseSensitive' in verify_obj['kwargs']['meta']['url']
|
|
assert 'key2=another' in verify_obj['kwargs']['meta']['url']
|
|
|
|
# Tidy
|
|
N_MGR.remove('utiltest')
|
|
|
|
|
|
def test_notify_multi_instance_decoration(tmpdir):
|
|
"""decorators: Test multi-instance @notify
|
|
"""
|
|
|
|
# Verify our schema we're about to declare doesn't already exist
|
|
# in our schema map:
|
|
assert 'multi' not in N_MGR
|
|
|
|
verify_obj = []
|
|
|
|
# Define a function here on the spot
|
|
@notify(on="multi", name="Apprise @notify Decorator Testing")
|
|
def my_inline_notify_wrapper(
|
|
body, title, notify_type, attach, meta, *args, **kwargs):
|
|
|
|
# Track what is added
|
|
verify_obj.append({
|
|
'body': body,
|
|
'title': title,
|
|
'notify_type': notify_type,
|
|
'attach': attach,
|
|
'meta': meta,
|
|
'args': args,
|
|
'kwargs': kwargs,
|
|
})
|
|
|
|
# Now after our hook being inline... it's been loaded
|
|
assert 'multi' in N_MGR
|
|
|
|
# Prepare our config
|
|
t = tmpdir.mkdir("multi-test").join("apprise.yml")
|
|
t.write("""urls:
|
|
- multi://user1:pass@hostname
|
|
- multi://user2:pass2@hostname?verify=no
|
|
""")
|
|
|
|
# Create ourselves a config object
|
|
ac = AppriseConfig(paths=str(t))
|
|
|
|
# Create ourselves an apprise object
|
|
aobj = Apprise()
|
|
|
|
# Add our configuration
|
|
aobj.add(ac)
|
|
|
|
# The number of configuration files that exist
|
|
assert len(ac) == 1
|
|
|
|
# no notifications are loaded
|
|
assert len(ac.servers()) == 2
|
|
|
|
# Nothing stored yet in our object
|
|
assert len(verify_obj) == 0
|
|
|
|
assert aobj.notify("Hello World", title="My Title") is True
|
|
|
|
assert len(verify_obj) == 2
|
|
|
|
# Python 3.6 does not nessisarily return list in order
|
|
# So let's be sure it's sorted by the user id field to make the remaining
|
|
# checks on this test easy
|
|
verify_obj = sorted(verify_obj, key=lambda x: x['meta']['user'])
|
|
|
|
# Our content was populated after the notify() call
|
|
obj = verify_obj[0]
|
|
assert obj['body'] == "Hello World"
|
|
assert obj['title'] == "My Title"
|
|
assert obj['notify_type'] == common.NotifyType.INFO
|
|
|
|
meta = obj['meta']
|
|
assert isinstance(meta, dict)
|
|
|
|
# No format was defined
|
|
assert 'body_format' in obj['kwargs']
|
|
assert obj['kwargs']['body_format'] is None
|
|
|
|
# The meta argument allows us to further parse the URL parameters
|
|
# specified
|
|
assert isinstance(obj['kwargs'], dict)
|
|
|
|
assert 'asset' in meta
|
|
assert isinstance(meta['asset'], AppriseAsset)
|
|
|
|
assert 'tag' in meta
|
|
assert isinstance(meta['tag'], set)
|
|
|
|
assert len(meta) == 8
|
|
# We carry all of our default arguments from the @notify's initialization
|
|
assert meta['schema'] == 'multi'
|
|
assert meta['host'] == 'hostname'
|
|
assert meta['user'] == 'user1'
|
|
assert meta['verify'] is True
|
|
assert meta['password'] == 'pass'
|
|
|
|
# Verify our URL is correct
|
|
assert meta['url'] == 'multi://user1:pass@hostname'
|
|
|
|
#
|
|
# Now verify our second URL saved correct
|
|
#
|
|
|
|
# Our content was populated after the notify() call
|
|
obj = verify_obj[1]
|
|
assert obj['body'] == "Hello World"
|
|
assert obj['title'] == "My Title"
|
|
assert obj['notify_type'] == common.NotifyType.INFO
|
|
|
|
meta = obj['meta']
|
|
assert isinstance(meta, dict)
|
|
|
|
# No format was defined
|
|
assert 'body_format' in obj['kwargs']
|
|
assert obj['kwargs']['body_format'] is None
|
|
|
|
# The meta argument allows us to further parse the URL parameters
|
|
# specified
|
|
assert isinstance(obj['kwargs'], dict)
|
|
|
|
assert 'asset' in meta
|
|
assert isinstance(meta['asset'], AppriseAsset)
|
|
|
|
assert 'tag' in meta
|
|
assert isinstance(meta['tag'], set)
|
|
|
|
assert len(meta) == 9
|
|
# We carry all of our default arguments from the @notify's initialization
|
|
assert meta['schema'] == 'multi'
|
|
assert meta['host'] == 'hostname'
|
|
assert meta['user'] == 'user2'
|
|
assert meta['password'] == 'pass2'
|
|
assert meta['verify'] is False
|
|
assert meta['qsd']['verify'] == 'no'
|
|
|
|
# Verify our URL is correct
|
|
assert meta['url'] == 'multi://user2:pass2@hostname?verify=no'
|
|
|
|
# Tidy
|
|
N_MGR.remove('multi')
|
|
|
|
|
|
def test_custom_notify_plugin_decoration():
|
|
"""decorators: CustomNotifyPlugin testing
|
|
"""
|
|
|
|
CustomNotifyPlugin()
|