mirror of
https://github.com/hackdiaz-dev/free-turn-proxy.git
synced 2026-10-12 09:41:53 +02:00
samosvalishe
22 lines
809 B
Go
22 lines
809 B
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/samosvalishe/free-turn-proxy/internal/clientsdb"
|
|
"github.com/samosvalishe/free-turn-proxy/internal/config"
|
|
)
|
|
|
|
func TestAdmitTCPAndBond(t *testing.T) {
|
|
for _, mode := range []config.ProxyMode{config.ProxyModeTCP, config.ProxyModeUDP} {
|
|
cfg := &config.Server{Proxy: config.ProxyOpts{Mode: mode}}
|
|
for _, tag := range []byte{clientsdb.ModeUnset, clientsdb.ModeUDP, clientsdb.ModeTCP, clientsdb.ModeTCPBond, 255} {
|
|
err := admitClient(cfg, nil, "test-client", tag)
|
|
want := (mode == config.ProxyModeTCP && (tag == clientsdb.ModeTCP || tag == clientsdb.ModeTCPBond)) || (mode == config.ProxyModeUDP && (tag == clientsdb.ModeUnset || tag == clientsdb.ModeUDP))
|
|
if (err == nil) != want {
|
|
t.Fatalf("mode=%s tag=%d err=%v", mode, tag, err)
|
|
}
|
|
}
|
|
}
|
|
}
|