syncthing/protocol/common_test.go

package protocol

import "io"

type TestModel struct {
	data   []byte
	name   string
	offset uint64
	size   uint32
	hash   []byte
	closed bool
}

func (t *TestModel) Index(nodeID string, files []FileInfo) {
}

func (t *TestModel) Request(nodeID, name string, offset uint64, size uint32, hash []byte) ([]byte, error) {
	t.name = name
	t.offset = offset
	t.size = size
	t.hash = hash
	return t.data, nil
}

func (t *TestModel) Close(nodeID string) {
	t.closed = true
}

type ErrPipe struct {
	io.PipeWriter
	written int
	max     int
	err     error
	closed  bool
}

func (e *ErrPipe) Write(data []byte) (int, error) {
	if e.closed {
		return 0, e.err
	}
	if e.written+len(data) > e.max {
		n, _ := e.PipeWriter.Write(data[:e.max-e.written])
		e.PipeWriter.CloseWithError(e.err)
		e.closed = true
		return n, e.err
	} else {
		return e.PipeWriter.Write(data)
	}
}