57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
Test bouncer management: pre-installed, run-time installation and removal.
|
|
"""
|
|
|
|
import hashlib
|
|
from http import HTTPStatus
|
|
import json
|
|
|
|
import pytest
|
|
|
|
pytestmark = pytest.mark.docker
|
|
|
|
|
|
def hex512(s):
|
|
"""Return the sha512 hash of a string as a hex string"""
|
|
return hashlib.sha512(s.encode()).hexdigest()
|
|
|
|
|
|
def test_register_bouncer_env(crowdsec, flavor):
|
|
"""Test installing bouncers at startup, from envvar"""
|
|
|
|
env = {
|
|
'BOUNCER_KEY_bouncer1name': 'bouncer1key',
|
|
'BOUNCER_KEY_bouncer2name': 'bouncer2key'
|
|
}
|
|
|
|
with crowdsec(flavor=flavor, environment=env) as cs:
|
|
cs.wait_for_log("*Starting processing data*")
|
|
cs.wait_for_http(8080, '/health', want_status=HTTPStatus.OK)
|
|
res = cs.cont.exec_run('cscli bouncers list -o json')
|
|
assert res.exit_code == 0
|
|
j = json.loads(res.output)
|
|
assert len(j) == 2
|
|
bouncer1, bouncer2 = j
|
|
assert bouncer1['name'] == 'bouncer1name'
|
|
assert bouncer2['name'] == 'bouncer2name'
|
|
|
|
# add a second bouncer at runtime
|
|
res = cs.cont.exec_run('cscli bouncers add bouncer3name -k bouncer3key')
|
|
assert res.exit_code == 0
|
|
res = cs.cont.exec_run('cscli bouncers list -o json')
|
|
assert res.exit_code == 0
|
|
j = json.loads(res.output)
|
|
assert len(j) == 3
|
|
bouncer3 = j[2]
|
|
assert bouncer3['name'] == 'bouncer3name'
|
|
|
|
# remove all bouncers
|
|
res = cs.cont.exec_run('cscli bouncers delete bouncer1name bouncer2name bouncer3name')
|
|
assert res.exit_code == 0
|
|
res = cs.cont.exec_run('cscli bouncers list -o json')
|
|
assert res.exit_code == 0
|
|
j = json.loads(res.output)
|
|
assert len(j) == 0
|