Skip to content

Commit 53ecdad

Browse files
committed
Move MockPacketConn to fakenet package
1 parent 8c5b099 commit 53ecdad

File tree

2 files changed

+28
-14
lines changed

2 files changed

+28
-14
lines changed

agent_test.go

+5-14
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ import (
1616
"testing"
1717
"time"
1818

19+
"github.com/pion/ice/v2/internal/fakenet"
1920
"github.com/pion/logging"
2021
"github.com/pion/stun"
2122
"github.com/pion/transport/v2/test"
@@ -24,16 +25,6 @@ import (
2425
"github.com/stretchr/testify/require"
2526
)
2627

27-
type mockPacketConn struct{}
28-
29-
func (m *mockPacketConn) ReadFrom([]byte) (n int, addr net.Addr, err error) { return 0, nil, nil }
30-
func (m *mockPacketConn) WriteTo([]byte, net.Addr) (n int, err error) { return 0, nil }
31-
func (m *mockPacketConn) Close() error { return nil }
32-
func (m *mockPacketConn) LocalAddr() net.Addr { return nil }
33-
func (m *mockPacketConn) SetDeadline(time.Time) error { return nil }
34-
func (m *mockPacketConn) SetReadDeadline(time.Time) error { return nil }
35-
func (m *mockPacketConn) SetWriteDeadline(time.Time) error { return nil }
36-
3728
func TestOnSelectedCandidatePairChange(t *testing.T) {
3829
report := test.CheckRoutines(t)
3930
defer report()
@@ -133,7 +124,7 @@ func TestHandlePeerReflexive(t *testing.T) {
133124
Component: 1,
134125
}
135126
local, err := NewCandidateHost(&hostConfig)
136-
local.conn = &mockPacketConn{}
127+
local.conn = &fakenet.MockPacketConn{}
137128
if err != nil {
138129
t.Fatalf("failed to create a new candidate: %v", err)
139130
}
@@ -224,7 +215,7 @@ func TestHandlePeerReflexive(t *testing.T) {
224215
Component: 1,
225216
}
226217
local, err := NewCandidateHost(&hostConfig)
227-
local.conn = &mockPacketConn{}
218+
local.conn = &fakenet.MockPacketConn{}
228219
if err != nil {
229220
t.Fatalf("failed to create a new candidate: %v", err)
230221
}
@@ -445,7 +436,7 @@ func TestInboundValidity(t *testing.T) {
445436
Component: 1,
446437
}
447438
local, err := NewCandidateHost(&hostConfig)
448-
local.conn = &mockPacketConn{}
439+
local.conn = &fakenet.MockPacketConn{}
449440
if err != nil {
450441
t.Fatalf("failed to create a new candidate: %v", err)
451442
}
@@ -547,7 +538,7 @@ func TestInboundValidity(t *testing.T) {
547538
Component: 1,
548539
}
549540
local, err := NewCandidateHost(&hostConfig)
550-
local.conn = &mockPacketConn{}
541+
local.conn = &fakenet.MockPacketConn{}
551542
if err != nil {
552543
t.Fatalf("failed to create a new candidate: %v", err)
553544
}

internal/fakenet/mock_conn.go

+23
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
2+
// SPDX-License-Identifier: MIT
3+
4+
//go:build !js
5+
// +build !js
6+
7+
package fakenet
8+
9+
import (
10+
"net"
11+
"time"
12+
)
13+
14+
// MockPacketConn for tests
15+
type MockPacketConn struct{}
16+
17+
func (m *MockPacketConn) ReadFrom([]byte) (n int, addr net.Addr, err error) { return 0, nil, nil } //nolint:revive
18+
func (m *MockPacketConn) WriteTo([]byte, net.Addr) (n int, err error) { return 0, nil } //nolint:revive
19+
func (m *MockPacketConn) Close() error { return nil } //nolint:revive
20+
func (m *MockPacketConn) LocalAddr() net.Addr { return nil } //nolint:revive
21+
func (m *MockPacketConn) SetDeadline(time.Time) error { return nil } //nolint:revive
22+
func (m *MockPacketConn) SetReadDeadline(time.Time) error { return nil } //nolint:revive
23+
func (m *MockPacketConn) SetWriteDeadline(time.Time) error { return nil } //nolint:revive

0 commit comments

Comments
 (0)