accept 403 (ForbiddenException) when trying to allocate (#23)

stf_appium_client/StfClient.py

@@ -4,6 +4,9 @@
 import json
 from pydash import filter_, map_, wrap, find, uniq
 import atexit
+
+from stf_client.exceptions import ForbiddenException
+
 from stf_appium_client.Logger import Logger
 from stf_appium_client.exceptions import DeviceNotFound, NotConnectedError
 from stf_client.api_client import ApiClient, Configuration
@@ -197,7 +200,7 @@ def allocate_first():
             def try_allocate(device):
                 try:
                     return self.allocate(device, timeout_seconds=timeout_seconds)
-                except AssertionError as error:
+                except (AssertionError, ForbiddenException) as error:
                     self.logger.warning(f"{device.get('serial')}Allocation fails: {error}")
                     return None
 

test/test_StfClient.py

@@ -2,6 +2,9 @@
 import logging
 import types
 from unittest.mock import patch, MagicMock
+
+from stf_client.exceptions import ForbiddenException
+
 from stf_appium_client.StfClient import StfClient
 from stf_appium_client.exceptions import *
 
@@ -134,12 +137,12 @@ def test_find_and_allocate_success(self):
     def test_find_and_allocate_second_success(self, mock_choise):
         invalid = {'serial': '12', 'present': False, 'ready': True, 'using': False, 'owner': None, 'status': 3}
         available = {'serial': '123', 'present': True, 'ready': True, 'using': False, 'owner': None, 'status': 3}
-        self.client.get_devices = MagicMock(return_value=[invalid, available])
+        self.client.get_devices = MagicMock(return_value=[invalid, available, available])
 
         def dummy_alloc(dev, timeout_seconds):
             return dev
 
-        self.client.allocate = MagicMock(side_effect=dummy_alloc)
+        self.client.allocate = MagicMock(side_effect=[ForbiddenException, dummy_alloc])
 
         device = self.client.find_and_allocate({})
         available['owner'] = 'me'