Add test

godbus · Mar 23, 2021 · c88335c · c88335c
1 parent 6f46e0d
commit c88335c
Showing 1 changed file with 34 additions and 2 deletions.
diff --git a/conn_test.go b/conn_test.go
@@ -13,17 +13,43 @@ import (
 )
 
 func TestSessionBus(t *testing.T) {
-	_, err := SessionBus()
+	oldConn, err := SessionBus()
 	if err != nil {
 		t.Error(err)
 	}
+	if err = oldConn.Close(); err != nil {
+		t.Fatal(err)
+	}
+	if oldConn.Connected() {
+		t.Fatal("Should be closed")
+	}
+	newConn, err := SessionBus()
+	if err != nil {
+		t.Error(err)
+	}
+	if newConn == oldConn {
+		t.Fatal("Should get a new connection")
+	}
 }
 
 func TestSystemBus(t *testing.T) {
-	_, err := SystemBus()
+	oldConn, err := SystemBus()
+	if err != nil {
+		t.Error(err)
+	}
+	if err = oldConn.Close(); err != nil {
+		t.Fatal(err)
+	}
+	if oldConn.Connected() {
+		t.Fatal("Should be closed")
+	}
+	newConn, err := SystemBus()
 	if err != nil {
 		t.Error(err)
 	}
+	if newConn == oldConn {
+		t.Fatal("Should get a new connection")
+	}
 }
 
 func TestConnectSessionBus(t *testing.T) {
@@ -34,6 +60,9 @@ func TestConnectSessionBus(t *testing.T) {
 	if err = conn.Close(); err != nil {
 		t.Fatal(err)
 	}
+	if conn.Connected() {
+		t.Fatal("Should be closed")
+	}
 }
 
 func TestConnectSystemBus(t *testing.T) {
@@ -44,6 +73,9 @@ func TestConnectSystemBus(t *testing.T) {
 	if err = conn.Close(); err != nil {
 		t.Fatal(err)
 	}
+	if conn.Connected() {
+		t.Fatal("Should be closed")
+	}
 }
 
 func TestSend(t *testing.T) {