mirror of
https://github.com/caronc/apprise.git
synced 2024-11-21 23:53:23 +01:00
398 lines
12 KiB
Python
398 lines
12 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright (C) 2020 Chris Caron <lead2gold@gmail.com>
|
|
# All rights reserved.
|
|
#
|
|
# This code is licensed under the MIT License.
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
# of this software and associated documentation files(the "Software"), to deal
|
|
# in the Software without restriction, including without limitation the rights
|
|
# to use, copy, modify, merge, publish, distribute, sublicense, and / or sell
|
|
# copies of the Software, and to permit persons to whom the Software is
|
|
# furnished to do so, subject to the following conditions :
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in
|
|
# all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.IN NO EVENT SHALL THE
|
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
# THE SOFTWARE.
|
|
|
|
import os
|
|
import six
|
|
import mock
|
|
import pytest
|
|
import requests
|
|
from datetime import datetime
|
|
from json import dumps
|
|
from apprise import Apprise
|
|
from apprise import plugins
|
|
from helpers import AppriseURLTester
|
|
|
|
# Disable logging for a cleaner testing output
|
|
import logging
|
|
logging.disable(logging.CRITICAL)
|
|
|
|
# Attachment Directory
|
|
TEST_VAR_DIR = os.path.join(os.path.dirname(__file__), 'var')
|
|
|
|
# Our Testing URLs
|
|
apprise_url_tests = (
|
|
##################################
|
|
# NotifyOffice365
|
|
##################################
|
|
('o365://', {
|
|
# Missing tenant, client_id, secret, and targets!
|
|
'instance': TypeError,
|
|
}),
|
|
('o365://:@/', {
|
|
# invalid url
|
|
'instance': TypeError,
|
|
}),
|
|
('o365://{tenant}:{aid}/{cid}/{secret}/{targets}'.format(
|
|
# invalid tenant
|
|
tenant=',',
|
|
cid='ab-cd-ef-gh',
|
|
aid='user@example.com',
|
|
secret='abcd/123/3343/@jack/test',
|
|
targets='/'.join(['email1@test.ca'])), {
|
|
|
|
# We're valid and good to go
|
|
'instance': TypeError,
|
|
}),
|
|
('o365://{tenant}:{aid}/{cid}/{secret}/{targets}'.format(
|
|
tenant='tenant',
|
|
# invalid client id
|
|
cid='ab.',
|
|
aid='user@example.com',
|
|
secret='abcd/123/3343/@jack/test',
|
|
targets='/'.join(['email1@test.ca'])), {
|
|
|
|
# We're valid and good to go
|
|
'instance': TypeError,
|
|
}),
|
|
('o365://{tenant}:{aid}/{cid}/{secret}/{targets}'.format(
|
|
tenant='tenant',
|
|
cid='ab-cd-ef-gh',
|
|
aid='user@example.com',
|
|
secret='abcd/123/3343/@jack/test',
|
|
targets='/'.join(['email1@test.ca'])), {
|
|
|
|
# We're valid and good to go
|
|
'instance': plugins.NotifyOffice365,
|
|
|
|
# Test what happens if a batch send fails to return a messageCount
|
|
'requests_response_text': {
|
|
'expires_in': 2000,
|
|
'access_token': 'abcd1234',
|
|
},
|
|
|
|
# Our expected url(privacy=True) startswith() response:
|
|
'privacy_url': 'o365://t...t:user@example.com/a...h/' \
|
|
'****/email1%40test.ca/'}),
|
|
# test our arguments
|
|
('o365://_/?oauth_id={cid}&oauth_secret={secret}&tenant={tenant}'
|
|
'&to={targets}&from={aid}'.format(
|
|
tenant='tenant',
|
|
cid='ab-cd-ef-gh',
|
|
aid='user@example.com',
|
|
secret='abcd/123/3343/@jack/test',
|
|
targets='email1@test.ca'),
|
|
{
|
|
# We're valid and good to go
|
|
'instance': plugins.NotifyOffice365,
|
|
|
|
# Test what happens if a batch send fails to return a messageCount
|
|
'requests_response_text': {
|
|
'expires_in': 2000,
|
|
'access_token': 'abcd1234',
|
|
},
|
|
|
|
# Our expected url(privacy=True) startswith() response:
|
|
'privacy_url': 'o365://t...t:user@example.com/a...h/' \
|
|
'****/email1%40test.ca/'}),
|
|
# Test invalid JSON (no tenant defaults to email domain)
|
|
('o365://{tenant}:{aid}/{cid}/{secret}/{targets}'.format(
|
|
tenant='tenant',
|
|
cid='ab-cd-ef-gh',
|
|
aid='user@example.com',
|
|
secret='abcd/123/3343/@jack/test',
|
|
targets='/'.join(['email1@test.ca'])), {
|
|
|
|
# We're valid and good to go
|
|
'instance': plugins.NotifyOffice365,
|
|
|
|
# invalid JSON response
|
|
'requests_response_text': '{',
|
|
'notify_response': False,
|
|
}),
|
|
# No Targets specified
|
|
('o365://{tenant}:{aid}/{cid}/{secret}'.format(
|
|
tenant='tenant',
|
|
cid='ab-cd-ef-gh',
|
|
aid='user@example.com',
|
|
secret='abcd/123/3343/@jack/test'), {
|
|
|
|
# We're valid and good to go
|
|
'instance': plugins.NotifyOffice365,
|
|
|
|
# There were no targets to notify; so we use our own email
|
|
'requests_response_text': {
|
|
'expires_in': 2000,
|
|
'access_token': 'abcd1234',
|
|
},
|
|
}),
|
|
('o365://{tenant}:{aid}/{cid}/{secret}/{targets}'.format(
|
|
tenant='tenant',
|
|
cid='zz-zz-zz-zz',
|
|
aid='user@example.com',
|
|
secret='abcd/abc/dcba/@john/test',
|
|
targets='/'.join(['email1@test.ca'])), {
|
|
'instance': plugins.NotifyOffice365,
|
|
# throw a bizzare code forcing us to fail to look it up
|
|
'response': False,
|
|
'requests_response_code': 999,
|
|
}),
|
|
('o365://{tenant}:{aid}/{cid}/{secret}/{targets}'.format(
|
|
tenant='tenant',
|
|
cid='01-12-23-34',
|
|
aid='user@example.com',
|
|
secret='abcd/321/4321/@test/test',
|
|
targets='/'.join(['email1@test.ca'])), {
|
|
'instance': plugins.NotifyOffice365,
|
|
# Throws a series of connection and transfer exceptions when this flag
|
|
# is set and tests that we gracfully handle them
|
|
'test_requests_exceptions': True,
|
|
}),
|
|
)
|
|
|
|
|
|
def test_plugin_office365_urls():
|
|
"""
|
|
NotifyOffice365() Apprise URLs
|
|
|
|
"""
|
|
|
|
# Run our general tests
|
|
AppriseURLTester(tests=apprise_url_tests).run_all()
|
|
|
|
|
|
@mock.patch('requests.post')
|
|
def test_plugin_office365_general(mock_post):
|
|
"""
|
|
NotifyOffice365() General Testing
|
|
|
|
"""
|
|
# Disable Throttling to speed testing
|
|
plugins.NotifyBase.request_rate_per_sec = 0
|
|
|
|
# Initialize some generic (but valid) tokens
|
|
email = 'user@example.net'
|
|
tenant = 'ff-gg-hh-ii-jj'
|
|
client_id = 'aa-bb-cc-dd-ee'
|
|
secret = 'abcd/1234/abcd@ajd@/test'
|
|
targets = 'target@example.com'
|
|
|
|
# Prepare Mock return object
|
|
authentication = {
|
|
"token_type": "Bearer",
|
|
"expires_in": 6000,
|
|
"access_token": "abcd1234"
|
|
}
|
|
response = mock.Mock()
|
|
response.content = dumps(authentication)
|
|
response.status_code = requests.codes.ok
|
|
mock_post.return_value = response
|
|
|
|
# Instantiate our object
|
|
obj = Apprise.instantiate(
|
|
'o365://{tenant}:{email}/{tenant}/{secret}/{targets}'.format(
|
|
tenant=tenant,
|
|
email=email,
|
|
secret=secret,
|
|
targets=targets))
|
|
|
|
assert isinstance(obj, plugins.NotifyOffice365)
|
|
|
|
# Test our URL generation
|
|
assert isinstance(obj.url(), six.string_types)
|
|
|
|
# Test our notification
|
|
assert obj.notify(title='title', body='test') is True
|
|
|
|
# Instantiate our object
|
|
obj = Apprise.instantiate(
|
|
'o365://{tenant}:{email}/{tenant}/{secret}/{targets}'
|
|
'?bcc={bcc}&cc={cc}'.format(
|
|
tenant=tenant,
|
|
email=email,
|
|
secret=secret,
|
|
targets=targets,
|
|
# Test the cc and bcc list (use good and bad email)
|
|
cc='Chuck Norris cnorris@yahoo.ca, Sauron@lotr.me, invalid@!',
|
|
bcc='Bruce Willis bwillis@hotmail.com, Frodo@lotr.me invalid@!',
|
|
))
|
|
|
|
assert isinstance(obj, plugins.NotifyOffice365)
|
|
|
|
# Test our URL generation
|
|
assert isinstance(obj.url(), six.string_types)
|
|
|
|
# Test our notification
|
|
assert obj.notify(title='title', body='test') is True
|
|
|
|
with pytest.raises(TypeError):
|
|
# No secret
|
|
plugins.NotifyOffice365(
|
|
email=email,
|
|
client_id=client_id,
|
|
tenant=tenant,
|
|
secret=None,
|
|
targets=None,
|
|
)
|
|
|
|
with pytest.raises(TypeError):
|
|
# Invalid email
|
|
plugins.NotifyOffice365(
|
|
email=None,
|
|
client_id=client_id,
|
|
tenant=tenant,
|
|
secret=secret,
|
|
targets=None,
|
|
)
|
|
|
|
with pytest.raises(TypeError):
|
|
# Invalid email
|
|
plugins.NotifyOffice365(
|
|
email='garbage',
|
|
client_id=client_id,
|
|
tenant=tenant,
|
|
secret=secret,
|
|
targets=None,
|
|
)
|
|
|
|
# One of the targets are invalid
|
|
obj = plugins.NotifyOffice365(
|
|
email=email,
|
|
client_id=client_id,
|
|
tenant=tenant,
|
|
secret=secret,
|
|
targets=('Management abc@gmail.com', 'garbage'),
|
|
)
|
|
# Test our notification (this will work and only notify abc@gmail.com)
|
|
assert obj.notify(title='title', body='test') is True
|
|
|
|
# all of the targets are invalid
|
|
obj = plugins.NotifyOffice365(
|
|
email=email,
|
|
client_id=client_id,
|
|
tenant=tenant,
|
|
secret=secret,
|
|
targets=('invalid', 'garbage'),
|
|
)
|
|
|
|
# Test our notification (which will fail because of no entries)
|
|
assert obj.notify(title='title', body='test') is False
|
|
|
|
|
|
@mock.patch('requests.post')
|
|
def test_plugin_office365_authentication(mock_post):
|
|
"""
|
|
NotifyOffice365() Authentication Testing
|
|
|
|
"""
|
|
# Disable Throttling to speed testing
|
|
plugins.NotifyBase.request_rate_per_sec = 0
|
|
|
|
# Initialize some generic (but valid) tokens
|
|
tenant = 'ff-gg-hh-ii-jj'
|
|
email = 'user@example.net'
|
|
client_id = 'aa-bb-cc-dd-ee'
|
|
secret = 'abcd/1234/abcd@ajd@/test'
|
|
targets = 'target@example.com'
|
|
|
|
# Prepare Mock return object
|
|
authentication_okay = {
|
|
"token_type": "Bearer",
|
|
"expires_in": 6000,
|
|
"access_token": "abcd1234"
|
|
}
|
|
authentication_failure = {
|
|
"error": "invalid_scope",
|
|
"error_description": "AADSTS70011: Blah... Blah Blah... Blah",
|
|
"error_codes": [70011],
|
|
"timestamp": "2020-01-09 02:02:12Z",
|
|
"trace_id": "255d1aef-8c98-452f-ac51-23d051240864",
|
|
"correlation_id": "fb3d2015-bc17-4bb9-bb85-30c5cf1aaaa7",
|
|
}
|
|
response = mock.Mock()
|
|
response.content = dumps(authentication_okay)
|
|
response.status_code = requests.codes.ok
|
|
mock_post.return_value = response
|
|
|
|
# Instantiate our object
|
|
obj = Apprise.instantiate(
|
|
'o365://{tenant}:{email}/{client_id}/{secret}/{targets}'.format(
|
|
client_id=client_id,
|
|
tenant=tenant,
|
|
email=email,
|
|
secret=secret,
|
|
targets=targets))
|
|
|
|
assert isinstance(obj, plugins.NotifyOffice365)
|
|
|
|
# Authenticate
|
|
assert obj.authenticate() is True
|
|
|
|
# We're already authenticated
|
|
assert obj.authenticate() is True
|
|
|
|
# Expire our token
|
|
obj.token_expiry = datetime.now()
|
|
|
|
# Re-authentiate
|
|
assert obj.authenticate() is True
|
|
|
|
# Change our response
|
|
response.status_code = 400
|
|
|
|
# We'll fail to send a notification now...
|
|
assert obj.notify(title='title', body='test') is False
|
|
|
|
# Expire our token
|
|
obj.token_expiry = datetime.now()
|
|
|
|
# Set a failure response
|
|
response.content = dumps(authentication_failure)
|
|
|
|
# We will fail to authenticate at this point
|
|
assert obj.authenticate() is False
|
|
|
|
# Notifications will also fail in this case
|
|
assert obj.notify(title='title', body='test') is False
|
|
|
|
# We will fail to authenticate with invalid data
|
|
|
|
invalid_auth_entries = authentication_okay.copy()
|
|
invalid_auth_entries['expires_in'] = 'garbage'
|
|
response.content = dumps(invalid_auth_entries)
|
|
response.status_code = requests.codes.ok
|
|
assert obj.authenticate() is False
|
|
|
|
invalid_auth_entries['expires_in'] = None
|
|
response.content = dumps(invalid_auth_entries)
|
|
assert obj.authenticate() is False
|
|
|
|
invalid_auth_entries['expires_in'] = ''
|
|
response.content = dumps(invalid_auth_entries)
|
|
assert obj.authenticate() is False
|
|
|
|
del invalid_auth_entries['expires_in']
|
|
response.content = dumps(invalid_auth_entries)
|
|
assert obj.authenticate() is False
|