2
0

test_dhparam_generation.py 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. import backoff
  2. import docker
  3. docker_client = docker.from_env()
  4. ###############################################################################
  5. #
  6. # Tests helpers
  7. #
  8. ###############################################################################
  9. @backoff.on_exception(backoff.constant, AssertionError, interval=2, max_tries=15, jitter=None)
  10. def assert_log_contains(expected_log_line):
  11. """
  12. Check that the nginx-proxy container log contains a given string.
  13. The backoff decorator will retry the check 15 times with a 2 seconds delay.
  14. :param expected_log_line: string to search for
  15. :return: None
  16. :raises: AssertError if the expected string is not found in the log
  17. """
  18. sut_container = docker_client.containers.get("nginxproxy")
  19. docker_logs = sut_container.logs(stdout=True, stderr=True, stream=False, follow=False)
  20. assert expected_log_line in docker_logs
  21. ###############################################################################
  22. #
  23. # Tests
  24. #
  25. ###############################################################################
  26. def test_dhparam_is_generated_if_missing(docker_compose):
  27. sut_container = docker_client.containers.get("nginxproxy")
  28. assert sut_container.status == "running"
  29. assert_log_contains("Generating DH parameters")
  30. assert_log_contains("dhparam generation complete, reloading nginx")
  31. # Make sure the dhparam in use is not the default, pre-generated one
  32. default_checksum = sut_container.exec_run("md5sum /app/dhparam.pem.default").split()
  33. generated_checksum = sut_container.exec_run("md5sum /etc/nginx/dhparam/dhparam.pem").split()
  34. assert default_checksum[0] != generated_checksum[0]