From 2169891df8341a449ad5e4a7ab295a03f2eb93cc Mon Sep 17 00:00:00 2001 From: Kai Schwarz Date: Thu, 4 Apr 2019 12:05:08 +0200 Subject: [PATCH] fix(APIClient): return APIClient instance in setUserAgent method use this to improve code by method chaining --- hexonet/apiconnector/apiclient.py | 1 + tests/test_apiclient.py | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/hexonet/apiconnector/apiclient.py b/hexonet/apiconnector/apiclient.py index e3a4038..446f99a 100644 --- a/hexonet/apiconnector/apiclient.py +++ b/hexonet/apiconnector/apiclient.py @@ -93,6 +93,7 @@ def setUserAgent(self, pid, rv): pf = platform.system() arch = platform.architecture()[0] self.__ua = "%s (%s; %s; rv:%s) python-sdk/%s python/%s" % (pid, pf, arch, rv, self.getVersion(), pyv) + return self def getVersion(self): """ diff --git a/tests/test_apiclient.py b/tests/test_apiclient.py index 7e6d8c9..d82776b 100644 --- a/tests/test_apiclient.py +++ b/tests/test_apiclient.py @@ -122,7 +122,8 @@ def test_apiclientmethods(): pf = platform.system() arch = platform.architecture()[0] ua = "%s (%s; %s; rv:%s) %s/%s python/%s" % (pid, pf, arch, rv, pid2, cl.getVersion(), pyv) - cl.setUserAgent(pid, rv) + cl2 = cl.setUserAgent(pid, rv) + assert isinstance(cl2, AC) is True assert cl.getUserAgent() == ua # #.setURL()