Create separate _get_batch_script method for testing batch_script subst

Author: rkdarst

batchspawner/batchspawner.py

@@ -206,6 +206,12 @@ def run_command(self, cmd, input=None, env=None):
             out = out.decode().strip()
             return out
 
+    @gen.coroutine
+    def _get_batch_script(self, **subvars):
+        """Format batch script from vars"""
+        # Colud be overridden by subclasses, but mainly useful for testing
+        return format_template(self.batch_script, **subvars)
+
     @gen.coroutine
     def submit_batch_script(self):
         subvars = self.get_req_subvars()
@@ -214,8 +220,7 @@ def submit_batch_script(self):
         subvars['cmd'] = self.cmd_formatted_for_batch()
         if hasattr(self, 'user_options'):
             subvars.update(self.user_options)
-        script = format_template(self.batch_script, **subvars)
-        self._last_batch_script = script   # used for testing only
+        script = yield self._get_batch_script(**subvars)
         self.log.info('Spawner submitting job using ' + cmd)
         self.log.info('Spawner submitted script:\n' + script)
         out = yield self.run_command(cmd, input=script, env=self.get_env())