micro/tunnel/tunnel_test.go

349 lines
6.1 KiB
Go
Raw Normal View History

2019-08-07 20:44:33 +03:00
package tunnel
import (
"os"
2019-08-07 20:56:21 +03:00
"sync"
2019-08-07 20:44:33 +03:00
"testing"
"time"
2019-08-07 20:44:33 +03:00
"github.com/micro/go-micro/v3/transport"
2019-08-07 20:44:33 +03:00
)
2019-12-08 16:45:24 +03:00
func testBrokenTunAccept(t *testing.T, tun Tunnel, wait chan bool, wg *sync.WaitGroup) {
defer wg.Done()
// listen on some virtual address
tl, err := tun.Listen("test-tunnel")
if err != nil {
t.Fatal(err)
}
// receiver ready; notify sender
wait <- true
// accept a connection
c, err := tl.Accept()
if err != nil {
t.Fatal(err)
}
// accept the message and close the tunnel
// we do this to simulate loss of network connection
m := new(transport.Message)
if err := c.Recv(m); err != nil {
t.Fatal(err)
}
// close all the links
for _, link := range tun.Links() {
link.Close()
}
// receiver ready; notify sender
wait <- true
// accept the message
m = new(transport.Message)
if err := c.Recv(m); err != nil {
t.Fatal(err)
}
// notify the sender we have received
wait <- true
}
func testBrokenTunSend(t *testing.T, tun Tunnel, wait chan bool, wg *sync.WaitGroup, reconnect time.Duration) {
defer wg.Done()
// wait for the listener to get ready
<-wait
// dial a new session
c, err := tun.Dial("test-tunnel")
if err != nil {
t.Fatal(err)
}
defer c.Close()
m := transport.Message{
Header: map[string]string{
"test": "send",
},
}
// send the message
if err := c.Send(&m); err != nil {
t.Fatal(err)
}
// wait for the listener to get ready
<-wait
// give it time to reconnect
time.Sleep(reconnect)
// send the message
if err := c.Send(&m); err != nil {
t.Fatal(err)
}
// wait for the listener to receive the message
// c.Send merely enqueues the message to the link send queue and returns
// in order to verify it was received we wait for the listener to tell us
<-wait
}
2019-08-07 20:44:33 +03:00
// testAccept will accept connections on the transport, create a new link and tunnel on top
func testAccept(t *testing.T, tun Tunnel, wait chan bool, wg *sync.WaitGroup) {
defer wg.Done()
2019-08-07 20:44:33 +03:00
// listen on some virtual address
tl, err := tun.Listen("test-tunnel")
if err != nil {
t.Fatal(err)
}
// receiver ready; notify sender
wait <- true
2019-08-07 20:44:33 +03:00
// accept a connection
c, err := tl.Accept()
if err != nil {
t.Fatal(err)
}
// get a message
for {
2019-08-14 16:38:17 +03:00
// accept the message
2019-08-07 20:44:33 +03:00
m := new(transport.Message)
if err := c.Recv(m); err != nil {
t.Fatal(err)
}
2019-08-14 16:38:17 +03:00
if v := m.Header["test"]; v != "send" {
t.Fatalf("Accept side expected test:send header. Received: %s", v)
}
// now respond
m.Header["test"] = "accept"
if err := c.Send(m); err != nil {
t.Fatal(err)
}
wait <- true
2019-08-07 20:44:33 +03:00
return
}
}
// testSend will create a new link to an address and then a tunnel on top
func testSend(t *testing.T, tun Tunnel, wait chan bool, wg *sync.WaitGroup) {
defer wg.Done()
// wait for the listener to get ready
<-wait
2019-08-07 20:44:33 +03:00
// dial a new session
c, err := tun.Dial("test-tunnel")
if err != nil {
t.Fatal(err)
}
2019-08-07 20:56:21 +03:00
defer c.Close()
2019-08-07 20:44:33 +03:00
m := transport.Message{
Header: map[string]string{
2019-08-14 16:38:17 +03:00
"test": "send",
2019-08-07 20:44:33 +03:00
},
}
2019-08-14 16:38:17 +03:00
// send the message
2019-08-07 20:44:33 +03:00
if err := c.Send(&m); err != nil {
t.Fatal(err)
}
2019-08-14 16:38:17 +03:00
// now wait for the response
mr := new(transport.Message)
if err := c.Recv(mr); err != nil {
t.Fatal(err)
}
<-wait
2019-08-14 16:38:17 +03:00
if v := mr.Header["test"]; v != "accept" {
t.Fatalf("Message not received from accepted side. Received: %s", v)
}
2019-08-07 20:44:33 +03:00
}
func TestTunnel(t *testing.T) {
2019-08-07 20:56:21 +03:00
// create a new tunnel client
tunA := NewTunnel(
2019-08-08 00:02:58 +03:00
Address("127.0.0.1:9096"),
Nodes("127.0.0.1:9097"),
2019-08-07 20:56:21 +03:00
)
// create a new tunnel server
tunB := NewTunnel(
2019-08-08 00:02:58 +03:00
Address("127.0.0.1:9097"),
2019-08-07 20:56:21 +03:00
)
// start tunB
err := tunB.Connect()
if err != nil {
t.Fatal(err)
}
defer tunB.Close()
// start tunA
err = tunA.Connect()
if err != nil {
t.Fatal(err)
}
defer tunA.Close()
wait := make(chan bool)
2019-08-07 20:56:21 +03:00
var wg sync.WaitGroup
wg.Add(1)
// start the listener
go testAccept(t, tunB, wait, &wg)
2019-08-07 20:56:21 +03:00
wg.Add(1)
// start the client
go testSend(t, tunA, wait, &wg)
2019-08-07 20:56:21 +03:00
// wait until done
wg.Wait()
2019-08-07 20:44:33 +03:00
}
func TestLoopbackTunnel(t *testing.T) {
// create a new tunnel
tun := NewTunnel(
Address("127.0.0.1:9096"),
Nodes("127.0.0.1:9096"),
)
// start tunnel
err := tun.Connect()
if err != nil {
t.Fatal(err)
}
defer tun.Close()
time.Sleep(500 * time.Millisecond)
wait := make(chan bool)
var wg sync.WaitGroup
wg.Add(1)
// start the listener
go testAccept(t, tun, wait, &wg)
wg.Add(1)
// start the client
go testSend(t, tun, wait, &wg)
// wait until done
wg.Wait()
}
2019-12-08 16:45:24 +03:00
func TestTunnelRTTRate(t *testing.T) {
// create a new tunnel client
tunA := NewTunnel(
Address("127.0.0.1:9096"),
Nodes("127.0.0.1:9097"),
)
2019-12-08 16:45:24 +03:00
// create a new tunnel server
tunB := NewTunnel(
Address("127.0.0.1:9097"),
)
2019-12-08 16:45:24 +03:00
// start tunB
err := tunB.Connect()
if err != nil {
t.Fatal(err)
}
2019-12-08 16:45:24 +03:00
defer tunB.Close()
2019-12-08 16:45:24 +03:00
// start tunA
err = tunA.Connect()
if err != nil {
t.Fatal(err)
}
2019-12-08 16:45:24 +03:00
defer tunA.Close()
2019-12-08 16:45:24 +03:00
wait := make(chan bool)
2019-12-08 16:45:24 +03:00
var wg sync.WaitGroup
2019-12-08 16:45:24 +03:00
wg.Add(1)
// start the listener
go testAccept(t, tunB, wait, &wg)
2019-12-08 16:45:24 +03:00
wg.Add(1)
// start the client
go testSend(t, tunA, wait, &wg)
2019-12-08 16:45:24 +03:00
// wait until done
wg.Wait()
if len(os.Getenv("IN_TRAVIS_CI")) == 0 {
// only needed for debug
for _, link := range tunA.Links() {
t.Logf("Link %s length %v rate %v", link.Id(), link.Length(), link.Rate())
}
for _, link := range tunB.Links() {
t.Logf("Link %s length %v rate %v", link.Id(), link.Length(), link.Rate())
}
}
}
2019-12-08 16:45:24 +03:00
func TestReconnectTunnel(t *testing.T) {
// we manually override the tunnel.ReconnectTime value here
// this is so that we make the reconnects faster than the default 5s
ReconnectTime = 200 * time.Millisecond
2019-10-22 21:38:29 +03:00
// create a new tunnel client
tunA := NewTunnel(
2019-12-08 16:45:24 +03:00
Address("127.0.0.1:9098"),
Nodes("127.0.0.1:9099"),
2019-10-22 21:38:29 +03:00
)
// create a new tunnel server
tunB := NewTunnel(
2019-12-08 16:45:24 +03:00
Address("127.0.0.1:9099"),
2019-10-22 21:38:29 +03:00
)
2019-12-08 16:45:24 +03:00
// start tunnel
2019-10-22 21:38:29 +03:00
err := tunB.Connect()
if err != nil {
t.Fatal(err)
}
defer tunB.Close()
2019-12-08 16:45:24 +03:00
// start tunnel
2019-10-22 21:38:29 +03:00
err = tunA.Connect()
if err != nil {
t.Fatal(err)
}
defer tunA.Close()
wait := make(chan bool)
var wg sync.WaitGroup
wg.Add(1)
2019-12-08 16:45:24 +03:00
// start tunnel listener
go testBrokenTunAccept(t, tunB, wait, &wg)
2019-10-22 21:38:29 +03:00
wg.Add(1)
2019-12-08 16:45:24 +03:00
// start tunnel sender
go testBrokenTunSend(t, tunA, wait, &wg, ReconnectTime*5)
2019-10-22 21:38:29 +03:00
// wait until done
wg.Wait()
}