apprise/test/test_rest_plugins.py

4208 lines
137 KiB
Python
Raw Normal View History

2017-12-04 03:48:23 +01:00
# -*- coding: utf-8 -*-
# BSD 2-Clause License
2017-12-04 03:48:23 +01:00
#
# Apprise - Push Notification Library.
2024-01-27 21:35:11 +01:00
# Copyright (c) 2024, Chris Caron <lead2gold@gmail.com>
2017-12-04 03:48:23 +01:00
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
2017-12-04 03:48:23 +01:00
#
# 1. Redistributions of source code must retain the above copyright notice,
# this list of conditions and the following disclaimer.
2017-12-04 03:48:23 +01:00
#
# 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.
2017-12-04 03:48:23 +01:00
#
# 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.
2017-12-04 03:48:23 +01:00
import pytest
2023-12-16 03:59:58 +01:00
import json
from unittest import mock
import requests
from random import choice
from string import ascii_uppercase as str_alpha
from string import digits as str_num
from apprise import NotifyBase
from apprise import NotifyFormat
from apprise import OverflowMode
from apprise import AppriseAsset
from apprise import Apprise
# Disable logging for a cleaner testing output
import logging
logging.disable(logging.CRITICAL)
2017-12-04 03:48:23 +01:00
2017-12-25 21:07:41 +01:00
def test_notify_overflow_truncate_with_amalgamation():
"""
API: Overflow With Amalgamation Truncate Functionality Testing
"""
#
# A little preparation
#
# Number of characters per line
row = 24
# Some variables we use to control the data we work with
body_len = 1024
title_len = 1024
# Create a large body and title with random data
body = ''.join(choice(str_alpha + str_num + ' ') for _ in range(body_len))
body = '\r\n'.join([body[i: i + row] for i in range(0, len(body), row)])
# the new lines add a large amount to our body; lets force the content
# back to being 1024 characters.
body = body[0:1024]
# Create our title using random data
title = ''.join(choice(str_alpha + str_num) for _ in range(title_len))
#
# First Test: Truncated Title
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 10
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# We should throw an exception because our specified overflow is wrong.
with pytest.raises(TypeError):
# Load our object
obj = TestNotification(overflow='invalid')
# Load our object
obj = TestNotification(overflow=OverflowMode.TRUNCATE)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title, overflow=None)
chunks = obj._apply_overflow(body="", title="", overflow=None)
chunks = obj._apply_overflow(body=body, title="", overflow=None)
chunks = obj._apply_overflow(body=body, title=title, overflow=None)
chunks = obj._apply_overflow(
body=body, title=title, overflow=OverflowMode.SPLIT)
assert len(chunks) == 1
assert body.lstrip('\r\n\x0b\x0c').rstrip() == chunks[0].get('body')
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Line Count Control
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 5
# Maximum number of lines
body_max_line_count = 5
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.TRUNCATE)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert len(chunks[0].get('body').split('\n')) == obj.body_max_line_count
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Truncated body
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length of just 10
body_maxlen = 10
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.TRUNCATE)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
# Body is lost as the title prevails over everything
assert chunks[0].get('body') == ''
# Title is not longer then the maximum size the body can be due to
# amalgamationflag:
assert title[:obj.body_maxlen] == chunks[0].get('title')
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length of just 10
body_maxlen = 10
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.TRUNCATE)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert chunks[0].get('body') == ''
# body_maxlen prevails due to it being smaller and amalgamation flag set
assert title[0:obj.body_maxlen] == chunks[0].get('title')
#
# Next Test: Append title to body + Truncated body
#
class TestNotification(NotifyBase):
# Enforce no title
title_maxlen = 0
# Enforce a body length of just 100
body_maxlen = 100
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.TRUNCATE)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
2019-03-10 17:10:01 +01:00
obj.notify_format = NotifyFormat.HTML
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
2019-03-10 17:10:01 +01:00
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
obj.notify_format = NotifyFormat.MARKDOWN
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
2019-03-10 17:10:01 +01:00
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
obj.notify_format = NotifyFormat.TEXT
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
# The below line should be read carefully... We're actually testing to see
# that our title is matched against our body. Behind the scenes, the title
# was appended to the body. The body was then truncated to the maxlen.
# The thing is, since the title is so large, all of the body was lost
# and a good chunk of the title was too. The message sent will just be a
# small portion of the title
assert len(chunks[0].get('body')) == obj.body_maxlen
assert title[0:obj.body_maxlen] == chunks[0].get('body')
def test_notify_overflow_truncate_no_amalgamation():
"""
API: Overflow No Amalgamation Truncate Functionality Testing
"""
#
# A little preparation
#
# Number of characters per line
row = 24
# Some variables we use to control the data we work with
body_len = 1024
title_len = 1024
# Create a large body and title with random data
body = ''.join(choice(str_alpha + str_num + ' ') for _ in range(body_len))
body = '\r\n'.join([body[i: i + row] for i in range(0, len(body), row)])
# the new lines add a large amount to our body; lets force the content
# back to being 1024 characters.
body = body[0:1024]
# Create our title using random data
title = ''.join(choice(str_alpha + str_num) for _ in range(title_len))
#
# First Test: Truncated Title
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 10
# No amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# We should throw an exception because our specified overflow is wrong.
with pytest.raises(TypeError):
# Load our object
obj = TestNotification(overflow='invalid')
# Load our object
obj = TestNotification(overflow=OverflowMode.TRUNCATE)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title, overflow=None)
chunks = obj._apply_overflow(body="", title="", overflow=None)
chunks = obj._apply_overflow(body=body, title="", overflow=None)
chunks = obj._apply_overflow(body=body, title=title, overflow=None)
chunks = obj._apply_overflow(
body=body, title=title, overflow=OverflowMode.SPLIT)
assert len(chunks) == 1
assert body.lstrip('\r\n\x0b\x0c').rstrip() == chunks[0].get('body')
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Line Count Control
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 5
# Maximum number of lines
body_max_line_count = 5
# No amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.TRUNCATE)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert len(chunks[0].get('body').split('\n')) == obj.body_max_line_count
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Truncated body
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length of just 10
body_maxlen = 10
# No amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.TRUNCATE)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert body[0:obj.body_maxlen]\
.lstrip('\r\n\x0b\x0c').rstrip() == chunks[0].get('body')
assert title == chunks[0].get('title')
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length of just 10
body_maxlen = 10
# No amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.TRUNCATE)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
# No amalgamation set so our body aligns in size (no -2 like previous
# test)
assert body[0:obj.body_maxlen]\
.lstrip('\r\n\x0b\x0c').rstrip() == chunks[0].get('body')
assert title == chunks[0].get('title')
#
# Next Test: Append title to body + Truncated body
#
class TestNotification(NotifyBase):
# Enforce no title
title_maxlen = 0
# Enforce a body length of just 100
body_maxlen = 100
# No amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.TRUNCATE)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
obj.notify_format = NotifyFormat.HTML
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
obj.notify_format = NotifyFormat.MARKDOWN
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
obj.notify_format = NotifyFormat.TEXT
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
# The below line should be read carefully... We're actually testing to see
# that our title is matched against our body. Behind the scenes, the title
# was appended to the body. The body was then truncated to the maxlen.
# The thing is, since the title is so large, all of the body was lost
# and a good chunk of the title was too. The message sent will just be a
# small portion of the title
assert len(chunks[0].get('body')) == obj.body_maxlen
assert title[0:obj.body_maxlen] == chunks[0].get('body')
def test_notify_overflow_split_with_amalgamation():
"""
API: Overflow With Amalgamation Splits Functionality Testing
"""
#
# A little preparation
#
# Number of characters per line
row = 24
# Some variables we use to control the data we work with
body_len = 1024
title_len = 1024
# Create a large body and title with random data
body = ''.join(choice(str_alpha + str_num) for _ in range(body_len))
body = '\r\n'.join([body[i: i + row] for i in range(0, len(body), row)])
# the new lines add a large amount to our body; lets force the content
# back to being 1024 characters.
body = body[0:1024]
# Create our title using random data
title = ''.join(choice(str_alpha + str_num) for _ in range(title_len))
#
# First Test: Truncated Title
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 10
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert body.lstrip('\r\n\x0b\x0c').rstrip() == chunks[0].get('body')
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Line Count Control
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 5
# Maximum number of lines
body_max_line_count = 5
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert len(chunks[0].get('body').split('\n')) == obj.body_max_line_count
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Split body
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# No counter is displayed because our title is so enormous
# We switch to a display title on first message only
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The length of the body prevails our title due to it being
# so much smaller then our title length
assert len(chunk.get('title')) == obj.body_maxlen
assert title[:obj.body_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
# Another edge case where the title just isn't that long leaving
# a lot of space for the [xx/xx] entries (no truncation needed)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title[:20])
offset = 0
c_len = len(' [X/X]')
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has a counter added to it
assert title[:20] == chunk.get('title')[:-c_len]
assert chunk.get('title')[-c_len:] == \
' [{:01}/{:01}]'.format(idx, len(chunks))
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
#
# Test forcing overflow_display_title_once
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Only display title once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The body length prevails due to our amalgamation flag
assert len(chunk.get('title')) == obj.body_maxlen
assert title[:obj.body_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
# Test larger messages
# and that the body remains untouched
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 150
# Enforce a body length. Make sure it's an int.
body_maxlen = 400
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
new_body = (body * 500)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
c_len = len(' [XXXX/XXXX]')
assert len(chunks) == 2048
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has a counter added to it
assert title[:obj.title_maxlen][:-c_len] == chunk.get('title')[:-c_len]
assert chunk.get('title')[-c_len:] == \
' [{:04}/{:04}]'.format(idx, len(chunks))
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
# Body chunk is beyond 4 digits, so [XXXX/XXXX] is turned off
new_body = (body * 2500)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 10240
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
# Test larger messages
# and that the body remains untouched
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 150
# Enforce a body length. Make sure it's an int.
body_maxlen = 150
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
new_body = (body * 5)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 36
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# Because 150 is what we set the title limit to
assert len(chunk.get('title')) == obj.title_maxlen
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Next Test: Append title to body + split body
#
class TestNotification(NotifyBase):
# Enforce no title
title_maxlen = 0
# Enforce a body length based on the title. Make sure it's an int.
body_maxlen = int(title_len / 4)
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
# Our final product is that our title has been appended to our body to
# create one great big body. As a result we'll get quite a few lines back
# now.
offset = 0
# Our body will look like this in small chunks at the end of the day
bulk = title + '\r\n' + body
# Due to the new line added to the end
assert len(chunks) == (
# wrap division in int() so Python 3 doesn't convert it to a float on
# us
int(len(bulk) / obj.body_maxlen) +
(1 if len(bulk) % obj.body_maxlen else 0))
for chunk in chunks:
# Verification
assert len(chunk.get('title')) == 0
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title is empty every time
assert chunk.get('title') == ''
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(bulk[offset: len(_body) + offset]) - \
len(bulk[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert bulk[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Test case where our title_len is shorter then the value
# that would otherwise trigger the [XX/XX] elements
#
class TestNotification(NotifyBase):
# Set a small title length
title_maxlen = 100
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 7
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title is truncated and no counter added
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(body[offset: len(_body) + offset]) - \
len(body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Scenario where the title length is larger than the body
#
class TestNotification(NotifyBase):
# Set a small title length
title_maxlen = 100
# Enforce a body length. Make sure it's an int.
body_maxlen = 50
# With amalgamation
overflow_amalgamate_title = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 22
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it due to it's length
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# Because 150 is what we set the title limit to 50 due
# to amalamation. The lowest value always prevails
assert len(chunk.get('title')) == obj.body_maxlen
assert title[:obj.body_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(body[offset: len(_body) + offset]) - \
len(body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
def test_notify_overflow_split_with_amalgamation_force_title_always():
"""
API: Overflow With Amalgamation (title alaways Split Functionality Testing
"""
#
# A little preparation
#
# Number of characters per line
row = 24
# Some variables we use to control the data we work with
body_len = 1024
title_len = 1024
# Create a large body and title with random data
body = ''.join(choice(str_alpha + str_num) for _ in range(body_len))
body = '\r\n'.join([body[i: i + row] for i in range(0, len(body), row)])
# the new lines add a large amount to our body; lets force the content
# back to being 1024 characters.
body = body[0:1024]
# Create our title using random data
title = ''.join(choice(str_alpha + str_num) for _ in range(title_len))
#
# First Test: Truncated Title
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 10
# With amalgamation
overflow_amalgamate_title = True
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert body.lstrip('\r\n\x0b\x0c').rstrip() == chunks[0].get('body')
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Line Count Control
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 5
# Maximum number of lines
body_max_line_count = 5
# With amalgamation
overflow_amalgamate_title = True
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert len(chunks[0].get('body').split('\n')) == obj.body_max_line_count
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Split body
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# No counter is displayed because our title is so enormous
# We switch to a display title on first message only
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The length of the body prevails our title due to it being
# so much smaller then our title length
assert len(chunk.get('title')) == obj.body_maxlen
assert title[:obj.body_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
# Another edge case where the title just isn't that long leaving
# a lot of space for the [xx/xx] entries (no truncation needed)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title[:20])
offset = 0
c_len = len(' [X/X]')
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has a counter added to it
assert title[:20] == chunk.get('title')[:-c_len]
assert chunk.get('title')[-c_len:] == \
' [{:01}/{:01}]'.format(idx, len(chunks))
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
#
# Test forcing overflow_display_title_once
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The body length prevails due to our amalgamation flag
assert len(chunk.get('title')) == obj.body_maxlen
assert title[:obj.body_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
# Test larger messages
# and that the body remains untouched
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 150
# Enforce a body length. Make sure it's an int.
body_maxlen = 400
# With amalgamation
overflow_amalgamate_title = True
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
new_body = (body * 500)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
c_len = len(' [XXXX/XXXX]')
assert len(chunks) == 2048
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has a counter added to it
assert title[:obj.title_maxlen][:-c_len] == chunk.get('title')[:-c_len]
assert chunk.get('title')[-c_len:] == \
' [{:04}/{:04}]'.format(idx, len(chunks))
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
# Body chunk is beyond 4 digits, so [XXXX/XXXX] is turned off
new_body = (body * 2500)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 10240
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
# Test larger messages
# and that the body remains untouched
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 150
# Enforce a body length. Make sure it's an int.
body_maxlen = 150
# With amalgamation
overflow_amalgamate_title = True
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
new_body = (body * 5)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
# overflow_display_title_once whle set to False is still ignored
# because our title_maxlen > body_maxlen and a full title was
# provided
assert len(chunks) == 36
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# Because 150 is what we set the title limit to
assert len(chunk.get('title')) == obj.title_maxlen
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
# overflow_display_title_once whle set to False is still ignored
# because our title_maxlen > body_maxlen and a full title was
# provided
assert len(chunks) == 36
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# Because 150 is what we set the title limit to
assert len(chunk.get('title')) == obj.title_maxlen
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Run again but with a smaller title
#
chunks = obj._apply_overflow(body=new_body, title=title[:30])
offset = 0
# overflow_display_title_once whle set to False is still ignored
# because our body_maxlen (after title has been calculated with it)
# is less then the overflow_display_count_threshold; hence a message
# must be a certain minimum size in order to kick in
assert len(chunks) == 48
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title doesn't change
assert len(chunk.get('title')) == 30
assert title[:30] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset].rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Next Test: Append title to body + split body
#
class TestNotification(NotifyBase):
# Enforce no title
title_maxlen = 0
# Enforce a body length based on the title. Make sure it's an int.
body_maxlen = int(title_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
# Our final product is that our title has been appended to our body to
# create one great big body. As a result we'll get quite a few lines back
# now.
offset = 0
# Our body will look like this in small chunks at the end of the day
bulk = title + '\r\n' + body
# Due to the new line added to the end
assert len(chunks) == (
# wrap division in int() so Python 3 doesn't convert it to a float on
# us
int(len(bulk) / obj.body_maxlen) +
(1 if len(bulk) % obj.body_maxlen else 0))
for chunk in chunks:
# Verification
assert len(chunk.get('title')) == 0
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title is empty every time
assert chunk.get('title') == ''
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(bulk[offset: len(_body) + offset]) - \
len(bulk[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert bulk[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Test case where our title_len is shorter then the value
# that would otherwise trigger the [XX/XX] elements
#
class TestNotification(NotifyBase):
# Set a small title length
title_maxlen = 100
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 7
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title is truncated and no counter added
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(body[offset: len(_body) + offset]) - \
len(body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Scenario where the title length is larger than the body
#
class TestNotification(NotifyBase):
# Set a small title length
title_maxlen = 100
# Enforce a body length. Make sure it's an int.
body_maxlen = 50
# With amalgamation
overflow_amalgamate_title = True
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
# overflow_display_title_once whle set to False is still ignored
# because our title_maxlen > body_maxlen and a full title was
# provided
assert len(chunks) == 22
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it due to it's length
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# Because 150 is what we set the title limit to 50 due
# to amalamation. The lowest value always prevails
assert len(chunk.get('title')) == obj.body_maxlen
assert title[:obj.body_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(body[offset: len(_body) + offset]) - \
len(body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
def test_notify_overflow_split_with_amalgamation_force_title_once():
"""
API: Overflow With Amalgamation (title once) Split Functionality Testing
"""
#
# A little preparation
#
# Number of characters per line
row = 24
# Some variables we use to control the data we work with
body_len = 1024
title_len = 1024
# Create a large body and title with random data
body = ''.join(choice(str_alpha + str_num) for _ in range(body_len))
body = '\r\n'.join([body[i: i + row] for i in range(0, len(body), row)])
# the new lines add a large amount to our body; lets force the content
# back to being 1024 characters.
body = body[0:1024]
# Create our title using random data
title = ''.join(choice(str_alpha + str_num) for _ in range(title_len))
#
# First Test: Truncated Title
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 10
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert body.lstrip('\r\n\x0b\x0c').rstrip() == chunks[0].get('body')
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Line Count Control
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 5
# Maximum number of lines
body_max_line_count = 5
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert len(chunks[0].get('body').split('\n')) == obj.body_max_line_count
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Split body
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# No counter is displayed because our title is so enormous
# We switch to a display title on first message only
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The length of the body prevails our title due to it being
# so much smaller then our title length
assert len(chunk.get('title')) == obj.body_maxlen
assert title[:obj.body_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
# Another edge case where the title just isn't that long leaving
# a lot of space for the [xx/xx] entries (no truncation needed)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title[:20])
offset = 0
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The body length prevails due to our amalgamation flag
assert len(chunk.get('title')) == 20
assert title[:20] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
#
# Test forcing overflow_display_title_once
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Only display title once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The body length prevails due to our amalgamation flag
assert len(chunk.get('title')) == obj.body_maxlen
assert title[:obj.body_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
# Test larger messages
# and that the body remains untouched
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 150
# Enforce a body length. Make sure it's an int.
body_maxlen = 400
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
new_body = (body * 500)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 1281
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The body length prevails due to our amalgamation flag
assert len(chunk.get('title')) == obj.title_maxlen
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
# Body chunk is beyond 4 digits, so [XXXX/XXXX] is turned off
new_body = (body * 2500)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 6401
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The body length prevails due to our amalgamation flag
assert len(chunk.get('title')) == obj.title_maxlen
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
# Test larger messages
# and that the body remains untouched
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 150
# Enforce a body length. Make sure it's an int.
body_maxlen = 150
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
new_body = (body * 5)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 36
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# Because 150 is what we set the title limit to
assert len(chunk.get('title')) == obj.title_maxlen
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Next Test: Append title to body + split body
#
class TestNotification(NotifyBase):
# Enforce no title
title_maxlen = 0
# Enforce a body length based on the title. Make sure it's an int.
body_maxlen = int(title_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
# Our final product is that our title has been appended to our body to
# create one great big body. As a result we'll get quite a few lines back
# now.
offset = 0
# Our body will look like this in small chunks at the end of the day
bulk = title + '\r\n' + body
# Due to the new line added to the end
assert len(chunks) == (
# wrap division in int() so Python 3 doesn't convert it to a float on
# us
int(len(bulk) / obj.body_maxlen) +
(1 if len(bulk) % obj.body_maxlen else 0))
for chunk in chunks:
# Verification
assert len(chunk.get('title')) == 0
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title is empty every time
assert chunk.get('title') == ''
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(bulk[offset: len(_body) + offset]) - \
len(bulk[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert bulk[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Test case where our title_len is shorter then the value
# that would otherwise trigger the [XX/XX] elements
#
class TestNotification(NotifyBase):
# Set a small title length
title_maxlen = 100
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The body length prevails due to our amalgamation flag
assert len(chunk.get('title')) == obj.title_maxlen
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(body[offset: len(_body) + offset]) - \
len(body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Scenario where the title length is larger than the body
#
class TestNotification(NotifyBase):
# Set a small title length
title_maxlen = 100
# Enforce a body length. Make sure it's an int.
body_maxlen = 50
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 22
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it due to it's length
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# Because 150 is what we set the title limit to 50 due
# to amalamation. The lowest value always prevails
assert len(chunk.get('title')) == obj.body_maxlen
assert title[:obj.body_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(body[offset: len(_body) + offset]) - \
len(body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
def test_notify_overflow_split_no_amalgamation():
"""
API: Overflow No Amalgamation Splits Functionality Testing
"""
#
# A little preparation
#
# Number of characters per line
row = 24
# Some variables we use to control the data we work with
body_len = 1024
title_len = 1024
# Create a large body and title with random data
body = ''.join(choice(str_alpha + str_num) for _ in range(body_len))
body = '\r\n'.join([body[i: i + row] for i in range(0, len(body), row)])
# the new lines add a large amount to our body; lets force the content
# back to being 1024 characters.
body = body[0:1024]
# Create our title using random data
title = ''.join(choice(str_alpha + str_num) for _ in range(title_len))
#
# First Test: Truncated Title
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 10
# No amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert body.lstrip('\r\n\x0b\x0c').rstrip() == chunks[0].get('body')
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Line Count Control
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 5
# Maximum number of lines
body_max_line_count = 5
# No amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert len(chunks[0].get('body').split('\n')) == obj.body_max_line_count
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Split body
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# No amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
c_len = len(' [X/X]')
assert len(chunks) == 4
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has a counter added to it
assert title[:-c_len] == chunk.get('title')[:-c_len]
assert chunk.get('title')[-c_len:] == \
' [{:01}/{:01}]'.format(idx, len(chunks))
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
# Another edge case where the title just isn't that long leaving
# a lot of space for the [xx/xx] entries (no truncation needed)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title[:20])
offset = 0
assert len(chunks) == 4
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has a counter added to it
assert title[:20] == chunk.get('title')[:-c_len]
assert chunk.get('title')[-c_len:] == \
' [{:01}/{:01}]'.format(idx, len(chunks))
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
# Test larger messages
# and that the body remains untouched
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 150
# Enforce a body length. Make sure it's an int.
body_maxlen = 400
# No amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
new_body = (body * 500)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
c_len = len(' [XXXX/XXXX]')
assert len(chunks) == 1280
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has a counter added to it
assert title[:obj.title_maxlen][:-c_len] == chunk.get('title')[:-c_len]
assert chunk.get('title')[-c_len:] == \
' [{:04}/{:04}]'.format(idx, len(chunks))
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
# Body chunk is beyond 4 digits, so [XXXX/XXXX] is turned off
new_body = (body * 4500)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 11520
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
# Test larger messages
# and that the body remains untouched
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 150
# Enforce a body length. Make sure it's an int.
body_maxlen = 150
# No amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
new_body = (body * 5)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 35
c_len = len(' [XX/XX]')
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has a counter added to it
assert title[:obj.title_maxlen][:-c_len] == chunk.get('title')[:-c_len]
assert chunk.get('title')[-c_len:] == \
' [{:02}/{:02}]'.format(idx, len(chunks))
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Next Test: Append title to body + split body
#
class TestNotification(NotifyBase):
# Enforce no title
title_maxlen = 0
# Enforce a body length based on the title. Make sure it's an int.
body_maxlen = int(title_len / 4)
# No Amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
# Our final product is that our title has been appended to our body to
# create one great big body. As a result we'll get quite a few lines back
# now.
offset = 0
# Our body will look like this in small chunks at the end of the day
bulk = title + '\r\n' + body
# Due to the new line added to the end
assert len(chunks) == (
# wrap division in int() so Python 3 doesn't convert it to a float on
# us
int(len(bulk) / obj.body_maxlen) +
(1 if len(bulk) % obj.body_maxlen else 0))
for chunk in chunks:
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title is empty every time
assert chunk.get('title') == ''
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(bulk[offset: len(_body) + offset]) - \
len(bulk[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert bulk[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Test case where our title_len is shorter then the value
# that would otherwise trigger the [XX/XX] elements
#
class TestNotification(NotifyBase):
# Set a small title length
title_maxlen = 100
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# No Amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 4
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title is truncated and no counter added
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
#
# Scenario where the title length is larger than the body
#
class TestNotification(NotifyBase):
# Set a small title length
title_maxlen = 100
# Enforce a body length. Make sure it's an int.
body_maxlen = 50
# No Amalgamation
overflow_amalgamate_title = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 21
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title is truncated and no counter added
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(body[offset: len(_body) + offset]) - \
len(body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
def test_notify_overflow_split_no_amalgamation_force_title_always():
"""
API: Overflow No Amalgamation (title always) Split Functionality Testing
"""
#
# A little preparation
#
# Number of characters per line
row = 24
# Some variables we use to control the data we work with
body_len = 1024
title_len = 1024
# Create a large body and title with random data
body = ''.join(choice(str_alpha + str_num) for _ in range(body_len))
body = '\r\n'.join([body[i: i + row] for i in range(0, len(body), row)])
# the new lines add a large amount to our body; lets force the content
# back to being 1024 characters.
body = body[0:1024]
# Create our title using random data
title = ''.join(choice(str_alpha + str_num) for _ in range(title_len))
#
# First Test: Truncated Title
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 10
# No amalgamation
overflow_amalgamate_title = False
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert body.lstrip('\r\n\x0b\x0c').rstrip() == chunks[0].get('body')
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Line Count Control
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 5
# Maximum number of lines
body_max_line_count = 5
# No amalgamation
overflow_amalgamate_title = False
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert len(chunks[0].get('body').split('\n')) == obj.body_max_line_count
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Split body
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# No amalgamation
overflow_amalgamate_title = False
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
c_len = len(' [X/X]')
assert len(chunks) == 4
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has a counter added to it
assert title[:-c_len] == chunk.get('title')[:-c_len]
assert chunk.get('title')[-c_len:] == \
' [{:01}/{:01}]'.format(idx, len(chunks))
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
# Another edge case where the title just isn't that long leaving
# a lot of space for the [xx/xx] entries (no truncation needed)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title[:20])
offset = 0
assert len(chunks) == 4
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has a counter added to it
assert title[:20] == chunk.get('title')[:-c_len]
assert chunk.get('title')[-c_len:] == \
' [{:01}/{:01}]'.format(idx, len(chunks))
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
# Test larger messages
# and that the body remains untouched
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 150
# Enforce a body length. Make sure it's an int.
body_maxlen = 400
# No amalgamation
overflow_amalgamate_title = False
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
new_body = (body * 500)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
c_len = len(' [XXXX/XXXX]')
assert len(chunks) == 1280
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has a counter added to it
assert title[:obj.title_maxlen][:-c_len] == chunk.get('title')[:-c_len]
assert chunk.get('title')[-c_len:] == \
' [{:04}/{:04}]'.format(idx, len(chunks))
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
# Body chunk is beyond 4 digits, so [XXXX/XXXX] is turned off
new_body = (body * 4500)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 11520
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
# Test larger messages
# and that the body remains untouched
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 150
# Enforce a body length. Make sure it's an int.
body_maxlen = 150
# No amalgamation
overflow_amalgamate_title = False
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
new_body = (body * 5)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 35
c_len = len(' [XX/XX]')
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has a counter added to it
assert title[:obj.title_maxlen][:-c_len] == chunk.get('title')[:-c_len]
assert chunk.get('title')[-c_len:] == \
' [{:02}/{:02}]'.format(idx, len(chunks))
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Next Test: Append title to body + split body
#
class TestNotification(NotifyBase):
# Enforce no title
title_maxlen = 0
# Enforce a body length based on the title. Make sure it's an int.
body_maxlen = int(title_len / 4)
# No Amalgamation
overflow_amalgamate_title = False
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
# Our final product is that our title has been appended to our body to
# create one great big body. As a result we'll get quite a few lines back
# now.
offset = 0
# Our body will look like this in small chunks at the end of the day
bulk = title + '\r\n' + body
# Due to the new line added to the end
assert len(chunks) == (
# wrap division in int() so Python 3 doesn't convert it to a float on
# us
int(len(bulk) / obj.body_maxlen) +
(1 if len(bulk) % obj.body_maxlen else 0))
for chunk in chunks:
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title is empty every time
assert chunk.get('title') == ''
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(bulk[offset: len(_body) + offset]) - \
len(bulk[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert bulk[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Test case where our title_len is shorter then the value
# that would otherwise trigger the [XX/XX] elements
#
class TestNotification(NotifyBase):
# Set a small title length
title_maxlen = 100
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# No Amalgamation
overflow_amalgamate_title = False
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 4
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title is truncated and no counter added
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
#
# Scenario where the title length is larger than the body
#
class TestNotification(NotifyBase):
# Set a small title length
title_maxlen = 100
# Enforce a body length. Make sure it's an int.
body_maxlen = 50
# No Amalgamation
overflow_amalgamate_title = False
# Force title to be displayed always
overflow_display_title_once = False
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 21
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title is truncated and no counter added
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(body[offset: len(_body) + offset]) - \
len(body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
def test_notify_overflow_split_no_amalgamation_force_title_once():
"""
API: Overflow No Amalgamation (title once) Split Functionality Testing
"""
#
# A little preparation
#
# Number of characters per line
row = 24
# Some variables we use to control the data we work with
body_len = 1024
title_len = 1024
# Create a large body and title with random data
body = ''.join(choice(str_alpha + str_num) for _ in range(body_len))
body = '\r\n'.join([body[i: i + row] for i in range(0, len(body), row)])
# the new lines add a large amount to our body; lets force the content
# back to being 1024 characters.
body = body[0:1024]
# Create our title using random data
title = ''.join(choice(str_alpha + str_num) for _ in range(title_len))
#
# First Test: Truncated Title
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 10
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert body.lstrip('\r\n\x0b\x0c').rstrip() == chunks[0].get('body')
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Line Count Control
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 5
# Maximum number of lines
body_max_line_count = 5
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
assert len(chunks[0].get('body').split('\n')) == obj.body_max_line_count
assert title[0:obj.title_maxlen] == chunks[0].get('title')
#
# Next Test: Split body
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# No counter is displayed because our title is so enormous
# We switch to a display title on first message only
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The length of the body prevails our title due to it being
# so much smaller then our title length
assert len(chunk.get('title')) == obj.body_maxlen
assert title[:obj.body_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
# Another edge case where the title just isn't that long leaving
# a lot of space for the [xx/xx] entries (no truncation needed)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title[:20])
offset = 0
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# No counter is displayed because our title is so enormous
# We switch to a display title on first message only
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The length of the body prevails our title due to it being
# so much smaller then our title length
assert len(chunk.get('title')) == 20
assert title[:20] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
#
# Test forcing overflow_display_title_once
#
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = title_len
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The body length prevails due to our amalgamation flag
assert len(chunk.get('title')) == obj.body_maxlen
assert title[:obj.body_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
assert body[offset: len(_body) + offset]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += len(_body)
# Test larger messages
# and that the body remains untouched
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 150
# Enforce a body length. Make sure it's an int.
body_maxlen = 400
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
new_body = (body * 500)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 1281
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# No counter is displayed because our title is so enormous
# We switch to a display title on first message only
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The length of the body prevails our title due to it being
# so much smaller then our title length
assert len(chunk.get('title')) == obj.title_maxlen
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
# Body chunk is beyond 4 digits, so [XXXX/XXXX] is turned off
new_body = (body * 2500)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 6401
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# No counter is displayed because our title is so enormous
# We switch to a display title on first message only
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The length of the body prevails our title due to it being
# so much smaller then our title length
assert len(chunk.get('title')) == obj.title_maxlen
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
# Test larger messages
# and that the body remains untouched
class TestNotification(NotifyBase):
# Test title max length
title_maxlen = 150
# Enforce a body length. Make sure it's an int.
body_maxlen = 150
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
new_body = (body * 5)
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=new_body, title="")
chunks = obj._apply_overflow(body=new_body, title=title)
offset = 0
assert len(chunks) == 36
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# Because 150 is what we set the title limit to
assert len(chunk.get('title')) == obj.title_maxlen
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(new_body[offset: len(_body) + offset]) - \
len(new_body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert new_body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Next Test: Append title to body + split body
#
class TestNotification(NotifyBase):
# Enforce no title
title_maxlen = 0
# Enforce a body length based on the title. Make sure it's an int.
body_maxlen = int(title_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
# Our final product is that our title has been appended to our body to
# create one great big body. As a result we'll get quite a few lines back
# now.
offset = 0
# Our body will look like this in small chunks at the end of the day
bulk = title + '\r\n' + body
# Due to the new line added to the end
assert len(chunks) == (
# wrap division in int() so Python 3 doesn't convert it to a float on
# us
int(len(bulk) / obj.body_maxlen) +
(1 if len(bulk) % obj.body_maxlen else 0))
for chunk in chunks:
# Verification
assert len(chunk.get('title')) == 0
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title is empty every time
assert chunk.get('title') == ''
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(bulk[offset: len(_body) + offset]) - \
len(bulk[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert bulk[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Test case where our title_len is shorter then the value
# that would otherwise trigger the [XX/XX] elements
#
class TestNotification(NotifyBase):
# Set a small title length
title_maxlen = 100
# Enforce a body length. Make sure it's an int.
body_maxlen = int(body_len / 4)
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 5
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# No counter is displayed because our title is so enormous
# We switch to a display title on first message only
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# The length of the body prevails our title due to it being
# so much smaller then our title length
assert len(chunk.get('title')) == obj.title_maxlen
assert title[:obj.title_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(body[offset: len(_body) + offset]) - \
len(body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
#
# Scenario where the title length is larger than the body
#
class TestNotification(NotifyBase):
# Set a small title length
title_maxlen = 100
# Enforce a body length. Make sure it's an int.
body_maxlen = 50
# With amalgamation
overflow_amalgamate_title = True
# Force title displayed once
overflow_display_title_once = True
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestNotification(overflow=OverflowMode.SPLIT)
assert obj is not None
# Verify that we break the title to a max length of our title_max
# and that the body remains untouched
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
offset = 0
assert len(chunks) == 22
for idx, chunk in enumerate(chunks, start=1):
# Verification
assert len(chunk.get('title')) <= obj.title_maxlen
assert len(chunk.get('body')) <= obj.body_maxlen
# Our title has no counter added to it due to it's length
if idx > 1:
# Empty (no title displayed on following entries
assert chunk.get('title') == ""
else:
# Because 150 is what we set the title limit to 50 due
# to amalamation. The lowest value always prevails
assert len(chunk.get('title')) == obj.body_maxlen
assert title[:obj.body_maxlen] == chunk.get('title')
# Our body is only broken up; not lost
_body = chunk.get('body')
# Un-used whitespace is always cleaned up; make sure we account for
# this in our new calculation
ws_diff = len(body[offset: len(_body) + offset]) - \
len(body[offset: len(_body) + offset]
.lstrip('\r\n\x0b\x0c').rstrip())
assert body[offset: len(_body) + offset + ws_diff]\
.lstrip('\r\n\x0b\x0c').rstrip() == _body
offset += (len(_body) + ws_diff)
2023-12-16 03:59:58 +01:00
def test_notify_markdown_general():
"""
2023-12-16 03:59:58 +01:00
API: Markdown General Testing
"""
#
# A little preparation
#
#
# First Test: Truncated Title
#
class TestMarkdownNotification(NotifyBase):
# Force our title to wrap
title_maxlen = 0
# Default Notify Format
notify_format = NotifyFormat.MARKDOWN
def __init__(self, *args, **kwargs):
super().__init__(**kwargs)
def notify(self, *args, **kwargs):
# Pretend everything is okay
return True
# Load our object
obj = TestMarkdownNotification()
assert obj is not None
# A bad header
title = " # "
body = "**Test Body**"
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(body=body, title=title)
assert len(chunks) == 1
# whitspace is trimmed
assert '#\r\n**Test Body**' == chunks[0].get('body')
assert chunks[0].get('title') == ""
# If we know our input is text however, we perform manipulation
chunks = obj._apply_overflow(body="", title=title)
chunks = obj._apply_overflow(body="", title="")
chunks = obj._apply_overflow(body=body, title="")
chunks = obj._apply_overflow(
body=body, title=title, body_format=NotifyFormat.TEXT)
assert len(chunks) == 1
# Our title get's stripped off since it's not of valid markdown
assert body.lstrip('\r\n\x0b\x0c').rstrip() == chunks[0].get('body')
assert chunks[0].get('title') == ""
2023-12-16 03:59:58 +01:00
@mock.patch('requests.post')
def test_notify_emoji_general(mock_post):
"""
API: Emoji General Testing
"""
# Prepare our response
response = requests.Request()
response.status_code = requests.codes.ok
# Prepare Mock
mock_post.return_value = response
# Set up our emojis
title = ":smile:"
body = ":grin:"
# general reference used below (using default values)
asset = AppriseAsset()
#
# Test default emoji asset value
#
# Load our object
ap_obj = Apprise(asset=asset)
ap_obj.add('json://localhost')
assert len(ap_obj) == 1
assert ap_obj.notify(title=title, body=body) is True
assert mock_post.call_count == 1
details = mock_post.call_args_list[0]
dataset = json.loads(details[1]['data'])
# No changes
assert dataset['title'] == title
assert dataset['message'] == body
mock_post.reset_mock()
#
# Test URL over-ride while default value set in asset
#
# Load our object
ap_obj = Apprise(asset=asset)
ap_obj.add('json://localhost?emojis=no')
assert len(ap_obj) == 1
assert ap_obj.notify(title=title, body=body) is True
assert mock_post.call_count == 1
details = mock_post.call_args_list[0]
dataset = json.loads(details[1]['data'])
# No changes
assert dataset['title'] == title
assert dataset['message'] == body
mock_post.reset_mock()
#
# Test URL over-ride while default value set in asset pt 2
#
# Load our object
ap_obj = Apprise(asset=asset)
ap_obj.add('json://localhost?emojis=yes')
assert len(ap_obj) == 1
assert ap_obj.notify(title=title, body=body) is True
assert mock_post.call_count == 1
details = mock_post.call_args_list[0]
dataset = json.loads(details[1]['data'])
# Emoji's are displayed
assert dataset['title'] == '😄'
assert dataset['message'] == '😃'
mock_post.reset_mock()
#
# Test URL over-ride while default value set in asset pt 2
#
# Load our object
ap_obj = Apprise(asset=asset)
ap_obj.add('json://localhost?emojis=no')
assert len(ap_obj) == 1
assert ap_obj.notify(title=title, body=body) is True
assert mock_post.call_count == 1
details = mock_post.call_args_list[0]
dataset = json.loads(details[1]['data'])
# No changes
assert dataset['title'] == title
assert dataset['message'] == body
mock_post.reset_mock()
#
# Test Default Emoji settings
#
# Set our interpret emoji's flag
asset = AppriseAsset(interpret_emojis=True)
# Re-create our object
ap_obj = Apprise(asset=asset)
# Load our object
ap_obj.add('json://localhost')
assert len(ap_obj) == 1
assert ap_obj.notify(title=title, body=body) is True
assert mock_post.call_count == 1
details = mock_post.call_args_list[0]
dataset = json.loads(details[1]['data'])
# emoji's are displayed
assert dataset['title'] == '😄'
assert dataset['message'] == '😃'
mock_post.reset_mock()
#
# With Emoji's turned on by default, the user can optionally turn them
# off.
#
# Re-create our object
ap_obj = Apprise(asset=asset)
ap_obj.add('json://localhost?emojis=no')
assert len(ap_obj) == 1
assert ap_obj.notify(title=title, body=body) is True
assert mock_post.call_count == 1
details = mock_post.call_args_list[0]
dataset = json.loads(details[1]['data'])
# No changes
assert dataset['title'] == title
assert dataset['message'] == body
mock_post.reset_mock()
#
# With Emoji's turned on by default, there is no change when emojis
# flag is set to on
#
# Re-create our object
ap_obj = Apprise(asset=asset)
ap_obj.add('json://localhost?emojis=yes')
assert len(ap_obj) == 1
assert ap_obj.notify(title=title, body=body) is True
assert mock_post.call_count == 1
details = mock_post.call_args_list[0]
dataset = json.loads(details[1]['data'])
# emoji's are displayed
assert dataset['title'] == '😄'
assert dataset['message'] == '😃'
mock_post.reset_mock()
#
# Enforce the disabling of emojis
#
# Set our interpret emoji's flag
asset = AppriseAsset(interpret_emojis=False)
# Re-create our object
ap_obj = Apprise(asset=asset)
# Load our object
ap_obj.add('json://localhost')
assert len(ap_obj) == 1
assert ap_obj.notify(title=title, body=body) is True
assert mock_post.call_count == 1
details = mock_post.call_args_list[0]
dataset = json.loads(details[1]['data'])
# Disabled - no emojis
assert dataset['title'] == title
assert dataset['message'] == body
mock_post.reset_mock()
#
# Enforce the disabling of emojis
#
# Set our interpret emoji's flag
asset = AppriseAsset(interpret_emojis=False)
# Re-create our object
ap_obj = Apprise(asset=asset)
# Load our object
ap_obj.add('json://localhost?emojis=yes')
assert len(ap_obj) == 1
assert ap_obj.notify(title=title, body=body) is True
assert mock_post.call_count == 1
details = mock_post.call_args_list[0]
dataset = json.loads(details[1]['data'])
# Disabled - no emojis
assert dataset['title'] == title
assert dataset['message'] == body
mock_post.reset_mock()