1 files changed, 106 insertions, 0 deletions
diff --git a/ca/tests/test-rrdp.py b/ca/tests/test-rrdp.py
new file mode 100755
index 00000000..cff17a87
--- /dev/null
+++ b/ca/tests/test-rrdp.py
@@ -0,0 +1,106 @@
+#!/usr/bin/env python
+# $Id$
+#
+# Copyright (C) 2013  Dragon Research Labs ("DRL")
+#
+# Permission to use, copy, modify, and distribute this software for any
+# purpose with or without fee is hereby granted, provided that the above
+# copyright notice and this permission notice appear in all copies.
+#
+# THE SOFTWARE IS PROVIDED "AS IS" AND DRL DISCLAIMS ALL WARRANTIES WITH
+# REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
+# AND FITNESS.  IN NO EVENT SHALL DRL BE LIABLE FOR ANY SPECIAL, DIRECT,
+# INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+# LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
+# OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+# PERFORMANCE OF THIS SOFTWARE.
+
+"""
+Proof-of-concept test driver for RRDP code.  Still fairly kludgy in places.
+"""
+
+import os
+import sys
+import glob
+import time
+import textwrap
+import argparse
+import subprocess 
+
+parser = argparse.ArgumentParser(description = __doc__)
+parser.add_argument("--use-smoketest", action = "store_true")
+parser.add_argument("--yaml-file", default = "smoketest.2.yaml")
+parser.add_argument("--delay", type = int, default = 300)
+parser.add_argument("--exhaustive", action = "store_true")
+args = parser.parse_args()
+
+def log(msg):
+  sys.stdout.write(msg + "\n")
+  sys.stdout.flush()
+
+def run(*argv):
+  log("Running: " + " ".join(argv))
+  subprocess.check_call(argv)
+
+def dataglob(pattern):
+  return glob.iglob(os.path.join(("smoketest.dir" if args.use_smoketest else "yamltest.dir/RIR"), pattern))
+
+def snapshot_to_serial(fn):
+  return os.path.splitext(os.path.basename(fn))[0]
+
+def delta_to_serial(fn):
+  return os.path.splitext(os.path.basename(fn))[0].split("-")[1]
+
+top = os.path.expanduser("~/rpki/subvert-rpki.hactrn.net/branches/tk705")
+
+rrdp_test_tool = os.path.join(top, "potpourri/rrdp-test-tool")
+rcynic         = os.path.join(top, "rp/rcynic/rcynic")
+rcynic_text    = os.path.join(top, "rp/rcynic/rcynic-text")
+
+with open("rcynic-rrdp.conf", "w") as f:
+  f.write(textwrap.dedent('''# Automatically generated for RRDP tests, do not edit.
+    [rcynic]
+    xml-summary  = rcynic.xml
+    jitter       = 0
+    use-links    = yes
+    use-syslog   = no
+    use-stderr   = yes
+    log-level    = log_debug
+    run-rsync    = no
+    '''))
+  if args.use_smoketest:
+    f.write("trust-anchor = smoketest.dir/root.cer\n")
+  else:
+    f.write("trust-anchor = yamltest.dir/RIR/publication/RIR-root/root.cer\n")
+
+if args.use_smoketest:
+  run("python", "smoketest.py", args.yaml_file)
+else:
+  run("python", "sql-cleaner.py")
+  argv = ("python", "yamltest.py", args.yaml_file)
+  log("Running: " + " ".join(argv))
+  yamltest = subprocess.Popen(argv)
+  log("Sleeping %s" % args.delay)
+  time.sleep(args.delay)
+  yamltest.terminate()
+
+snapshots = dict((int(snapshot_to_serial(fn)), fn) for fn in dataglob("rrdp-publication/*/snapshot/*.xml"))
+deltas    = dict((int(delta_to_serial(fn)),    fn) for fn in dataglob("rrdp-publication/*/deltas/*.xml"))
+
+for snapshot in sorted(snapshots):
+
+  time.sleep(1)
+  run("rm", "-rf", "rcynic-data")
+  run(rrdp_test_tool, snapshots[snapshot])
+  run(rcynic, "-c", "rcynic-rrdp.conf")
+  run(rcynic_text, "rcynic.xml")
+
+  for delta in sorted(deltas):
+    if delta > snapshot:
+      time.sleep(1)
+      run(rrdp_test_tool, deltas[delta])
+      run(rcynic, "-c", "rcynic-rrdp.conf")
+      run(rcynic_text, "rcynic.xml")
+
+  if not args.exhaustive:
+    break