def setUp(self, test_interval=BCAST_TEST_INTERVAL):
self.factory = VOEventBroadcasterFactory(DUMMY_SERVICE_IVOID.decode(),
test_interval)
self.factory.alive_loop.clock = task.Clock()
self.factory.test_loop.clock = task.Clock()
self.factory.broadcasters.append(DummyBroadcaster())
self.connector = reactor.listenTCP(0, self.factory)
def test_receive_iamalive(self):
self.proto.stringReceived(DUMMY_IAMALIVE)
received_element = etree.fromstring(self.tr.value()[4:])
self.assertEqual("iamalive", received_element.attrib['role'])
if self.IVOID:
self.assertEqual(DUMMY_SERVICE_IVOID.decode(),
received_element.find('Response').text)
def test_receive_authenticate(self):
self.proto.stringReceived(DUMMY_AUTHENTICATE)
received_element = etree.fromstring(self.tr.value()[4:])
self.assertEqual("authenticate", received_element.attrib['role'])
if self.IVOID:
self.assertEqual(DUMMY_SERVICE_IVOID.decode(),
received_element.find('Response').text)
def test_receive_authenticate(self):
self.proto.stringReceived(DUMMY_AUTHENTICATE)
received_element = etree.fromstring(self.tr.value()[4:])
self.assertEqual("authenticate", received_element.attrib['role'])
if self.IVOID:
self.assertEqual(DUMMY_SERVICE_IVOID.decode(),
received_element.find('Response').text)
def test_local_ivoid(self):
# The local IVOID should be set correctly.
self.config.parseOptions(self.cmd_line)
self.assertEqual(self.config['local_ivo'], DUMMY_SERVICE_IVOID.decode())
# And a bad one rejected.
self._check_bad_parse(["--local-ivo", "bad_ivo"])
def test_receive_iamalive(self):
self.proto.stringReceived(DUMMY_IAMALIVE)
received_element = etree.fromstring(self.tr.value()[4:])
self.assertEqual("iamalive", received_element.attrib['role'])
if self.IVOID:
self.assertEqual(DUMMY_SERVICE_IVOID.decode(),
received_element.find('Response').text)
def test_sendIAmAlive(self):
self.tr.clear()
init_alive_count = self.proto.alive_count
self.factory.alive_loop.clock.advance(self.factory.IAMALIVE_INTERVAL)
received_element = etree.fromstring(self.tr.value()[4:])
self.assertEqual("iamalive", received_element.attrib['role'])
self.assertEqual(DUMMY_SERVICE_IVOID.decode(), received_element.find('Origin').text)
self.assertEqual(self.proto.alive_count - init_alive_count, 1)
def setUp(self, test_interval=BCAST_TEST_INTERVAL):
self.factory = VOEventBroadcasterFactory(
DUMMY_SERVICE_IVOID.decode(), test_interval
)
self.factory.alive_loop.clock = task.Clock()
self.factory.test_loop.clock = task.Clock()
self.factory.broadcasters.append(DummyBroadcaster())
self.connector = reactor.listenTCP(0, self.factory)
def test_local_ivoid(self):
# The local IVOID should be set correctly.
self.config.parseOptions(self.cmd_line)
self.assertEqual(self.config['local_ivo'],
DUMMY_SERVICE_IVOID.decode())
# And a bad one rejected.
self._check_bad_parse(["--local-ivo", "bad_ivo"])
def test_receive_valid_voevent(self):
self.proto.stringReceived(DUMMY_VOEVENT)
received_element = etree.fromstring(self.tr.value()[4:])
self.assertEqual("ack", received_element.attrib['role'])
if self.IVOID:
self.assertEqual(DUMMY_SERVICE_IVOID.decode(),
received_element.find('Response').text)
self.assertEqual(DUMMY_EVENT_IVOID.decode(),
received_element.find('Origin').text)
def test_receive_valid_voevent(self):
self.proto.stringReceived(DUMMY_VOEVENT)
received_element = etree.fromstring(self.tr.value()[4:])
self.assertEqual("ack", received_element.attrib['role'])
if self.IVOID:
self.assertEqual(DUMMY_SERVICE_IVOID.decode(),
received_element.find('Response').text)
self.assertEqual(DUMMY_EVENT_IVOID.decode(),
received_element.find('Origin').text)
def test_sendIAmAlive(self):
self.tr.clear()
init_alive_count = self.proto.alive_count
self.factory.alive_loop.clock.advance(self.factory.IAMALIVE_INTERVAL)
received_element = etree.fromstring(self.tr.value()[4:])
self.assertEqual("iamalive", received_element.attrib['role'])
self.assertEqual(DUMMY_SERVICE_IVOID.decode(),
received_element.find('Origin').text)
self.assertEqual(self.proto.alive_count - init_alive_count, 1)
def test_valid(self):
schema = etree.XMLSchema(
etree.parse(
os.path.join(comet.__path__[0], "schema/VOEvent-v2.0.xsd")
)
)
self.assertTrue(schema.validate(self.message.element))
self.assertEqual(self.message.role, "test")
self.assertTrue(self.message.ivoid.startswith(DUMMY_SERVICE_IVOID.decode()))
def setUp(self):
self.factory = VOEventBroadcasterFactory(DUMMY_SERVICE_IVOID.decode(),
BCAST_TEST_INTERVAL)
self.factory.alive_loop.clock = task.Clock()
self.factory.test_loop.clock = task.Clock()
self.connector = reactor.listenTCP(0, self.factory)
self.proto = self.factory.buildProtocol(('127.0.0.1', 0))
self.tr = proto_helpers.StringTransportWithDisconnection()
self.proto.makeConnection(self.tr)
self.tr.protocol = self.proto
def setUp(self):
self.factory = VOEventBroadcasterFactory(
DUMMY_SERVICE_IVOID.decode(), BCAST_TEST_INTERVAL
)
self.factory.alive_loop.clock = task.Clock()
self.factory.test_loop.clock = task.Clock()
self.connector = reactor.listenTCP(0, self.factory)
self.proto = self.factory.buildProtocol(('127.0.0.1', 0))
self.tr = proto_helpers.StringTransportWithDisconnection()
self.proto.makeConnection(self.tr)
self.tr.protocol = self.proto
def test_receive_invalid_voevent(self):
# This should not be accepted, but *should not* generate a NAK.
def invalid(event): raise Exception("Failed")
self.proto.factory.validators.append(invalid)
self.proto.stringReceived(DUMMY_VOEVENT)
received_element = etree.fromstring(self.tr.value()[4:])
self.assertEqual("ack", received_element.attrib['role'])
if self.IVOID:
self.assertEqual(DUMMY_SERVICE_IVOID.decode(),
received_element.find('Response').text)
self.assertEqual(DUMMY_EVENT_IVOID.decode(),
received_element.find('Origin').text)
def test_receive_invalid_voevent(self):
# This should not be accepted, but *should not* generate a NAK.
def invalid(event):
raise Exception("Failed")
self.proto.factory.validators.append(invalid)
self.proto.stringReceived(DUMMY_VOEVENT)
received_element = etree.fromstring(self.tr.value()[4:])
self.assertEqual("ack", received_element.attrib['role'])
if self.IVOID:
self.assertEqual(DUMMY_SERVICE_IVOID.decode(),
received_element.find('Response').text)
self.assertEqual(DUMMY_EVENT_IVOID.decode(),
received_element.find('Origin').text)
def test_sent_authenticate(self):
received_element = etree.fromstring(self.tr.value()[4:])
self.assertEqual("authenticate", received_element.attrib['role'])
self.assertEqual(DUMMY_SERVICE_IVOID.decode(), received_element.find('Origin').text)
def setUp(self):
self.message = broker_test_message(DUMMY_SERVICE_IVOID.decode())
def test_sent_authenticate(self):
received_element = etree.fromstring(self.tr.value()[4:])
self.assertEqual("authenticate", received_element.attrib['role'])
self.assertEqual(DUMMY_SERVICE_IVOID.decode(),
received_element.find('Origin').text)
def setUp(self):
self.config = Options()
self.cmd_line = ["--local-ivo", DUMMY_SERVICE_IVOID.decode()]
def setUp(self):
self.config = Options()
# Note that parsing from the command line always results in a str,
# regardless of whether we're running Python 2.
self.cmd_line = ["--local-ivo", DUMMY_SERVICE_IVOID.decode()]
def test_valid_ivoid(self):
self.config.parseOptions(self.cmd_line)
self.assertEqual(self.config['local-ivo'], DUMMY_SERVICE_IVOID.decode())
def setUp(self):
self.message = VOEventMessage.broker_test(DUMMY_SERVICE_IVOID.decode())
def setUp(self):
self.config = Options()
self.cmd_line = ["--local-ivo", DUMMY_SERVICE_IVOID.decode()]