Skip to content

fix: psycopg instrumentation fixes #29

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
sohankshirsagar merged 37 commits into from
Jan 16, 2026
Merged

fix: psycopg instrumentation fixes #29

Show file tree
Hide file tree
Changes from 17 commits
Commits
Show all changes
37 commits
Select commit Hold shift + click to select a range
c8b7bad
fix cursor-stream
sohankshirsagar Jan 13, 2026
1bdc5ca
refactor psycopg instrumentations to use mock utils
sohankshirsagar Jan 13, 2026
132b556
instrument ServerCursor
sohankshirsagar Jan 13, 2026
6e568a1
add psycopg cursor.copy() instrumentation for COPY operations
sohankshirsagar Jan 13, 2026
8feb8a7
fix multiple queries on same cursor
sohankshirsagar Jan 13, 2026
8723755
fix psycopg pipeline mode: defer result capture until sync()
sohankshirsagar Jan 14, 2026
da7329d
fix cursor iteration
sohankshirsagar Jan 14, 2026
8421029
fix executemany with returning=True instrumentation for psycopg
sohankshirsagar Jan 14, 2026
9c4a03e
fix cursor.scroll() support in psycopg replay mode
sohankshirsagar Jan 14, 2026
3f801cf
fix cursor.rownumber property returning null during REPLAY mode
sohankshirsagar Jan 14, 2026
b7c8502
fix psycopg cursor.statusmessage capture and replay
sohankshirsagar Jan 14, 2026
15a32a8
fix nextset() iteration for executemany with returning=True
sohankshirsagar Jan 14, 2026
02a5b45
fix cursor.scroll() position tracking in RECORD mode
sohankshirsagar Jan 14, 2026
acbb31b
refactor
sohankshirsagar Jan 14, 2026
35b3765
resolve cursor reuse hang in RECORD mode
sohankshirsagar Jan 14, 2026
d22ed0a
Fix UUID parameter serialization mismatch in psycopg REPLAY mode
sohankshirsagar Jan 14, 2026
765fd9b
fix bytea serialization to preserve binary data during record/replay
sohankshirsagar Jan 14, 2026
c95337f
Fix kwargs_row row factory handling in psycopg instrumentation
sohankshirsagar Jan 14, 2026
9a407ab
check number of tests replayed in e2e tests
sohankshirsagar Jan 14, 2026
be8de29
Fix scalar_row factory handling in psycopg instrumentation
sohankshirsagar Jan 15, 2026
0a2a1c9
handle replaying uuid properly
sohankshirsagar Jan 15, 2026
f6d4d32
Fix binary format hang by deferring result capture until fetch
sohankshirsagar Jan 15, 2026
8ad42ac
Enable null-values test for psycopg instrumentation
sohankshirsagar Jan 15, 2026
0526956
Clean up null-values test comments
sohankshirsagar Jan 15, 2026
2449d40
Enable transaction context manager test for psycopg instrumentation
sohankshirsagar Jan 15, 2026
a9ab04f
Clean up transaction context manager test comments
sohankshirsagar Jan 15, 2026
7dbf9eb
fix logging OOM issue during e2e tests
sohankshirsagar Jan 15, 2026
5fc28b2
Fix cursor.set_result() not mocked in REPLAY mode for executemany wit…
sohankshirsagar Jan 15, 2026
6903b87
Refactor psycopg instrumentation to reduce code duplication
sohankshirsagar Jan 15, 2026
f04b88b
Fix Decimal and timedelta serialization for consistent RECORD/REPLAY …
sohankshirsagar Jan 15, 2026
0170747
Refactor psycopg instrumentation: extract common helper methods
sohankshirsagar Jan 15, 2026
66453ad
Fix inet/cidr network type serialization for REPLAY mode
sohankshirsagar Jan 15, 2026
3418673
Fix psycopg Range type serialization for REPLAY mode
sohankshirsagar Jan 15, 2026
08f5ce7
fix format + lint issues
sohankshirsagar Jan 15, 2026
e5101ed
fix type errors
sohankshirsagar Jan 15, 2026
70e04b7
format
sohankshirsagar Jan 16, 2026
26f92f7
try catch record mode
sohankshirsagar Jan 16, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions drift/instrumentation/httpx/instrumentation.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -806,6 +806,7 @@ def _try_get_mock_from_request_sync(
input_value=input_value,
kind=SpanKind.CLIENT,
input_schema_merges=input_schema_merges,
is_pre_app_start=not sdk.app_ready,
)

if not mock_response_output or not mock_response_output.found:
Expand Down
Loading
Loading