From 2159d1c959a8cec3aa990d83ed9845699e3af571 Mon Sep 17 00:00:00 2001
From: The Surgeon <deadly.surgery@gmail.com>
Date: Sat, 11 Jul 2020 15:43:19 -0500
Subject: [PATCH] Adds Close Function

---
 net/rcon.go      | 2 ++
 net/rcon_test.go | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/net/rcon.go b/net/rcon.go
index f0960c7..651b7e8 100644
--- a/net/rcon.go
+++ b/net/rcon.go
@@ -177,12 +177,14 @@ func (r *RCONConn) RespCmd(resp string) error {
 type RCONClientConn interface {
 	Cmd(cmd string) error
 	Resp() (resp string, err error)
+	Close() error
 }
 
 type RCONServerConn interface {
 	AcceptLogin(password string) error
 	AcceptCmd() (cmd string, err error)
 	RespCmd(resp string) error
+	Close() error
 }
 
 func ListenRCON(addr string) (*RCONListener, error) {
diff --git a/net/rcon_test.go b/net/rcon_test.go
index a93a0e3..e2a45bc 100644
--- a/net/rcon_test.go
+++ b/net/rcon_test.go
@@ -17,6 +17,8 @@ func server(t *testing.T, prepare chan<- int) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	defer l.Close()
+
 	prepare <- 1
 
 	for {
@@ -55,6 +57,7 @@ func client(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	defer conn.Close()
 
 	err = conn.Cmd("TEST COMMAND")
 	if err != nil {