forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-16 00:43:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add test for function doPortMapping

Browse source
This commit is contained in:
Rafael Sampaio 2024-10-04 18:50:39 -03:00
parent 120be2f1ee
commit 4549996a9b
1 changed files with 33 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
cmd/shisui/main_test.go Normal file
View file

@ -0,0 +1,33 @@
package main
import (
"net"
"testing"
"github.com/ethereum/go-ethereum/crypto"
"github.com/ethereum/go-ethereum/p2p/enode"
"github.com/ethereum/go-ethereum/p2p/nat"
"github.com/stretchr/testify/assert"
)
func newLocalNodeForTesting() (*enode.LocalNode, *enode.DB) {
db, _ := enode.OpenDB("")
key, _ := crypto.GenerateKey()
return enode.NewLocalNode(db, key), db
}
func TestDoPortMapping(t *testing.T) {
nat := nat.ExtIP{33, 44, 55, 66}
localNode, _ := newLocalNodeForTesting()
listenerAddr := &net.UDPAddr{IP: net.IP{127, 0, 0, 1}, Port: 1234}
doPortMapping(nat, localNode, listenerAddr)
assert.Equal(t, localNode.Seq(), uint64(1))
assert.Equal(t, localNode.Node().IP(), net.IP{33, 44, 55, 66})
assert.Equal(t, localNode.Node().UDP(), 1234)
assert.Equal(t, localNode.Node().TCP(), 0)
_ = localNode.Node().UDP()
assert.Equal(t, localNode.Seq(), uint64(2))
}