add transaction tests

Author: greg-rychlewski

test/query_test.exs

@@ -1978,7 +1978,21 @@ defmodule QueryTest do
     assert {:ok, _} = P.prepare(pid, "42", "SELECT 42")
   end
 
-  test "disconnect_and_retry with transaction" do
+  test "disconnect_and_retry with prepare in transaction" do
+    # Start new connection so we can retry on disconnect
+    opts = [database: "postgrex_test", backoff_min: 1, backoff_max: 1]
+    {:ok, pid} = P.start_link(opts)
+
+    # Dropping socket in middle of transaction should return error
+    fun = fn conn ->
+      disconnect(conn)
+      P.prepare(conn, "42", "SELECT 42")
+    end
+
+    assert {:error, :rollback} = P.transaction(pid, fun)
+  end
+
+  test "disconnect_and_retry with begin transaction" do
     # Start new connection so we can retry on disconnect
     opts = [database: "postgrex_test", backoff_min: 1, backoff_max: 1]
     {:ok, pid} = P.start_link(opts)
@@ -2005,7 +2019,24 @@ defmodule QueryTest do
     assert :ok = P.close(pid, query)
   end
 
-  test "disconnect_and_retry on attempting execution of prepared statement" do
+  test "disconnect_and_retry with closing prepared statement in transaction" do
+    # Start new connection so we can retry on disconnect
+    opts = [database: "postgrex_test", backoff_min: 1, backoff_max: 1]
+    {:ok, pid} = P.start_link(opts)
+
+    # Prepare query that we wil try to close after disconnecting
+    {:ok, query} = P.prepare(pid, "42", "SELECT 42")
+
+    # Dropping socket in middle of transaction should return error
+    fun = fn conn ->
+      disconnect(conn)
+      P.close(conn, query)
+    end
+
+    assert {:error, :rollback} = P.transaction(pid, fun)
+  end
+
+  test "disconnect_and_retry with binding prepared statement" do
     # Start new connection so we can retry on disconnect
     opts = [database: "postgrex_test", backoff_min: 1, backoff_max: 1]
     {:ok, pid} = P.start_link(opts)
@@ -2020,6 +2051,28 @@ defmodule QueryTest do
     assert {:ok, _, _} = P.execute(pid, query, [])
   end
 
+  test "disconnect_and_retry with binding prepared statement in transaction" do
+    # Start new connection so we can retry on disconnect
+    opts = [database: "postgrex_test", backoff_min: 1, backoff_max: 1]
+    {:ok, pid} = P.start_link(opts)
+
+    # Prepare query that we wil try to execute after disconnecting
+    {:ok, query} = P.prepare(pid, "42", "SELECT 42")
+
+    # Dropping socket in middle of transaction should return error
+    fun = fn conn ->
+      disconnect(conn)
+      P.execute(conn, query, [])
+    end
+
+    assert {:error, :rollback} = P.transaction(pid, fun)
+  end
+
+  defp disconnect(%DBConnection{} = conn) do
+    sock = get_socket(conn)
+    :gen_tcp.shutdown(sock, :read_write)
+  end
+
   defp disconnect(pid) do
     sock = DBConnection.run(pid, &get_socket/1)
     :gen_tcp.shutdown(sock, :read_write)