From b36bda1fc9a48a4c43514dbbe5363db43cdd05f6 Mon Sep 17 00:00:00 2001
From: Tom Dohrmann <erbse.13@gmx.de>
Date: Fri, 20 Mar 2020 21:05:25 +0100
Subject: [PATCH] add test for fetch one messing up stream states

---
 tests/mysql.rs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/tests/mysql.rs b/tests/mysql.rs
index 6c059a6f8..c6e17e9a5 100644
--- a/tests/mysql.rs
+++ b/tests/mysql.rs
@@ -210,3 +210,21 @@ async fn pool_smoke_test() -> anyhow::Result<()> {
 
     Ok(())
 }
+
+#[cfg_attr(feature = "runtime-async-std", async_std::test)]
+#[cfg_attr(feature = "runtime-tokio", tokio::test)]
+async fn test_fetch_one_and_ping() -> anyhow::Result<()> {
+    let mut conn = new::<MySql>().await?;
+
+    let (_id,): (i32,) = sqlx::query_as("SELECT 1 as id")
+        .fetch_one(&mut conn)
+        .await?;
+
+    conn.ping().await?;
+
+    let (_id,): (i32,) = sqlx::query_as("SELECT 1 as id")
+        .fetch_one(&mut conn)
+        .await?;
+
+    Ok(())
+}