Port last integration tests to python (getsentry#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,
+    )