timeout toxic drops more than just the first segment

Shopify · Apr 4, 2017 · 7cda467 · 7cda467
1 parent 476c3aa
commit 7cda467
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 12 deletions.
diff --git a/toxics/timeout.go b/toxics/timeout.go
@@ -24,11 +24,13 @@ func (t *TimeoutToxic) Pipe(stub *ToxicStub) {
 			}
 		}
 	} else {
-		select {
-		case <-stub.Interrupt:
-			return
-		case <-stub.Input:
-			// Drop the data on the ground.
+		for {
+			select {
+			case <-stub.Interrupt:
+				return
+			case <-stub.Input:
+				// Drop the data on the ground.
+			}
 		}
 	}
 }

diff --git a/toxics/timeout_test.go b/toxics/timeout_test.go
@@ -70,16 +70,17 @@ func WithEstablishedProxy(t *testing.T, f func(net.Conn, net.Conn, *toxiproxy.Pr
 func TestTimeoutToxicDoesNotCauseHang(t *testing.T) {
 	WithEstablishedProxy(t, func(conn, _ net.Conn, proxy *toxiproxy.Proxy) {
 		proxy.Toxics.AddToxicJson(ToxicToJson(t, "might_block", "latency", "upstream", &toxics.LatencyToxic{Latency: 10}))
-		proxy.Toxics.AddToxicJson(ToxicToJson(t, "to_delete", "timeout", "upstream", &toxics.TimeoutToxic{Timeout: 0}))
+		proxy.Toxics.AddToxicJson(ToxicToJson(t, "timeout", "timeout", "upstream", &toxics.TimeoutToxic{Timeout: 0}))
 
-		_, err := conn.Write([]byte("hello"))
-		if err != nil {
-			t.Fatal("Unable to write to proxy", err)
+		for i := 0; i < 5; i++ {
+			_, err := conn.Write([]byte("hello"))
+			if err != nil {
+				t.Fatal("Unable to write to proxy", err)
+			}
+			time.Sleep(200 * time.Millisecond) // Shitty sync waiting for latency toxi to get data.
 		}
 
-		time.Sleep(1 * time.Second) // Shitty sync waiting for latency toxi to get data.
-
-		err = testhelper.TimeoutAfter(time.Second, func() {
+		err := testhelper.TimeoutAfter(time.Second, func() {
 			proxy.Toxics.RemoveToxic("might_block")
 		})
 		if err != nil {