Issue 69: Tests for Headers support

apprise/plugins/NotifyBase.py

@@ -425,5 +425,6 @@ class NotifyBase(object):
         if 'user' in results['qsd']:
             results['user'] = results['qsd']['user']
 
-        results['headers'] = {k[1:]: v for k, v in results['qsd'].items() if re.match(r'^-.', k)}
+        results['headers'] = {k[1:]: v for k, v in results['qsd'].items()
+                              if re.match(r'^-.', k)}
         return results

apprise/plugins/NotifyDiscord.py

@@ -181,7 +181,7 @@ class NotifyDiscord(NotifyBase):
         else:
             # not markdown
             payload['content'] = body if not title \
-                    else "{}\r\n{}".format(title, body)
+                else "{}\r\n{}".format(title, body)
 
         if self.avatar and image_url:
             payload['avatar_url'] = image_url
@@ -298,7 +298,7 @@ class NotifyDiscord(NotifyBase):
         """
         regex = re.compile(
             r'^\s*#+\s*(?P<name>[^#\n]+)([ \r\t\v#])?'
-            r'(?P<value>([^ \r\t\v#].+?)(\n(?!\s#))|\s*$)', flags=re.S|re.M)
+            r'(?P<value>([^ \r\t\v#].+?)(\n(?!\s#))|\s*$)', flags=re.S | re.M)
 
         common = regex.finditer(markdown)
         fields = list()

apprise/plugins/NotifySNS.py

@@ -58,8 +58,8 @@ LIST_DELIM = re.compile(r'[ \t\r\n,\\/]+')
 # region as a delimiter. This is a bit hacky; but it's much easier than having
 # users of this product search though this Access Key Secret and escape all
 # of the forward slashes!
-IS_REGION = re.compile(
+IS_REGION = re.compile(r'^\s*(?P<country>[a-z]{2})-'
-        r'^\s*(?P<country>[a-z]{2})-(?P<area>[a-z]+)-(?P<no>[0-9]+)\s*$', re.I)
+                       r'(?P<area>[a-z]+)-(?P<no>[0-9]+)\s*$', re.I)
 
 # Extend HTTP Error Messages
 AWS_HTTP_ERROR_MAP = HTTP_ERROR_MAP.copy()

test/test_notify_base.py

@@ -166,6 +166,12 @@ def test_notify_base_urls():
     assert 'password' in results
     assert results['password'] == "newpassword"
 
+    # pass headers
+    results = NotifyBase.parse_url(
+        'https://localhost:8080?-HeaderKey=HeaderValue')
+    assert 'headerkey' in results['headers']
+    assert results['headers']['headerkey'] == 'HeaderValue'
+
     # User Handling
 
     # user keyword over-rides default password

test/test_rest_plugins.py

@@ -431,6 +431,10 @@ TEST_URLS = (
         # is set and tests that we gracfully handle them
         'test_requests_exceptions': True,
     }),
+    ('json://localhost:8080/path?-HeaderKey=HeaderValue', {
+        'instance': plugins.NotifyJSON,
+    }),
+
 
     ##################################
     # NotifyKODI
@@ -1470,6 +1474,9 @@ TEST_URLS = (
         # is set and tests that we gracfully handle them
         'test_requests_exceptions': True,
     }),
+    ('xml://localhost:8080/path?-HeaderKey=HeaderValue', {
+        'instance': plugins.NotifyXML,
+    }),
 )