crowdsec/docker/test/tests/test_flavors.py
2025-02-17 11:04:26 +01:00

58 lines
2.2 KiB
Python

"""
Test basic behavior of all the image variants
"""
from http import HTTPStatus
import pytest
pytestmark = pytest.mark.docker
def test_cscli_lapi(crowdsec, flavor: str) -> None:
"""Test if cscli can talk to lapi"""
with crowdsec(flavor=flavor) as cs:
cs.wait_for_log("*Starting processing data*")
cs.wait_for_http(8080, "/health", want_status=HTTPStatus.OK)
x = cs.cont.exec_run("cscli lapi status")
assert x.exit_code == 0
stdout = x.output.decode()
assert "You can successfully interact with Local API (LAPI)" in stdout
@pytest.mark.skip(reason="currently broken by hub upgrade")
def test_flavor_content(crowdsec, flavor: str) -> None:
"""Test flavor contents"""
with crowdsec(flavor=flavor) as cs:
cs.wait_for_log("*Starting processing data*")
cs.wait_for_http(8080, "/health", want_status=HTTPStatus.OK)
x = cs.cont.exec_run("ls -1 /var/lib/crowdsec/data/")
assert x.exit_code == 0
stdout = x.output.decode()
if "slim" in flavor or "plugins" in flavor:
assert "GeoLite2-City.mmdb" not in stdout
assert "GeoLite2-ASN.mmdb" not in stdout
else:
assert "GeoLite2-City.mmdb" in stdout
assert "GeoLite2-ASN.mmdb" in stdout
assert "crowdsec.db" in stdout
x = cs.cont.exec_run("ls -1 /usr/local/lib/crowdsec/plugins/")
stdout = x.output.decode()
if "slim" in flavor:
# the exact return code and full message depend
# on the 'ls' implementation (busybox vs coreutils)
assert x.exit_code != 0
assert "No such file or directory" in stdout
assert "notification-email" not in stdout
assert "notification-http" not in stdout
assert "notification-slack" not in stdout
assert "notification-splunk" not in stdout
assert "notification-sentinel" not in stdout
else:
assert x.exit_code == 0
assert "notification-email" in stdout
assert "notification-http" in stdout
assert "notification-slack" in stdout
assert "notification-splunk" in stdout
assert "notification-sentinel" in stdout