conn_mock.go

package net

import (
	"errors"
	"github.com/spacemeshos/go-spacemesh/crypto"
	"github.com/spacemeshos/go-spacemesh/p2p/p2pcrypto"
	"net"
	"sync/atomic"
	"time"
)

var defaultAddr net.Addr = &net.TCPAddr{
	IP:   IPv4LoopbackAddress,
	Port: 7513,
	Zone: "",
}

type ConnectionMock struct {
	id        string
	remotePub p2pcrypto.PublicKey
	session   NetworkSession
	source    ConnectionSource

	created time.Time

	Addr net.Addr

	sendDelayMs int
	sendRes     error
	sendCnt     int32

	closed bool
}

func NewConnectionMock(key p2pcrypto.PublicKey) *ConnectionMock {
	return &ConnectionMock{
		id:        crypto.UUIDString(),
		remotePub: key,
		closed:    false,
		created:   time.Now(),
	}
}

func (cm ConnectionMock) ID() string {
	return cm.id
}
func (cm ConnectionMock) Created() time.Time {
	return cm.created
}
func (cm ConnectionMock) SetCreated(time2 time.Time) {
	cm.created = time2
}

func (cm ConnectionMock) RemotePublicKey() p2pcrypto.PublicKey {
	return cm.remotePub
}

func (cm *ConnectionMock) SetRemotePublicKey(key p2pcrypto.PublicKey) {
	cm.remotePub = key
}

func (cm *ConnectionMock) RemoteAddr() net.Addr {
	return cm.Addr
}

func (cm *ConnectionMock) SetSession(session NetworkSession) {
	cm.session = session
}

func (cm ConnectionMock) Session() NetworkSession {
	return cm.session
}

func (cm ConnectionMock) IncomingChannel() chan []byte {
	return nil
}

func (cm *ConnectionMock) SetSendDelay(delayMs int) {
	cm.sendDelayMs = delayMs
}

func (cm *ConnectionMock) SetSendResult(err error) {
	cm.sendRes = err
}

func (cm ConnectionMock) SendCount() int32 {
	return atomic.LoadInt32(&cm.sendCnt)
}

func (cm *ConnectionMock) Send(m []byte) error {
	atomic.AddInt32(&cm.sendCnt, int32(1))
	time.Sleep(time.Duration(cm.sendDelayMs) * time.Millisecond)
	return cm.sendRes
}

func (cm ConnectionMock) Closed() bool {
	return cm.closed
}

func (cm *ConnectionMock) Close() error {
	if cm.closed == true {
		return errors.New("already closed")
	}
	cm.closed = true
	return nil
}

func (cm *ConnectionMock) beginEventProcessing() {

}

func (cm ConnectionMock) String() string {
	return cm.id
}