From 823bdeda9ead30ae989a8c46cb32a51581bc4753 Mon Sep 17 00:00:00 2001
From: hanhxiao <hanhxiao@tencent.com>
Date: Fri, 2 Aug 2019 13:10:56 +0800
Subject: [PATCH] fix(test): fix grpc gentle shutdown

---
 gnes/composer/base.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gnes/composer/base.py b/gnes/composer/base.py
index 76e524a7..e6fe7137 100644
--- 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