From 24d216cd7f06f037423c82dd3d0aa6ffe82378cb Mon Sep 17 00:00:00 2001 From: Steven Fackler Date: Wed, 20 Jun 2018 21:08:08 -0400 Subject: [PATCH] simplify test a bit --- tokio-postgres/tests/test.rs | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/tokio-postgres/tests/test.rs b/tokio-postgres/tests/test.rs index 168baf1d..50c22c7c 100644 --- a/tokio-postgres/tests/test.rs +++ b/tokio-postgres/tests/test.rs @@ -46,10 +46,14 @@ fn insert_select() { let connection = connection.map_err(|e| panic!("{}", e)); runtime.handle().spawn(connection).unwrap(); - let create = client.prepare("CREATE TEMPORARY TABLE foo (id SERIAL, name TEXT)"); - let create = runtime.block_on(create).unwrap(); - let create = client.execute(&create, &[]).map(|n| assert_eq!(n, 0)); - runtime.block_on(create).unwrap(); + runtime + .block_on( + client + .prepare("CREATE TEMPORARY TABLE foo (id SERIAL, name TEXT)") + .and_then(|create| client.execute(&create, &[])) + .map(|n| assert_eq!(n, 0)), + ) + .unwrap(); let insert = client.prepare("INSERT INTO foo (name) VALUES ($1), ($2)"); let select = client.prepare("SELECT id, name FROM foo ORDER BY id");