1
//go:build !no_mocks
2
 
3
package mocks
4
 
5
import (
6
	"context"
7
	"net"
8
 
9
	"go.miragespace.co/specter/spec/protocol"
10
	"go.miragespace.co/specter/spec/transport"
11
 
12
	"github.com/stretchr/testify/mock"
13
)
14
 
15
type Transport struct {
16
	mock.Mock
17
}
18
 
19
var _ transport.Transport = (*Transport)(nil)
20
 
21
func (t *Transport) Identity() *protocol.Node {
22
	args := t.Called()
23
	v := args.Get(0)
24
	if v == nil {
25
		return nil
26
	}
27
	return v.(*protocol.Node)
28
}
29
 
30
func (t *Transport) DialStream(ctx context.Context, peer *protocol.Node, kind protocol.Stream_Type) (net.Conn, error) {
31
	args := t.Called(ctx, peer, kind)
32
	v := args.Get(0)
33
	e := args.Error(1)
34
	if v == nil {
35
		return nil, e
36
	}
37
	return v.(net.Conn), e
38
}
39
 
40
func (t *Transport) AcceptStream() <-chan *transport.StreamDelegate {
41
	args := t.Called()
42
	v := args.Get(0)
43
	return v.(chan *transport.StreamDelegate)
44
}
45
 
46
func (t *Transport) ListConnected() []transport.ConnectedPeer {
47
	args := t.Called()
48
	v := args.Get(0)
49
	return v.([]transport.ConnectedPeer)
50
}
51
 
52
func (t *Transport) SupportDatagram() bool {
53
	args := t.Called()
54
	v := args.Bool(0)
55
	return v
56
}
57
 
58
func (t *Transport) ReceiveDatagram() <-chan *transport.DatagramDelegate {
59
	args := t.Called()
60
	v := args.Get(0)
61
	return v.(chan *transport.DatagramDelegate)
62
}
63
 
64
func (t *Transport) SendDatagram(n *protocol.Node, b []byte) error {
65
	args := t.Called(n, b)
66
	e := args.Error(0)
67
	return e
68
}
69
 
70
func (t *Transport) TransportEstablished() <-chan *protocol.Node {
71
	args := t.Called()
72
	v := args.Get(0)
73
	return v.(chan *protocol.Node)
74
}
75
 
76
func (t *Transport) TransportDestroyed() <-chan *protocol.Node {
77
	args := t.Called()
78
	v := args.Get(0)
79
	return v.(chan *protocol.Node)
80
}