fix(APIClient): return APIClient instance in setUserAgent method

use this to improve code by method chaining
centralnicgroup-opensource · Apr 4, 2019 · 2169891 · 2169891
1 parent 1823117
commit 2169891
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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()