def test_backend_with_health_check(): with CleanupEtcdFolders( ['/upstreams/test-health-check', '/hosts/test-health-check.abc.com']): with MockHttpServer() as node1: _add_upstream('test-health-check', health_uri='/', health_timeout='2s', health_interval='1m') _add_node('test-health-check', 'node1', node1) _add_location('test-health-check.abc.com', 'test-health-check') # Wait for sometime for changes to apply sleep(PROXY_REFRESH_TIME) for protocol in ['http', 'https']: resp = _request_proxy('test-health-check.abc.com', protocol=protocol) assert_equals(resp.status_code, 200)
def test_tcp_proxy_backend(): upstream = 'test-proxy-tcp-backend' with CleanupEtcdFolders( ['/upstreams/%s' % upstream, '/listeners/tcp/test-tcp']): with MockHttpServer() as node1: with MockHttpServer() as node2: _add_upstream(upstream, mode='tcp') _add_tcp_listener('test-tcp', '*:%d' % MOCK_TCP_PORT, upstream) _add_node(upstream, 'node1', node1) _add_node(upstream, 'node2', node2) # Wait 5s for changes to apply sleep(PROXY_REFRESH_TIME) for protocol in ['http']: resp = _request_proxy('localhost', protocol=protocol, port=MOCK_TCP_PORT) assert_equals(resp.status_code, 200)