docker-compose/server/proxy/streamwriter_test.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 {
	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 {
	s, _ := m.(*v1.LogsResponse)
	ls.logs = s.Logs
	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)
}