mirror of
https://github.com/coder/coder.git
synced 2025-07-08 11:39:50 +00:00
44 lines
1.0 KiB
Go
44 lines
1.0 KiB
Go
package terraform
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/provisionersdk/proto"
|
|
)
|
|
|
|
type mockLogger struct {
|
|
logs []*proto.Log
|
|
}
|
|
|
|
var _ logSink = &mockLogger{}
|
|
|
|
func (m *mockLogger) Log(l *proto.Log) {
|
|
m.logs = append(m.logs, l)
|
|
}
|
|
|
|
func TestLogWriter_Mainline(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
logr := &mockLogger{}
|
|
writer, doneLogging := logWriter(logr, proto.LogLevel_INFO)
|
|
|
|
_, err := writer.Write([]byte(`Sitting in an English garden
|
|
Waiting for the sun
|
|
If the sun don't come you get a tan
|
|
From standing in the English rain`))
|
|
require.NoError(t, err)
|
|
err = writer.Close()
|
|
require.NoError(t, err)
|
|
<-doneLogging
|
|
|
|
expected := []*proto.Log{
|
|
{Level: proto.LogLevel_INFO, Output: "Sitting in an English garden"},
|
|
{Level: proto.LogLevel_INFO, Output: "Waiting for the sun"},
|
|
{Level: proto.LogLevel_INFO, Output: "If the sun don't come you get a tan"},
|
|
{Level: proto.LogLevel_INFO, Output: "From standing in the English rain"},
|
|
}
|
|
require.Equal(t, expected, logr.logs)
|
|
}
|