snapshot of https://github.com/pion/webrtc.git (main, 2026-09-15; blobs > 131072B omitted)

vnet_test.go

@@ -0,0 +1,75 @@
+// SPDX-FileCopyrightText: 2026 The Pion community <https://pion.ly>
+// SPDX-License-Identifier: MIT
+
+//go:build !js
+
+package webrtc
+
+import (
+	"testing"
+	"time"
+
+	"github.com/pion/interceptor"
+	"github.com/pion/logging"
+	"github.com/pion/transport/v4/vnet"
+	"github.com/stretchr/testify/assert"
+)
+
+func createVNetPair(t *testing.T, interceptorRegistry *interceptor.Registry) (
+	*PeerConnection,
+	*PeerConnection,
+	*vnet.Router,
+) {
+	t.Helper()
+	// Create a root router
+	wan, err := vnet.NewRouter(&vnet.RouterConfig{
+		CIDR:          "1.2.3.0/24",
+		LoggerFactory: logging.NewDefaultLoggerFactory(),
+	})
+	assert.NoError(t, err)
+
+	// Create a network interface for offerer
+	offerVNet, err := vnet.NewNet(&vnet.NetConfig{
+		StaticIPs: []string{"1.2.3.4"},
+	})
+	assert.NoError(t, err)
+
+	// Add the network interface to the router
+	assert.NoError(t, wan.AddNet(offerVNet))
+
+	offerSettingEngine := SettingEngine{}
+	offerSettingEngine.SetNet(offerVNet)
+	offerSettingEngine.SetICETimeouts(time.Second, time.Second, time.Millisecond*200)
+
+	// Create a network interface for answerer
+	answerVNet, err := vnet.NewNet(&vnet.NetConfig{
+		StaticIPs: []string{"1.2.3.5"},
+	})
+	assert.NoError(t, err)
+
+	// Add the network interface to the router
+	assert.NoError(t, wan.AddNet(answerVNet))
+
+	answerSettingEngine := SettingEngine{}
+	answerSettingEngine.SetNet(answerVNet)
+	answerSettingEngine.SetICETimeouts(time.Second, time.Second, time.Millisecond*200)
+
+	// Start the virtual network by calling Start() on the root router
+	assert.NoError(t, wan.Start())
+
+	offerOptions := []func(*API){WithSettingEngine(offerSettingEngine)}
+	if interceptorRegistry != nil {
+		offerOptions = append(offerOptions, WithInterceptorRegistry(interceptorRegistry))
+	}
+	offerPeerConnection, err := NewAPI(offerOptions...).NewPeerConnection(Configuration{})
+	assert.NoError(t, err)
+
+	answerOptions := []func(*API){WithSettingEngine(answerSettingEngine)}
+	if interceptorRegistry != nil {
+		answerOptions = append(answerOptions, WithInterceptorRegistry(interceptorRegistry))
+	}
+	answerPeerConnection, err := NewAPI(answerOptions...).NewPeerConnection(Configuration{})
+	assert.NoError(t, err)
+
+	return offerPeerConnection, answerPeerConnection, wan
+}