fix(test): fix grpc gentle shutdown

gnes-ai · Aug 2, 2019 · 823bded · 823bded
1 parent c5c811d
commit 823bded
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gnes/composer/base.py b/gnes/composer/base.py
@@ -160,7 +160,7 @@ def build_layers(self) -> List['YamlComposer.Layer']:
 
         # gRPCfrontend should always on the bind role
         assert all_layers[0].is_single_component
-        assert all_layers[0].components[0].name == 'gRPCFrontend'
+        assert all_layers[0].components[0]['name'] == 'gRPCFrontend'
 
         if all_layers[0].components[0]['socket_in'] == str(SocketType.SUB_CONNECT):
             # change to sub bind
@@ -173,7 +173,7 @@ def build_layers(self) -> List['YamlComposer.Layer']:
             all_layers[0].components[0]['socket_in'] = str(SocketType.PULL_BIND)
             for c in all_layers[-1].components:
                 c['socket_out'] = str(SocketType.PUSH_CONNECT)
-         
+
         return all_layers
 
     @staticmethod