From 596492baa23623411d143186d9cad33a1305cd24 Mon Sep 17 00:00:00 2001
From: Waylan Limberg <waylan@gmail.com>
Date: Thu, 4 Aug 2011 12:57:25 -0400
Subject: Added some fadfile commands to (re)generate tests when they are added
 or updated.

---
 fabfile.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'fabfile.py')

diff --git a/fabfile.py b/fabfile.py
index 9680bd1..007fd7f 100644
--- a/fabfile.py
+++ b/fabfile.py
@@ -61,6 +61,22 @@ def build_tests(version=_pyversion[:3]):
         local('2to3-%s -w build/test.%s/tests' % (version, version))
         local('2to3-%s -w build/test.%s/run-tests.py' % (version, version))
 
+def generate_test(file):
+    """ Generate a given test. """
+    import tests
+    config = tests.get_config(os.path.dirname(file))
+    root, ext = os.path.splitext(file)
+    if ext == config.get(get_section(os.path.basename(root), config), 
+                         'input_ext'):
+        tests.generate(root, config)
+    else:
+        print test, 'does not have a valid file extension. Check config.'
+
+def generate_tests():
+    """ Generate all outdated tests. """
+    from tests import generate_all
+    generate_all()
+
 def build_env(version=_pyversion[:3]):
     """ Build testing environment for given Python version. """
     if version in confirmed_versions:
-- 
cgit v1.2.3