tinygo test: simplify output buffering

This commit is contained in:
Nia Waldvogel
2021-12-30 10:17:35 -05:00
committed by Nia
parent f1b15db258
commit f9f2349850
+12 -16
View File
@@ -1641,15 +1641,13 @@ func (out *testStdout) Write(data []byte) (int, error) {
defer buf.mu.Unlock()
// Append the output.
var prev []byte
if len(buf.output) > 0 && !buf.output[len(buf.output)-1].stderr {
prev = buf.output[len(buf.output)-1].data
buf.output = buf.output[:len(buf.output)-1]
if len(buf.output) == 0 || buf.output[len(buf.output)-1].stderr {
buf.output = append(buf.output, outputEntry{
stderr: false,
})
}
buf.output = append(buf.output, outputEntry{
stderr: false,
data: append(prev, data...),
})
last := &buf.output[len(buf.output)-1]
last.data = append(last.data, data...)
return len(data), nil
}
@@ -1674,15 +1672,13 @@ func (out *testStderr) Write(data []byte) (int, error) {
defer buf.mu.Unlock()
// Append the output.
var prev []byte
if len(buf.output) > 0 && buf.output[len(buf.output)-1].stderr {
prev = buf.output[len(buf.output)-1].data
buf.output = buf.output[:len(buf.output)-1]
if len(buf.output) == 0 || !buf.output[len(buf.output)-1].stderr {
buf.output = append(buf.output, outputEntry{
stderr: true,
})
}
buf.output = append(buf.output, outputEntry{
stderr: true,
data: append(prev, data...),
})
last := &buf.output[len(buf.output)-1]
last.data = append(last.data, data...)
return len(data), nil
}