Browse code

fix error in utils tests

Victor Vieux authored on 2013/07/19 22:56:36
Showing 1 changed files
... ...
@@ -60,9 +60,9 @@ func TestWriteBroadcaster(t *testing.T) {
60 60
 
61 61
 	// Test 1: Both bufferA and bufferB should contain "foo"
62 62
 	bufferA := &dummyWriter{}
63
-	writer.AddWriter(bufferA)
63
+	writer.AddWriter(bufferA, "")
64 64
 	bufferB := &dummyWriter{}
65
-	writer.AddWriter(bufferB)
65
+	writer.AddWriter(bufferB, "")
66 66
 	writer.Write([]byte("foo"))
67 67
 
68 68
 	if bufferA.String() != "foo" {
... ...
@@ -76,7 +76,7 @@ func TestWriteBroadcaster(t *testing.T) {
76 76
 	// Test2: bufferA and bufferB should contain "foobar",
77 77
 	// while bufferC should only contain "bar"
78 78
 	bufferC := &dummyWriter{}
79
-	writer.AddWriter(bufferC)
79
+	writer.AddWriter(bufferC, "")
80 80
 	writer.Write([]byte("bar"))
81 81
 
82 82
 	if bufferA.String() != "foobar" {
... ...
@@ -91,35 +91,22 @@ func TestWriteBroadcaster(t *testing.T) {
91 91
 		t.Errorf("Buffer contains %v", bufferC.String())
92 92
 	}
93 93
 
94
-	// Test3: Test removal
95
-	writer.RemoveWriter(bufferB)
96
-	writer.Write([]byte("42"))
97
-	if bufferA.String() != "foobar42" {
98
-		t.Errorf("Buffer contains %v", bufferA.String())
99
-	}
100
-	if bufferB.String() != "foobar" {
101
-		t.Errorf("Buffer contains %v", bufferB.String())
102
-	}
103
-	if bufferC.String() != "bar42" {
104
-		t.Errorf("Buffer contains %v", bufferC.String())
105
-	}
106
-
107
-	// Test4: Test eviction on failure
94
+	// Test3: Test eviction on failure
108 95
 	bufferA.failOnWrite = true
109 96
 	writer.Write([]byte("fail"))
110
-	if bufferA.String() != "foobar42" {
97
+	if bufferA.String() != "foobar" {
111 98
 		t.Errorf("Buffer contains %v", bufferA.String())
112 99
 	}
113
-	if bufferC.String() != "bar42fail" {
100
+	if bufferC.String() != "barfail" {
114 101
 		t.Errorf("Buffer contains %v", bufferC.String())
115 102
 	}
116 103
 	// Even though we reset the flag, no more writes should go in there
117 104
 	bufferA.failOnWrite = false
118 105
 	writer.Write([]byte("test"))
119
-	if bufferA.String() != "foobar42" {
106
+	if bufferA.String() != "foobar" {
120 107
 		t.Errorf("Buffer contains %v", bufferA.String())
121 108
 	}
122
-	if bufferC.String() != "bar42failtest" {
109
+	if bufferC.String() != "barfailtest" {
123 110
 		t.Errorf("Buffer contains %v", bufferC.String())
124 111
 	}
125 112
 
... ...
@@ -141,7 +128,7 @@ func TestRaceWriteBroadcaster(t *testing.T) {
141 141
 	writer := NewWriteBroadcaster()
142 142
 	c := make(chan bool)
143 143
 	go func() {
144
-		writer.AddWriter(devNullCloser(0))
144
+		writer.AddWriter(devNullCloser(0), "")
145 145
 		c <- true
146 146
 	}()
147 147
 	writer.Write([]byte("hello"))