add default status code

shanbay · Oct 31, 2017 · f24d25a · f24d25a
1 parent 8134fae
commit f24d25a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/sea/test/stub.py b/sea/test/stub.py
@@ -1,6 +1,9 @@
+import grpc
+
+
 class Context():
     def __init__(self):
-        self.code = None
+        self.code = grpc.StatusCode.OK
         self.details = None
         self.metadata = None
 

diff --git a/tests/test_test.py b/tests/test_test.py
@@ -1,3 +1,5 @@
+import grpc
+
 from sea.exceptions import BadRequestException
 from sea.test.stub import Context, Stub
 from sea.servicer import ServicerMeta
@@ -19,6 +21,7 @@ def return_normal(self, request, context):
     assert stub.ctx.invocation_metadata() == data
     res = stub.return_normal(None)
     assert res is None
+    assert stub.ctx.code == grpc.StatusCode.OK
     assert stub.ctx.invocation_metadata() is None
 
     res = stub.return_error(None)