Added reauth test

Author: fronzbot

tests/mock_responses.py

@@ -214,10 +214,16 @@ def json(self):
         def raw(self):
             """Return raw data from get request."""
             return self.raw_data
-
+    rx_header = kwargs.pop('headers')
+    expected_token = LOGIN_RESPONSE['authtoken']['authtoken']
     # pylint: disable=unused-variable
+    if ('Content-Type' not in rx_header
+            and rx_header['TOKEN_AUTH'] != expected_token):
+        return MockGetResponse({'message': 'Not Authorized', 'code': 400}, 400)
+
     (region_id, region), = LOGIN_RESPONSE['region'].items()
-    if args[0] != 'use_bad_response' and args[0] != 'reauth' and args[0] is not None:
+    if (args[0] != 'use_bad_response'
+            and args[0] != 'reauth' and args[0] is not None):
         set_region_id = args[0].split('/')[2].split('.')[0]
     else:
         set_region_id = 'ciri'
@@ -297,4 +303,4 @@ def __init__(self, region_id):
         self.home_url = 'reauth'
         self.event_url = self.base_url + '/events/network/'
         self.network_url = self.base_url + '/network/'
-        self.networks_url = self.base_url + '/networks'
+        self.networks_url = self.base_url + '/networks'

tests/test_blink_system.py

@@ -181,22 +181,21 @@ def test_setup_backup_subdomain(self, mock_get, mock_post):
     def test_continuous_bad_auth(self, mock_url, mock_get, mock_post):
         """Check that we don't get stuck if we can't reauthorize."""
         with self.assertRaises(blinkpy.BlinkAuthenticationException):
-            self.blink.setup_system()      
+            self.blink.setup_system()
 
-'''
     @mock.patch('blinkpy.blinkpy.requests.post',
                 side_effect=mresp.mocked_requests_post)
     @mock.patch('blinkpy.blinkpy.requests.get',
                 side_effect=mresp.mocked_requests_get)
     def test_reauthorization_attempt(self, mock_get, mock_post):
         """Check that we can reauthorize after first unsuccessful attempt."""
         self.blink.setup_system()
-        old_token = self.blink._token
-        self.blink._token = 'BADTOKEN'
-        (region_id, region), = mresp.LOGIN_RESPONSE['region'].items()
-        test_urls = blinkpy.BlinkURLHandler(region_id)
-        self.blink.urls.home_url = 'reauth'
-        self.assertEqual(self.blink.refresh(), None)
-        self.assertEqual(self.blink.urls.home_url, test_urls.home_url)
-
-'''
+        # pylint: disable=protected-access
+        original_header = self.blink._auth_header
+        bad_header = {'Host': self.blink._host, 'TOKEN_AUTH': 'BADTOKEN'}
+        self.blink._auth_header = bad_header
+        # pylint: disable=protected-access
+        self.assertEqual(self.blink._auth_header, bad_header)
+        self.blink.get_summary()
+        # pylint: disable=protected-access
+        self.assertEqual(self.blink._auth_header, original_header)