Skip to content

Handle write functions and advisory locks #181

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
merged 9 commits into from
May 22, 2025
Merged

Handle write functions and advisory locks #181

Changes from 1 commit
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
5b7d153
More read/write tests
levkk May 22, 2025
b43e1b6
save
levkk May 22, 2025
d959335
save
levkk May 22, 2025
892461f
Add a dirty guard test
levkk May 22, 2025
b5b6d27
Test
levkk May 22, 2025
d26119f
add another test
levkk May 22, 2025
da58475
test
levkk May 22, 2025
810c0b4
test unlocked by default
levkk May 22, 2025
b0908a6
really test
levkk May 22, 2025
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
Prev Previous commit
Next Next commit
add another test
  • Loading branch information
@levkk
levkk committed May 22, 2025
commit d26119fa8219291aa8aa377dd42112d7dc22429d
20 changes: 14 additions & 6 deletions pgdog/src/frontend/client/test/mod.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -405,12 +405,17 @@ async fn test_abrupt_disconnect() {
async fn test_lock_session() {
let (mut conn, mut client, mut inner) = new_client!(true);

conn.write_all(&buffer!({ Query::new("SELECT pg_advisory_lock(1234)") }))
.await
.unwrap();

client.buffer().await.unwrap();
client.client_messages(inner.get()).await.unwrap();
conn.write_all(&buffer!(
{ Query::new("SET application_name TO 'blah'") },
{ Query::new("SELECT pg_advisory_lock(1234)") }
))
.await
.unwrap();

for _ in 0..2 {
client.buffer().await.unwrap();
client.client_messages(inner.get()).await.unwrap();
}

for c in ['T', 'D', 'C', 'Z'] {
let msg = inner.backend.read().await.unwrap();
Expand All @@ -421,4 +426,7 @@ async fn test_lock_session() {
// Session locked.
assert!(inner.backend.is_dirty());
assert!(!inner.backend.done());
assert!(client.params.contains_key("application_name"));

inner.disconnect();
}
Loading