Port last integration tests to python (#2966)

* port custom ca cert test to python

_integration-test/conftest.py

@@ -1,6 +1,7 @@
-import subprocess
 import os
+import subprocess
 import time
+
 import httpx
 import pytest
 
@@ -10,8 +11,10 @@
 TEST_PASS = "test123TEST"
 TIMEOUT_SECONDS = 60
 
+
 def pytest_addoption(parser):
- parser.addoption("--customizations", default="disabled")
+ parser.addoption("--customizations", default="disabled")
+
 
 @pytest.fixture(scope="session", autouse=True)
 def configure_self_hosted_environment(request):
@@ -28,23 +31,23 @@ def configure_self_hosted_environment(request):
  raise AssertionError("timeout waiting for self-hosted to come up")
 
  if request.config.getoption("--customizations") == "enabled":
- os.environ['TEST_CUSTOMIZATIONS'] = "enabled"
- script_content = '''\
+ os.environ["TEST_CUSTOMIZATIONS"] = "enabled"
+ script_content = """\
 #!/bin/bash
 touch /created-by-enhance-image
 apt-get update
 apt-get install -y gcc libsasl2-dev python-dev libldap2-dev libssl-dev
-'''
+"""
 
- with open('sentry/enhance-image.sh', 'w') as script_file:
+ with open("sentry/enhance-image.sh", "w") as script_file:
  script_file.write(script_content)
  # Set executable permissions for the shell script
- os.chmod('sentry/enhance-image.sh', 0o755)
+ os.chmod("sentry/enhance-image.sh", 0o755)
 
  # Write content to the requirements.txt file
- with open('sentry/requirements.txt', 'w') as req_file:
- req_file.write('python-ldap\n')
- os.environ['MINIMIZE_DOWNTIME'] = "1"
+ with open("sentry/requirements.txt", "w") as req_file:
+ req_file.write("python-ldap\n")
+ os.environ["MINIMIZE_DOWNTIME"] = "1"
  subprocess.run(["./install.sh"], check=True)
  # Create test user
  subprocess.run(
@@ -68,7 +71,8 @@ def configure_self_hosted_environment(request):
  text=True,
  )
 
+
 @pytest.fixture()
 def setup_backup_restore_env_variables():
- os.environ['SENTRY_DOCKER_IO_DIR'] = os.path.join(os.getcwd(), 'sentry')
- os.environ['SKIP_USER_CREATION'] = "1"
+ os.environ["SENTRY_DOCKER_IO_DIR"] = os.path.join(os.getcwd(), "sentry")
+ os.environ["SKIP_USER_CREATION"] = "1"

_integration-test/custom-ca-roots/test.py

@@ -1,15 +1,16 @@
 import unittest
+
 import requests
 
 
 class CustomCATests(unittest.TestCase):
  def test_valid_self_signed(self):
- self.assertEqual(requests.get("https://self.test").text, 'ok')
+ self.assertEqual(requests.get("https://self.test").text, "ok")
 
  def test_invalid_self_signed(self):
  with self.assertRaises(requests.exceptions.SSLError):
  requests.get("https://fail.test")
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
  unittest.main()

_integration-test/test_backup.py

@@ -1,25 +1,69 @@
-import subprocess
 import os
+import subprocess
+
+
+def test_sentry_admin(setup_backup_restore_env_variables):
+ sentry_admin_sh = os.path.join(os.getcwd(), "sentry-admin.sh")
+ output = subprocess.run(
+ [sentry_admin_sh, "--help"], check=True, capture_output=True, encoding="utf8"
+ ).stdout
+ assert "Usage: ./sentry-admin.sh" in output
+ assert "SENTRY_DOCKER_IO_DIR" in output
+
+ output = subprocess.run(
+ [sentry_admin_sh, "permissions", "--help"],
+ check=True,
+ capture_output=True,
+ encoding="utf8",
+ ).stdout
+ assert "Usage: ./sentry-admin.sh permissions" in output
 
 
 def test_backup(setup_backup_restore_env_variables):
  # Docker was giving me permissioning issues when trying to create this file and write to it even after giving read + write access
  # to group and owner. Instead, try creating the empty file and then give everyone write access to the backup file
- file_path = os.path.join(os.getcwd(), 'sentry', 'backup.json')
- sentry_admin_sh = os.path.join(os.getcwd(), 'sentry-admin.sh')
- open(file_path, 'a', encoding='utf8').close()
+ file_path = os.path.join(os.getcwd(), "sentry", "backup.json")
+ sentry_admin_sh = os.path.join(os.getcwd(), "sentry-admin.sh")
+ open(file_path, "a", encoding="utf8").close()
  os.chmod(file_path, 0o666)
  assert os.path.getsize(file_path) == 0
- subprocess.run([sentry_admin_sh, "export", "global", "/sentry-admin/backup.json", "--no-prompt"], check=True)
+ subprocess.run(
+ [
+ sentry_admin_sh,
+ "export",
+ "global",
+ "/sentry-admin/backup.json",
+ "--no-prompt",
+ ],
+ check=True,
+ )
  assert os.path.getsize(file_path) > 0
 
+
 def test_import(setup_backup_restore_env_variables):
  # Bring postgres down and recreate the docker volume
- subprocess.run(["docker", "compose", "--ansi", "never", "stop", "postgres"], check=True)
- subprocess.run(["docker", "compose", "--ansi", "never", "rm", "-f", "-v", "postgres"], check=True)
+ subprocess.run(
+ ["docker", "compose", "--ansi", "never", "stop", "postgres"], check=True
+ )
+ subprocess.run(
+ ["docker", "compose", "--ansi", "never", "rm", "-f", "-v", "postgres"],
+ check=True,
+ )
  subprocess.run(["docker", "volume", "rm", "sentry-postgres"], check=True)
  subprocess.run(["docker", "volume", "create", "--name=sentry-postgres"], check=True)
- subprocess.run(["docker", "compose", "--ansi", "never", "run", "web", "upgrade", "--noinput"], check=True)
+ subprocess.run(
+ ["docker", "compose", "--ansi", "never", "run", "web", "upgrade", "--noinput"],
+ check=True,
+ )
  subprocess.run(["docker", "compose", "--ansi", "never", "up", "-d"], check=True)
- sentry_admin_sh = os.path.join(os.getcwd(), 'sentry-admin.sh')
- subprocess.run([sentry_admin_sh, "import", "global", "/sentry-admin/backup.json", "--no-prompt"], check=True)
+ sentry_admin_sh = os.path.join(os.getcwd(), "sentry-admin.sh")
+ subprocess.run(
+ [
+ sentry_admin_sh,
+ "import",
+ "global",
+ "/sentry-admin/backup.json",
+ "--no-prompt",
+ ],
+ check=True,
+ )