mirror of https://github.com/docker/compose.git
56 lines
928 B
Go
56 lines
928 B
Go
package proxy
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/grpc/metadata"
|
|
|
|
v1 "github.com/docker/api/protos/containers/v1"
|
|
)
|
|
|
|
type logServer struct {
|
|
logs []byte
|
|
}
|
|
|
|
func (ls *logServer) Send(response *v1.LogsResponse) error {
|
|
ls.logs = response.Logs
|
|
return nil
|
|
}
|
|
|
|
func (ls *logServer) SetHeader(metadata.MD) error {
|
|
return nil
|
|
}
|
|
|
|
func (ls *logServer) SendHeader(metadata.MD) error {
|
|
return nil
|
|
}
|
|
|
|
func (ls *logServer) SetTrailer(metadata.MD) {
|
|
}
|
|
|
|
func (ls *logServer) Context() context.Context {
|
|
return nil
|
|
}
|
|
|
|
func (ls *logServer) SendMsg(m interface{}) error {
|
|
return nil
|
|
}
|
|
|
|
func (ls *logServer) RecvMsg(m interface{}) error {
|
|
return nil
|
|
}
|
|
|
|
func TestStreamWriter(t *testing.T) {
|
|
ls := &logServer{}
|
|
sw := newStreamWriter(ls)
|
|
in := []byte{104, 101, 108, 108, 111}
|
|
|
|
l, err := sw.Write(in)
|
|
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, len(in), l)
|
|
assert.Equal(t, in, ls.logs)
|
|
}
|