fix(composer): fix unit test and add tear down

gnes-ai · Jul 16, 2019 · 743ec3b · 743ec3b
1 parent 6ec4233
commit 743ec3b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/gnes/cli/parser.py b/gnes/cli/parser.py
@@ -47,7 +47,7 @@ def set_composer_parser(parser=None):
     parser.add_argument('--yaml_path', type=argparse.FileType('r'),
                         help='yaml config of the service')
     parser.add_argument('--html_path', type=argparse.FileType('w', encoding='utf8'),
-                        help='render the network graph in HTML with mermaid.js')
+                        help='output path of the HTML file, will contain all possible generations')
     parser.add_argument('--shell_path', type=argparse.FileType('w', encoding='utf8'),
                         help='output path of the shell-based starting script')
     parser.add_argument('--swarm_path', type=argparse.FileType('w', encoding='utf8'),

diff --git a/tests/test_compose.py b/tests/test_compose.py
@@ -32,6 +32,6 @@ def _test_topology(self, yaml_path: str, num_layer_before: int, num_layer_after:
         os.path.exists(self.html_path)
         print(a.build_dockerswarm(r))
 
-    # def tearDown(self):
-    #     if os.path.exists(self.html_path):
-    #         os.remove(self.html_path)
+    def tearDown(self):
+        if os.path.exists(self.html_path):
+            os.remove(self.html_path)