mirror of
https://github.com/datarhei/core.git
synced 2025-09-27 04:16:25 +08:00
eb1cc37456
Commits (Ingo Oppermann): - Add experimental SRT connection stats and logs - Hide /config/reload endpoint in reade-only mode - Add SRT server - Create v16 in go.mod - Fix data races, tests, lint, and update dependencies - Add trailing slash for routed directories (datarhei/restreamer#340) - Allow relative URLs in content in static routes Co-Authored-By: Ingo Oppermann <57445+ioppermann@users.noreply.github.com>
89 lines
1.6 KiB
Go
89 lines
1.6 KiB
Go
package net
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewPortrange(t *testing.T) {
|
|
_, err := NewPortrange(1000, 1999)
|
|
|
|
assert.Nil(t, err, "Valid port range not accepted: %s", err)
|
|
}
|
|
|
|
func TestInvalidPortrange(t *testing.T) {
|
|
_, err := NewPortrange(1999, 1000)
|
|
|
|
assert.NotNil(t, err, "Invalid port range accepted")
|
|
}
|
|
|
|
func TestGetPort(t *testing.T) {
|
|
portrange, _ := NewPortrange(1000, 1999)
|
|
|
|
port, err := portrange.Get()
|
|
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1000, port)
|
|
}
|
|
|
|
func TestGetPutPort(t *testing.T) {
|
|
portrange, _ := NewPortrange(1000, 1999)
|
|
|
|
port, err := portrange.Get()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1000, port)
|
|
|
|
port, err = portrange.Get()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1001, port)
|
|
|
|
portrange.Put(1000)
|
|
|
|
port, err = portrange.Get()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1000, port)
|
|
}
|
|
|
|
func TestPortUnavailable(t *testing.T) {
|
|
portrange, _ := NewPortrange(1000, 1999)
|
|
|
|
for i := 0; i < 1000; i++ {
|
|
port, _ := portrange.Get()
|
|
assert.Equal(t, 1000+i, port, "at index %d", i)
|
|
}
|
|
|
|
port, err := portrange.Get()
|
|
assert.NotNil(t, err)
|
|
assert.Less(t, port, 0)
|
|
}
|
|
|
|
func TestPutPort(t *testing.T) {
|
|
portrange, _ := NewPortrange(1000, 1999)
|
|
|
|
portrange.Put(999)
|
|
portrange.Put(1000)
|
|
|
|
portrange.Put(1999)
|
|
portrange.Put(2000)
|
|
}
|
|
|
|
func TestClampRange(t *testing.T) {
|
|
portrange, _ := NewPortrange(65000, 70000)
|
|
|
|
port, _ := portrange.Get()
|
|
|
|
assert.Equal(t, 65000, port)
|
|
|
|
portrange.Put(65000)
|
|
|
|
for i := 65000; i <= 65535; i++ {
|
|
port, _ := portrange.Get()
|
|
assert.Equal(t, i, port, "at index %d", i)
|
|
}
|
|
|
|
port, _ = portrange.Get()
|
|
|
|
assert.Less(t, port, 0)
|
|
}
|