def setUp(self):
setup_asyncio_reactor(self)
prometheus.reset()
self.loop = asyncio.get_event_loop()
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_client_connections') == 0
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_connection_made') == 0
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_connection_ready', {'ident': 'test'}) is None
self.sock = sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(('127.0.0.1', 0))
self.port = sock.getsockname()[1]
authenticator = Authenticator({
'test': {
'secret': 'secret',
'subchans': ['test-chan'],
'pubchans': ['test-chan'],
'owner': 'some-owner',
}
})
self.server = Server(authenticator, sock=self.sock)
def setUp(self):
prometheus.reset()
assert prometheus.REGISTRY.get_sample_value(
'hpfeeds_broker_client_connections') == 0
import ssl
ssl_context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
ssl_context.load_cert_chain('hpfeeds/tests/testcert.crt',
'hpfeeds/tests/testcert.key')
self.sock = sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(('127.0.0.1', 0))
self.port = sock.getsockname()[1]
authenticator = Authenticator({
'test': {
'secret': 'secret',
'subchans': ['test-chan'],
'pubchans': ['test-chan'],
'owner': 'some-owner',
}
})
self.server = Server(authenticator, sock=self.sock, ssl=ssl_context)
def test_subscribe_and_publish(self):
prometheus.reset()
async def inner():
self.log.debug('Starting server')
server_future = asyncio.ensure_future(self.server.serve_forever())
await self.server.when_started
self.port = self.server.endpoints[0]['port']
import ssl
ssl_context = ssl.create_default_context(ssl.Purpose.SERVER_AUTH, cafile='tests/testcert.crt')
ssl_context.check_hostname = False
self.log.debug('Creating client service')
client = ClientSession('127.0.0.1', self.port, 'test', 'secret', ssl=ssl_context)
client.subscribe('test-chan')
# Wait till client connected
await client.when_connected
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_client_connections') == 1
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_connection_made') == 1
self.log.debug('Publishing test message')
client.publish('test-chan', b'test message')
self.log.debug('Waiting for read()')
assert ('test', 'test-chan', b'test message') == await client.read()
# We would test this after call to subscribe, but need to wait until sure server has processed command
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_subscriptions', {'ident': 'test', 'chan': 'test-chan'}) == 1
# This will only have incremented when server has processed auth message
# Test can only reliably assert this is the case after reading a message
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_connection_ready', {'ident': 'test'}) == 1
self.log.debug('Stopping client')
await client.close()
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_connection_send_buffer_fill', {'ident': 'test'}) == 12
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_connection_send_buffer_drain', {'ident': 'test'}) == 32
self.log.debug('Stopping server')
server_future.cancel()
await server_future
asyncio.get_event_loop().run_until_complete(inner())
assert len(self.server.connections) == 0, 'Connection left dangling'
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_client_connections') == 0
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_connection_lost', {'ident': 'test'}) == 1
# Closing should auto unsubscribe
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_subscriptions', {'ident': 'test', 'chan': 'test-chan'}) == 0
def setUp(self):
prometheus.reset()
assert prometheus.REGISTRY.get_sample_value(
'hpfeeds_broker_client_connections') == 0
self.sock = sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(('127.0.0.1', 0))
self.port = sock.getsockname()[1]
self.server_thread = threading.Thread(target=self._server_thread, )
self.server_thread.start()
def setUp(self):
prometheus.reset()
authenticator = Authenticator({
'test': {
'secret': 'secret',
'subchans': ['test-chan'],
'pubchans': ['test-chan'],
'owner': 'some-owner',
}
})
self.server = Server(authenticator, bind='127.0.0.1:20000')
def setUp(self):
prometheus.reset()
assert prometheus.REGISTRY.get_sample_value('hpfeeds_broker_client_connections') == 0
self.sock = sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(('127.0.0.1', 0))
self.port = sock.getsockname()[1]
authenticator = self.make_authenticator()
self.server = Server(authenticator)
self.server.add_endpoint_test(sock)
def setUp(self):
prometheus.reset()
authenticator = Authenticator({
'test': {
'secret': 'secret',
'subchans': ['test-chan'],
'pubchans': ['test-chan'],
'owner': 'some-owner',
}
})
self.server = Server(authenticator, exporter='127.0.0.1:20001')
self.server.add_endpoint_legacy('127.0.0.1:20000')
def setUp(self):
prometheus.reset()
assert prometheus.REGISTRY.get_sample_value(
'hpfeeds_broker_client_connections') == 0
self.sock = sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(('127.0.0.1', 0))
self.port = sock.getsockname()[1]
authenticator = Authenticator({
'test': {
'secret': 'secret',
'subchans': ['test-chan'],
'pubchans': ['test-chan'],
'owner': 'some-owner',
}
})
self.server = Server(authenticator, sock=self.sock)
