Fix flakey port forward test

Only simulate data for the data stream, not for the error stream too.

Fixes #6529
This commit is contained in:
Andy Goldstein 2015-04-09 13:18:09 -04:00
parent 6836b527ab
commit c07e98b7f5

View File

@ -138,7 +138,10 @@ func (c *fakeUpgradeConnection) CreateStream(headers http.Header) (httpstream.St
stream := &fakeUpgradeStream{}
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)]
}
return stream, nil
}