fix(test): Try to unflake the test_graceful_shutdown_with_sqlite_bufer

[olksdr]

Trying to improve and fix the flaky test.
The CI job logs have expired.
However I can reproduce some flakiness locally so I’ve fixed that part I saw in the reproductions.

fix: RELAY-176

[linear-code]

RELAY-176 [Flaky CI]: test_graceful_shutdown_with_sqlite_buffer

[cursor]

Cursor Bugbot has reviewed your changes and found 1 potential issue.

Bugbot Autofix prepared a fix for the issue found in the latest run.

• ✅ Fixed: sqlite3 context manager doesn't close the connection
  • Replaced with sqlite3.connect(...) with an explicit connection and finally: conn.close() so the SQLite handle is always closed before temporary directory cleanup.

Or push these changes by commenting:

@cursor push 9668ec0f9f

Preview (9668ec0f9f)

diff --git a/tests/integration/test_basic.py b/tests/integration/test_basic.py
--- a/tests/integration/test_basic.py
+++ b/tests/integration/test_basic.py
@@ -85,10 +85,13 @@
         assert mini_sentry.captured_envelopes.empty()
 
         # Check if there's data in the SQLite table `envelopes`.
-        with sqlite3.connect(db_file_path) as conn:
+        conn = sqlite3.connect(db_file_path)
+        try:
             cursor = conn.cursor()
             cursor.execute("SELECT COALESCE(SUM(count), 0) FROM envelopes")
             row_count = cursor.fetchone()[0]
+        finally:
+            conn.close()
 
         assert (
             row_count == n

_{This Bugbot Autofix run was free. To enable autofix for future PRs, go to the Cursor dashboard.}