move to defer

f0ssel · f0ssel · commit c18963bd8922 · 2024-08-21T16:22:16.000Z
diff --git a/coderd/database/dbtestutil/driver.go b/coderd/database/dbtestutil/driver.go
@@ -79,9 +79,10 @@ func (d *Driver) AddConnection(conn driver.Conn) {
 
 func (d *Driver) WaitForConnection() {
 	ch := make(chan struct{})
+	defer close(ch)
+	defer delete(d.listeners, ch)
 	d.listeners[ch] = ch
 	<-ch
-	delete(d.listeners, ch)
 }
 
 func (d *Driver) DropConnections() {
