mirror of
https://github.com/internetarchive/warcprox.git
synced 2025-01-18 13:22:09 +01:00
let test_choose_a_port_for_me pass when service registry is missing, i.e. when not running with rethinkdb
This commit is contained in:
parent
3d87ed61be
commit
47680cc17d
2
setup.py
2
setup.py
@ -51,7 +51,7 @@ except:
|
|||||||
|
|
||||||
setuptools.setup(
|
setuptools.setup(
|
||||||
name='warcprox',
|
name='warcprox',
|
||||||
version='2.1b1.dev70',
|
version='2.1b1.dev71',
|
||||||
description='WARC writing MITM HTTP/S proxy',
|
description='WARC writing MITM HTTP/S proxy',
|
||||||
url='https://github.com/internetarchive/warcprox',
|
url='https://github.com/internetarchive/warcprox',
|
||||||
author='Noah Levitt',
|
author='Noah Levitt',
|
||||||
|
@ -1369,23 +1369,24 @@ def test_choose_a_port_for_me(service_registry):
|
|||||||
th = threading.Thread(target=controller.run_until_shutdown)
|
th = threading.Thread(target=controller.run_until_shutdown)
|
||||||
th.start()
|
th.start()
|
||||||
try:
|
try:
|
||||||
|
|
||||||
# check that service registry entry lists the correct port
|
|
||||||
start = time.time()
|
|
||||||
ports = []
|
|
||||||
while time.time() - start < 30:
|
|
||||||
svcs = service_registry.available_services('warcprox')
|
|
||||||
ports = [svc['port'] for svc in svcs]
|
|
||||||
if controller.proxy.server_port in ports:
|
|
||||||
break
|
|
||||||
assert controller.proxy.server_port in ports
|
|
||||||
|
|
||||||
# check that the status api lists the correct port
|
# check that the status api lists the correct port
|
||||||
url = 'http://localhost:%s/status' % controller.proxy.server_port
|
url = 'http://localhost:%s/status' % controller.proxy.server_port
|
||||||
response = requests.get(url)
|
response = requests.get(url)
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
status = json.loads(response.content.decode('ascii'))
|
status = json.loads(response.content.decode('ascii'))
|
||||||
assert status['port'] == controller.proxy.server_port
|
assert status['port'] == controller.proxy.server_port
|
||||||
|
|
||||||
|
if service_registry:
|
||||||
|
# check that service registry entry lists the correct port
|
||||||
|
start = time.time()
|
||||||
|
ports = []
|
||||||
|
while time.time() - start < 30:
|
||||||
|
svcs = service_registry.available_services('warcprox')
|
||||||
|
ports = [svc['port'] for svc in svcs]
|
||||||
|
if controller.proxy.server_port in ports:
|
||||||
|
break
|
||||||
|
assert controller.proxy.server_port in ports
|
||||||
|
|
||||||
finally:
|
finally:
|
||||||
controller.stop.set()
|
controller.stop.set()
|
||||||
th.join()
|
th.join()
|
||||||
|
Loading…
x
Reference in New Issue
Block a user