Replace test_connection to class method

This commit is contained in:
Raimon Esteve 2016-11-30 23:25:29 +01:00
parent deccd3aa45
commit 90bf338e41

18
api.py
View file

@ -9,7 +9,7 @@ try:
from seurvalencia.picking import *
except ImportError:
logger = logging.getLogger(__name__)
message = 'Install Seur from Pypi: pip install seurvalencia'
message = 'Install Seur: pip install seurvalencia'
logger.error(message)
raise Exception(message)
@ -22,9 +22,7 @@ class CarrierApi:
@classmethod
def get_carrier_app(cls):
'''
Add Carrier Seur APP
'''
'Add Carrier Seur APP'
res = super(CarrierApi, cls).get_carrier_app()
res.append(('seurvalencia', 'Seur Valencia'))
return res
@ -36,13 +34,11 @@ class CarrierApi:
'invisible': Not(Equal(Eval('method'), 'seurvalencia')),
})]
def test_seurvalencia(self, api):
'''
Test Seur Valencia connection
:param api: obj
'''
@classmethod
def test_seurvalencia(cls, api):
'Test Seur Valencia connection'
message = 'Connection unknown result'
with API(api.username, api.password, api.debug) as seurvalencia_api:
message = seurvalencia_api.test_connection()
self.raise_user_error(message)
cls.raise_user_error(message)