aboutsummaryrefslogtreecommitdiff
path: root/rpkid/testbed.py
diff options
context:
space:
mode:
Diffstat (limited to 'rpkid/testbed.py')
-rw-r--r--rpkid/testbed.py6
1 files changed, 2 insertions, 4 deletions
diff --git a/rpkid/testbed.py b/rpkid/testbed.py
index c2be102c..698f986e 100644
--- a/rpkid/testbed.py
+++ b/rpkid/testbed.py
@@ -758,8 +758,7 @@ class allocation(object):
assert isinstance(pdus, (list, tuple))
assert self.rpki_port is not None
- msg = rpki.left_right.msg(pdus)
- msg.type = "query"
+ msg = rpki.left_right.msg.query(pdus)
cms, xml = rpki.left_right.cms_msg.wrap(msg, self.irbe_key, self.irbe_cert,
pretty_print = True)
rpki.log.debug(xml)
@@ -1146,8 +1145,7 @@ def call_pubd(pdus, cb):
response.
"""
rpki.log.info("Calling pubd")
- msg = rpki.publication.msg(pdus)
- msg.type = "query"
+ msg = rpki.publication.msg.query(pdus)
cms, xml = rpki.publication.cms_msg.wrap(msg, pubd_irbe_key, pubd_irbe_cert,
pretty_print = True)
rpki.log.debug(xml)