Added unittests for settings, render, and build. Fixed bug in render cmd

tgen · Feb 6, 2019 · ba4fc10 · ba4fc10
1 parent dab7bf2
commit ba4fc10
Show file tree

Hide file tree

Showing 3 changed files with 80 additions and 3 deletions.
diff --git a/jetstream/cli/subcommands/render.py b/jetstream/cli/subcommands/render.py
@@ -1,7 +1,5 @@
 """Render a template and print the results"""
 import logging
-import jetstream
-from jetstream import settings
 from jetstream.templates import context, render_template
 
 log = logging.getLogger(__name__)
@@ -19,7 +17,6 @@ def main(args):
     log.debug(f'{__name__} {args}')
 
     c = context(
-        constants=settings['constants'].get(dict),
         project=args.project,
         command_args=args.kvargs
     )

diff --git a/test/templates/should_pass/retry.jst b/test/templates/should_pass/retry.jst
@@ -0,0 +1,12 @@
+- name: fails_once
+  retry: 1
+  stdout: /dev/null
+  cmd: |
+    if [ ! -f foo.txt ]; then
+      echo "File not found!"
+      touch foo.txt
+      exit 1
+    else
+      echo "File was found!"
+    fi
+
diff --git a/test/test_cli.py b/test/test_cli.py
@@ -115,3 +115,71 @@ def test_tasks(self):
 
         with redirect_stdout(StringIO()):
             cli_main(args)
+
+    def test_render(self):
+        render_test = """
+                                                                        
+                              ##//#/##/##                            
+                            ###//##/###//##                          
+                           ###//###//##//###                         
+                           ##///##///##///##                         
+                           ##///##///###//##                         
+                           ##///##///###//##                         
+            #/##/#         ##///##///###//##                         
+           #/#/##/#        ##///##///###//##                         
+           #/#/##/#        ##///##///###//##                         
+           #/#/##/#        ##///##///###//##                         
+           #/#/##/#        ##///##///###//##                         
+           #/#/##/#        ##///##///###//##                         
+           #/#/##/#        ##///##///###//##        *#/#/#(/#        
+           #/#//##/////////##///##///###//##       /#//#//#/(#       
+           #/##////////////##///##///###//##       ##/##//#//#       
+           #///##############///##///###//##       ##/##//#//#       
+            #################///##///###//##       ##/##//#//#       
+                           ##///##///###//##       ##/##//#//#       
+                           ##///##///###//##       ##/##//#//#       
+                           ##///##///###//##       ##/##//#//#       
+                           ##///##///###//##########//##//#//#       
+                           ##///##///###//##############//#//#       
+                           ##///##///###//##/////////////##//#       
+                           ##///##///###//################//##       
+                           ##///##///###//##//////////////###      
+                           ##///##///###//##                         
+                           ##///##///###//##                         
+                           ##///##///###//##                         
+                           ##///##///###//##                         
+                           ##///##///###//##                         
+                           ##///##///###//##  
+                           ##///##///###//##                         
+                           ##///##///###//## 
+                           ##///##///###//##  
+                           ##///##///###//##                         
+                           ##///##///###//##                    
+        """
+        with open('testwf.jst', 'w') as fp:
+            fp.write(render_test)
+
+        args = CMD_ARGS + [
+            'render',
+            'testwf.jst'
+        ]
+
+        cli_main(args)
+
+    def test_build(self):
+        with open('testwf.jst', 'w') as fp:
+            fp.write('- cmd: true\n  stdout: /dev/null\n')
+
+        args = CMD_ARGS + [
+            'build',
+            'testwf.jst'
+        ]
+
+        cli_main(args)
+
+    def test_settings(self):
+        args = CMD_ARGS + [
+            'settings',
+        ]
+
+        cli_main(args)