Commit c07e98b7 authored by Andy Goldstein's avatar Andy Goldstein

Fix flakey port forward test

Only simulate data for the data stream, not for the error stream too. Fixes #6529
parent 6836b527
...@@ -138,7 +138,10 @@ func (c *fakeUpgradeConnection) CreateStream(headers http.Header) (httpstream.St ...@@ -138,7 +138,10 @@ func (c *fakeUpgradeConnection) CreateStream(headers http.Header) (httpstream.St
stream := &fakeUpgradeStream{} stream := &fakeUpgradeStream{}
c.streams[headers.Get(api.PortHeader)] = stream c.streams[headers.Get(api.PortHeader)] = stream
// only simulate data on the data stream for now, not the error stream
if headers.Get(api.StreamType) == api.StreamTypeData {
stream.data = c.portData[headers.Get(api.PortHeader)] stream.data = c.portData[headers.Get(api.PortHeader)]
}
return stream, nil return stream, nil
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment