Port backup tests to python

_integration-test/backup.py

@@ -0,0 +1,30 @@
+import subprocess
+import os
+import pytest
+
+@pytest.fixture()
+def setup_env():
+ os.environ['SENTRY_DOCKER_IO_DIR'] = os.path.join(os.getcwd(), 'sentry')
+ os.environ['SKIP_USER_CREATION'] = "1"
+
+def test_backup(setup_env):
+ # 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()
+ 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)
+ assert os.path.getsize(file_path) > 0
+
+def test_import(setup_env):
+ # 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", "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", "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)

_integration-test/conftest.py

@@ -0,0 +1,47 @@
+import subprocess
+import os
+import time
+import httpx
+import pytest
+
+SENTRY_CONFIG_PY = "sentry/sentry.conf.py"
+SENTRY_TEST_HOST = os.getenv("SENTRY_TEST_HOST", "http:https://localhost:9000")
+TEST_USER = "[email protected]"
+TEST_PASS = "test123TEST"
+TIMEOUT_SECONDS = 60
+
+
+@pytest.fixture(scope="session", autouse=True)
+def configure_self_hosted_environment():
+ subprocess.run(["docker", "compose", "--ansi", "never", "up", "-d"], check=True)
+ for i in range(TIMEOUT_SECONDS):
+ try:
+ response = httpx.get(SENTRY_TEST_HOST, follow_redirects=True)
+ except httpx.ConnectionError:
+ time.sleep(1)
+ else:
+ if response.status_code == 200:
+ break
+ else:
+ raise AssertionError("timeout waiting for self-hosted to come up")
+
+ # Create test user
+ subprocess.run(
+ [
+ "docker",
+ "compose",
+ "exec",
+ "web",
+ "sentry",
+ "createuser",
+ "--force-update",
+ "--superuser",
+ "--email",
+ TEST_USER,
+ "--password",
+ TEST_PASS,
+ "--no-input",
+ ],
+ check=True,
+ text=True,
+ )

_integration-test/run.py

@@ -46,42 +46,6 @@ def get_sentry_dsn(client: httpx.Client) -> str:
  return sentry_dsn
 
 
-@pytest.fixture(scope="session", autouse=True)
-def configure_self_hosted_environment():
- subprocess.run(["docker", "compose", "--ansi", "never", "up", "-d"], check=True)
- for i in range(TIMEOUT_SECONDS):
- try:
- response = httpx.get(SENTRY_TEST_HOST, follow_redirects=True)
- except httpx.ConnectionError:
- time.sleep(1)
- else:
- if response.status_code == 200:
- break
- else:
- raise AssertionError("timeout waiting for self-hosted to come up")
-
- # Create test user
- subprocess.run(
- [
- "docker",
- "compose",
- "exec",
- "web",
- "sentry",
- "createuser",
- "--force-update",
- "--superuser",
- "--email",
- TEST_USER,
- "--password",
- TEST_PASS,
- "--no-input",
- ],
- check=True,
- text=True,
- )
-
-
 @pytest.fixture()
 def client_login():
  client = httpx.Client()

integration-test.sh

@@ -17,7 +17,7 @@ if [[ "$test_option" == "--initial-install" ]]; then
  echo "Testing initial install"
  pytest --reruns 5 _integration-test/run.py
  source _integration-test/ensure-customizations-not-present.sh
- source _integration-test/ensure-backup-restore-works.sh
+ pytest _integration-test/backup.py
 elif [[ "$test_option" == "--customizations" ]]; then
  echo "Testing customizations"
  $dc up -d
@@ -36,5 +36,5 @@ EOT
  ./install.sh
  pytest --reruns 5 _integration-test/run.py
  source _integration-test/ensure-customizations-work.sh
- source _integration-test/ensure-backup-restore-works.sh
+ pytest _integration-test/backup.py
 fi