Add the drop replication slot functionality

2017-01-09 14:19:08 -08:00
parent c88c110169
commit af01afca00
2 changed files with 5 additions and 7 deletions
@@ -173,6 +173,10 @@ func TestSimpleReplicationConnection(t *testing.T) {
 		t.Errorf("Unexpected write position %d", status.WalWritePosition)
 	}

+	err = replicationConn.DropReplicationSlot("pgx_test")
+	if err != nil {
+		t.Fatalf("Failed to drop replication slot %v", err)
+	}
 	err = replicationConn.Close()
 	if err != nil {
 		t.Fatalf("Replication connection close failed: %v", err)
@@ -183,10 +187,4 @@ func TestSimpleReplicationConnection(t *testing.T) {
 	if integerRestartLsn != maxWal {
 		t.Fatalf("Wal offset update failed, expected %s found %s", pgx.FormatLSN(maxWal), restartLsn)
 	}
-
-	_, err = conn.Exec("select pg_drop_replication_slot($1)", "pgx_test")
-	if err != nil {
-		t.Fatalf("Failed to drop replication slot: %v", err)
-	}
-
 }