Checkpoint

svn path=/scripts/testdb.py; revision=1399

1 files changed, 19 insertions, 2 deletions

diff --git a/scripts/testdb.py b/scripts/testdb.py
index a19997fb..345cbaee 100644
--- a/scripts/testdb.py
+++ b/scripts/testdb.py
@@ -1,6 +1,6 @@
 # $Id$
 
-import rpki.resource_set, os, yaml
+import rpki.resource_set, os, yaml, MySQLdb
 
 debug          = True
 irbe_name      = "testdb"
@@ -10,6 +10,8 @@ max_twigs      = 10
 irdb_base_port = 4400
 rpki_base_port = irdb_base_port + max_twigs
 root_port      = rpki_base_port + max_twigs
+rpki_sql_file  = "../docs/rpki-db-schema.sql"
+irdb_sql_file  = "../docs/sample-irdb.sql"
 
 def main():
 
@@ -31,7 +33,15 @@ def main():
     if a.is_twig():
       a.setup_conf_file()
 
-  # 2: Initialize sql for rpki.py and irdb.py instances
+  # Initialize sql for rpki.py and irdb.py instances
+
+  rpki_sql = open(rpki_sql_file).read()
+  irdb_sql = open(irdb_sql_file).read()
+  
+  for a in db:
+    if a.is_twig():
+      a.setup_sql(rpki_sql, irdb_sql)
+
   # 4: Populate IRDB(s)
   # 5: Start RPKI and IRDB instances
   # 6: Create objects in RPKI engines
@@ -146,6 +156,13 @@ class allocation(object):
       f.write(s)
       f.close()
 
+  def setup_sql(self, rpki_sql, irdb_sql):
+    for name, user, passwd, sql in ((self.rpki_db_name, "rpki", rpki_db_pass, rpki_sql),
+                                    (self.irdb_db_name, "irdb", irdb_db_pass, irdb_sql)):
+      db = MySQLdb.connect(user = user, db = name, passwd = passwd)
+      db.cursor().execute(sql)
+      db.close()
+
 def setup_biz_cert_chain(name):
   s = ""
   for kind in ("EE", "CA", "TA"):