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)