| ... | ... |
@@ -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"))
|