Ticket #37: integrateConfigTester.diff

buildbot/scripts/runner.py

@@ -3 +3 @@
-# N.B.: don't import anything that might pull in a reactor yet. Some of our
-# subcommands want to load modules that need the gtk reactor.
-import os, sys, stat, re, time
+from shutil import copy, rmtree
+from tempfile import mkdtemp
+from os.path import isfile
+import traceback
-from twisted.python import usage, util, runtime
-
-# this is mostly just a front-end for mktap, twistd, and kill(1), but in the
@@ -809 +813 @@
-    os.rename(tmpfile, newfile)
-
-
+from buildbot import master
+
+class ConfigLoader(master.BuildMaster):
+    def __init__(self, configFileName="master.cfg"):
+        master.BuildMaster.__init__(self, ".", configFileName)
+        dir = os.getcwd()
+        # Use a temporary directory since loadConfig() creates a bunch of
+        # directories and compiles .py files
+        tempdir = mkdtemp()
+        file = configFileName
+        try:
+            copy(configFileName, tempdir)
+            for entry in os.listdir("."):
+                # Any code in a subdirectory will _not_ be copied! This is a bug
+                if isfile(entry):
+                    copy(entry, tempdir)
+    except:
+        raise Exception("Error copying file %s" % file)
+
+    try:
+        os.chdir(tempdir)
+        # Add the temp directory to the library path so local modules will work
+        sys.path.append(tempdir)
+        configFile = open(configFileName, "r")
+        self.loadConfig(configFile)
+    except:
+        raise
+    finally:
+        os.chdir(dir)
+        rmtree(tempdir)
+
+class CheckConfigOptions(usage.Options):
+    optFlags = [
+        ['quiet', 'q', "Don't display error messages or tracebacks"],
+    ]
+
+    def getSynopsis(self):
+        return "Usage           :buildbot checkconfig [configFile]\n" + \
+         "              If not specified, 'master.cfg' will be used as 'configFile'"
+
+    def parseArgs(self, *args):
+        if len(args) >= 1:
+            self['configFile'] = args[0]
+        else:
+            self['configFile'] = 'master.cfg'
+
+
+def doCheckConfig(config):
+    quiet = config.get('quiet')
+    configFile = config.get('configFile')
+    try:
+        ConfigLoader(configFile)
+    except:
+        if not quiet:
+            # Print out the traceback in a nice format
+            t, v, tb = sys.exc_info()
+            traceback.print_exception(t, v, tb)
+        sys.exit(1)
+
+    if not quiet:
+        print "Config file is good!"
+
+
-class Options(usage.Options):
-    synopsis = "Usage:    buildbot <command> [command options]"
-
@@ -847 +914 @@
-        ['tryserver', None, TryServerOptions,
-         "buildmaster-side 'try' support function, not for users"],
-
+        ['checkconfig', None, CheckConfigOptions,
+         "test the validity of a master.cfg config file"],
+
-        # TODO: 'watch'
-        ]
-
@@ -906 +976 @@
-        doTry(so)
-    elif command == "tryserver":
-        doTryServer(so)
+    elif command == "checkconfig":
+        doCheckConfig(so)
-
-