diff options
Diffstat (limited to 'ca')
| -rw-r--r-- | ca/tests/yamltest.py | 43 |
1 files changed, 27 insertions, 16 deletions
diff --git a/ca/tests/yamltest.py b/ca/tests/yamltest.py index f49968b3..308884a9 100644 --- a/ca/tests/yamltest.py +++ b/ca/tests/yamltest.py @@ -576,24 +576,35 @@ class allocation(object): database this week. """ - verbosity = 1 + # Fork a sub-process for each syncdb/migrate run, because it's + # easier than figuring out how to change Django settings after + # initialization. - if verbosity > 0: - print "Running Django setup for", self.name + def sync_app(app, verbosity = 1): - if not os.fork(): - os.environ.update(RPKI_CONF = self.path("rpki.conf"), - RPKI_GUI_ENABLE = "yes") - logging.getLogger().setLevel(logging.WARNING) - import django.core.management - django.core.management.call_command("syncdb", migrate = True, verbosity = verbosity, - load_initial_data = False, interactive = False) - from django.contrib.auth.models import User - User.objects.create_superuser("root", "root@example.org", "fnord") - sys.exit(0) + if verbosity > 0: + print "Running Django setup for", self.name - if os.wait()[1]: - raise RuntimeError("Django setup failed for %s" % self.name) + pid = os.fork() + + if pid == 0: + os.environ.update(RPKI_CONF = self.path("rpki.conf"), + DJANGO_SETTINGS_MODULE = "rpki.django_settings." + app) + logging.getLogger().setLevel(logging.WARNING) + import django.core.management + django.core.management.call_command("syncdb", migrate = True, verbosity = verbosity, + load_initial_data = False, interactive = False) + if app in ("gui", "irdb"): + from django.contrib.auth.models import User + User.objects.create_superuser("root", "root@example.org", "fnord") + sys.exit(0) + + else: + if os.waitpid(pid, 0)[1]: + raise RuntimeError("Django setup failed for %s %s" % (self.name, app)) + + for app in ("rpkid", "pubd", "gui"): + sync_app(app) def run_python_daemon(self, prog): """ @@ -705,7 +716,7 @@ def create_root_certificate(db_root): logger = logging.getLogger(__name__) -os.environ.update(DJANGO_SETTINGS_MODULE = "rpki.django_settings", +os.environ.update(DJANGO_SETTINGS_MODULE = "rpki.django_settings.irdb", TZ = "UTC") time.tzset() |