update readme

README.md

@@ -21,38 +21,42 @@ import (
 )
 
 func main() {
-    agent := gopubsub.NewAgent()
-    defer agent.Close()
-
-    // Subscribe to a topic
-    sub,_ := agent.Subscribe("foo")
-    sub2,cancel := agent.Subscribe("foo")
-    defer agent.Unsubscribe(sub2)
-    sub3 := agent.Subscribe("foo")
-    defer agent.Unsubscribe(sub3)
-
-    var wg = &sync.WaitGroup{}
-
-    wg.Add(3)
-    go func() {
-        defer wg.Done()
-        fmt.Println(<-sub.Msg)
-    }()
-
-    go func() {
-        defer wg.Done()
-        fmt.Println(<-sub2.Msg)
-    }()
-
-    go func() {
-        defer wg.Done()
-        fmt.Println(<-sub3.Msg)
-    }()
-
-
-    // Publish a message to the topic
-    go agent.Publish("foo", "hello world")
-
-    wg.Wait()
+    // channel to publish messages to
+	// Create a new agent
+	agent := gopubsub.NewAgent()
+	defer agent.Close()
+
+	// Subscribe to a topic
+	sub, _ := agent.Subscribe("foo")
+	sub2, cancel := agent.Subscribe("foo")
+	defer cancel(agent, sub2) // remember to unsubscribe !
+	sub3, cancel := agent.Subscribe("foo")
+	defer cancel(agent, sub3)
+
+	// Publish a message to the topic
+	var wg = &sync.WaitGroup{}
+
+	wg.Add(3)
+	go func() {
+		defer wg.Done()
+		fmt.Println(<-sub.Msg)
+	}()
+
+	go func() {
+		defer wg.Done()
+		fmt.Println(<-sub2.Msg)
+	}()
+
+	go func() {
+		defer wg.Done()
+		fmt.Println(<-sub3.Msg)
+	}()
+
+	go agent.Publish("foo", "hello world")
+
+	wg.Wait()
+
+	agent.Unsubscribe(sub)
+	fmt.Println("subscrip closed suber")
 }
 ```

test/pubsub_test.go

@@ -11,14 +11,13 @@ import (
 func TestChanSub(t *testing.T) {
 	// channel to publish messages to
 	// Create a new agent
-
 	agent := gopubsub.NewAgent()
 	defer agent.Close()
 
 	// Subscribe to a topic
 	sub, _ := agent.Subscribe("foo")
 	sub2, cancel := agent.Subscribe("foo")
-	defer cancel(agent, sub2)           // remember to unsubscribe !
+	defer cancel(agent, sub2) // remember to unsubscribe !
 	sub3, cancel := agent.Subscribe("foo")
 	defer cancel(agent, sub3)
 
@@ -54,9 +53,11 @@ func TestChanSub(t *testing.T) {
 		fmt.Println("channel is closed")
 	}
 
-	suber4,cancel := agent.Subscribe("foo2")
-	defer cancel(agent,suber4)
+	suber4, cancel := agent.Subscribe("foo2")
+	defer cancel(agent, suber4)
 	for _ = range 11 {
-		agent.Publish("foo2","hello")
+		if fail := agent.Publish("foo2", "hello"); fail != nil {
+			panic(fail)
+		}
 	}
 }