Merge pull request #20 from depaul-dice/feature-checkout

Checkout Feature for Sciunit Executions

sciunit2/cli.py

@@ -13,6 +13,7 @@
 from sciunit2.command.sort import SortCommand
 from sciunit2.command.push import PushCommand
 from sciunit2.command.copy import CopyCommand
+from sciunit2.command.checkout import CheckoutCommand
 from sciunit2.command.post_install import PostInstallCommand
 from sciunit2.command.diff import DiffCommand
 from sciunit2.command.export import ExportCommand
@@ -29,7 +30,7 @@
 __cmds__ = [CreateCommand, OpenCommand, ExecCommand, RepeatCommand,
             ListCommand, ShowCommand, GivenCommand, CommitCommand, RmCommand,
             SortCommand, PushCommand, CopyCommand, PostInstallCommand,
-            DiffCommand, RemoveCommand, ExportCommand]
+            DiffCommand, RemoveCommand, ExportCommand, CheckoutCommand]
 
 
 def short_usage(out):

sciunit2/command/checkout.py

@@ -0,0 +1,29 @@
+from __future__ import absolute_import
+
+from sciunit2.command import AbstractCommand
+from sciunit2.command.context import CheckoutContext
+from sciunit2.exceptions import CommandLineError
+from sciunit2.util import quoted_format
+import sciunit2.core
+
+from getopt import getopt
+import sys
+
+
+class CheckoutCommand(AbstractCommand):
+    name = 'checkout'
+
+    @property
+    def usage(self):
+        return [('checkout <execution id>',
+                 "Checkout the execution of <execution id>")]
+
+    def run(self, args):
+        optlist, args = getopt(args, '')
+        if len(args) != 1:
+            raise CommandLineError
+        with CheckoutContext(args[0]) as (pkgdir, _):
+            return pkgdir
+
+    def note(self, project_dir):
+        return quoted_format('Checked out at: {0}\n', project_dir)

tests/test_checkout.py

@@ -0,0 +1,36 @@
+from __future__ import absolute_import
+
+from nose.tools import *
+import os
+import testpath
+
+from tests import testit
+
+
+class TestCheckout(testit.LocalCase):
+    def test_all(self):
+        with assert_raises(SystemExit) as r:
+            testit.sciunit('checkout')
+        assert_equal(r.exception.code, 2)
+
+        with assert_raises(SystemExit) as r:
+            testit.sciunit('checkout', '-i')
+        assert_equal(r.exception.code, 2)
+
+        with assert_raises(SystemExit) as r:
+            testit.sciunit('checkout', 'e1')
+        assert_equal(r.exception.code, 1)
+
+        testit.sciunit('create', 'ok')
+
+        with assert_raises(SystemExit) as r:
+            testit.sciunit('checkout', 'asdf')
+        assert_equal(r.exception.code, 1)
+
+        with assert_raises(SystemExit) as r:
+            testit.sciunit('checkout', 'e1')
+        assert_equal(r.exception.code, 1)
+
+        testit.sciunit('exec', 'pwd')
+        testit.sciunit('checkout', 'e1')
+        assert_true(os.path.isdir('tmp/ok/cde-package'))