diff options
| -rw-r--r-- | rpkid/testbed.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/rpkid/testbed.py b/rpkid/testbed.py index 25612229..01cd8601 100644 --- a/rpkid/testbed.py +++ b/rpkid/testbed.py @@ -694,6 +694,8 @@ class allocation(object): db.autocommit(True) cur.execute("DELETE FROM registrant_asn") cur.execute("DELETE FROM registrant_net") + cur.execute("DELETE FROM roa_request_prefix") + cur.execute("DELETE FROM roa_request") for s in [self] + self.hosts: for kid in s.kids: cur.execute("SELECT registrant_id FROM registrant WHERE registrant_handle = %s AND registry_handle = %s", (kid.name, s.name)) @@ -705,6 +707,13 @@ class allocation(object): for v6_range in kid.resources.v6: cur.execute("INSERT registrant_net (start_ip, end_ip, version, registrant_id) VALUES (%s, %s, 6, %s)", (v6_range.min, v6_range.max, registrant_id)) cur.execute("UPDATE registrant SET valid_until = %s WHERE registrant_id = %s", (kid.resources.valid_until.to_sql(), registrant_id)) + for r in s.route_origins: + cur.execute("INSERT roa_request (roa_request_handle, as_number) VALUES (%s, %s)", (s.name, r.asn)) + roa_request_id = cur.lastrowid + for version, prefix_set in ((4, r.v4), (6, r.v6)): + if prefix_set: + cur.executemany("INSERT roa_request_prefix (roa_request_id, prefix, prefixlen, max_prefixlen, version) VALUES (%s, %s, %s, %s, %s)", + ((roa_request_id, x.address, x.prefixlen, x.max_prefixlen, version) for x in prefix_set)) db.close() def run_daemons(self): |