micro/tunnel/tunnel_test.go

133 lines
2.1 KiB
Go
Raw Normal View History

2019-08-07 20:44:33 +03:00
package tunnel
import (
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/transport"
)
// testAccept will accept connections on the transport, create a new link and tunnel on top
2019-08-07 20:56:21 +03:00
func testAccept(t *testing.T, tun Tunnel, wg *sync.WaitGroup) {
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)
}
// accept a connection
c, err := tl.Accept()
if err != nil {
t.Fatal(err)
}
// get a message
for {
m := new(transport.Message)
if err := c.Recv(m); err != nil {
t.Fatal(err)
}
2019-08-07 20:56:21 +03:00
wg.Done()
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) {
// 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{
"test": "header",
},
}
if err := c.Send(&m); err != nil {
t.Fatal(err)
}
}
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()
time.Sleep(time.Millisecond * 50)
2019-08-07 20:56:21 +03:00
// start tunA
err = tunA.Connect()
if err != nil {
t.Fatal(err)
}
defer tunA.Close()
time.Sleep(time.Millisecond * 50)
2019-08-07 20:56:21 +03:00
var wg sync.WaitGroup
// start accepting connections
// on tunnel A
2019-08-07 20:56:21 +03:00
wg.Add(1)
go testAccept(t, tunA, &wg)
2019-08-07 20:56:21 +03:00
time.Sleep(time.Millisecond * 50)
// dial and send via B
testSend(t, tunB)
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 client
tun := NewTunnel(
Address("127.0.0.1:9096"),
Nodes("127.0.0.1:9096"),
)
// start tunB
err := tun.Connect()
if err != nil {
t.Fatal(err)
}
defer tun.Close()
time.Sleep(time.Millisecond * 50)
var wg sync.WaitGroup
// start accepting connections
// on tunnel A
wg.Add(1)
go testAccept(t, tun, &wg)
time.Sleep(time.Millisecond * 50)
// dial and send via B
testSend(t, tun)
// wait until done
wg.Wait()
}