#167: Add Client class under Service class

nirum-lang · Feb 9, 2018 · 9f2c36d · 9f2c36d
1 parent f449697
commit 9f2c36d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/Nirum/Targets/Python.hs b/src/Nirum/Targets/Python.hs
@@ -1123,6 +1123,12 @@ class {className}_Client($className):
         self.__nirum_transport__ = transport  # type: transport_type
 
     {clientMethods'}
+
+{className}.Client = {className}_Client
+if hasattr({className}.Client, '__qualname__'):
+    {className}.Client.__qualname__ = '{className}.Client'
+else:
+    {className}.Client.__name__ = '{className}.Client'
 |]
   where
     nirumMapName :: T.Text

diff --git a/test/python/service_test.py b/test/python/service_test.py
@@ -3,7 +3,7 @@
 from nirum.transport import Transport
 
 from fixture.foo import (Dog, Gender, PingService, Product, RpcError,
-                         SampleService_Client, Way)
+                         SampleService, SampleService_Client, Way)
 
 
 def test_throws_error():
@@ -48,7 +48,8 @@ def latest_call(self):
 
 def test_service_client_payload_serialization():
     t = DumbTransport()
-    c = SampleService_Client(t)
+    c = SampleService.Client(t)
+    assert SampleService_Client is SampleService.Client
     c.sample_method(
         a=Dog(name=u'Dog.name', age=3),
         b=Product(name=u'Product.name', sale=False),
@@ -83,3 +84,8 @@ def test_service_client_payload_serialization():
         'g': 1234,
         'hh': 'text data',
     }
+
+
+def test_service_client_representation():
+    assert repr(SampleService.Client) == \
+        "<class 'fixture.foo.SampleService.Client'>"