test_dhparam_generation.py 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. import pytest
  2. import os
  3. import docker
  4. import time
  5. docker_client = docker.from_env()
  6. def wait_for_nginxproxy_to_be_ready():
  7. """
  8. If one (and only one) container started from image jwilder/nginx-proxy:test is found,
  9. wait for its log to contain substring "Watching docker events"
  10. """
  11. containers = docker_client.containers.list(filters={"ancestor": "jwilder/nginx-proxy:test"})
  12. if len(containers) != 1:
  13. return
  14. container = containers[0]
  15. for line in container.logs(stream=True):
  16. if "Watching docker events" in line:
  17. break
  18. def test_dhparam_is_generated_if_missing(docker_compose, nginxproxy):
  19. wait_for_nginxproxy_to_be_ready()
  20. containers = docker_client.containers.list(filters={"ancestor": "jwilder/nginx-proxy:test"})
  21. if len(containers) != 1:
  22. assert 0
  23. return
  24. sut_container = containers[0]
  25. docker_logs = sut_container.logs(stdout=True, stderr=True, stream=False, follow=False)
  26. assert "Generating DH parameters" in docker_logs
  27. expected_line = "dhparam generation complete, reloading nginx"
  28. max_wait = 30
  29. sleep_interval = 2
  30. current_wait = 0
  31. while current_wait < max_wait:
  32. docker_logs = sut_container.logs(stdout=True, stderr=True, stream=False, follow=False)
  33. if expected_line in docker_logs:
  34. break
  35. time.sleep(sleep_interval)
  36. current_wait += sleep_interval
  37. # Re-check the logs to get better assert output on failure
  38. assert expected_line in docker_logs
  39. # Make sure the dhparam in use is not the default, pre-generated one
  40. default_checksum = sut_container.exec_run("md5sum /app/dhparam.pem.default").split()
  41. generated_checksum = sut_container.exec_run("md5sum /etc/nginx/dhparam/dhparam.pem").split()
  42. assert default_checksum[0] != generated_checksum[0]