evio_test.go

package evio

import (
	"bufio"
	"fmt"
	"io"
	"math/rand"
	"net"
	"os"
	"strings"
	"sync"
	"sync/atomic"
	"testing"
	"time"
)

func TestServe(t *testing.T) {
	// start a server
	// connect 10 clients
	// each client will pipe random data for 1-3 seconds.
	// the writes to the server will be random sizes. 0KB - 1MB.
	// the server will echo back the data.
	// waits for graceful connection closing.
	var wg sync.WaitGroup
	wg.Add(1)
	go func() {
		defer wg.Done()
		testServe("tcp", ":9990", false, 10)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testServe("tcp", ":9991", true, 10)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testServe("tcp-net", ":9992", false, 10)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testServe("tcp-net", ":9993", true, 10)
	}()
	wg.Wait()
}

func testServe(network, addr string, unix bool, nclients int) {
	var started bool
	var connected int
	var disconnected int

	var events Events
	events.Serving = func(wake func(id int) bool, addrs []net.Addr) (action Action) {
		return
	}
	events.Opened = func(id int, addr Addr) (out []byte, opts Options, action Action) {
		connected++
		out = []byte("sweetness\r\n")
		opts.TCPKeepAlive = time.Minute * 5
		return
	}
	events.Closed = func(id int, err error) (action Action) {
		disconnected++
		if connected == disconnected && disconnected == nclients {
			action = Shutdown
		}
		return
	}
	events.Data = func(id int, in []byte) (out []byte, action Action) {
		out = in
		return
	}
	events.Tick = func() (delay time.Duration, action Action) {
		if !started {
			for i := 0; i < nclients; i++ {
				go startClient(network, addr)
			}
			started = true
		}
		delay = time.Second / 5
		return
	}
	var err error
	if unix {
		socket := strings.Replace(addr, ":", "socket", 1)
		os.RemoveAll(socket)
		defer os.RemoveAll(socket)
		err = Serve(events, network+"://"+addr, "unix://"+socket)
	} else {
		err = Serve(events, network+"://"+addr)
	}
	if err != nil {
		panic(err)
	}
}

func startClient(network, addr string) {
	network = strings.Replace(network, "-net", "", -1)
	rand.Seed(time.Now().UnixNano())
	c, err := net.Dial(network, addr)
	if err != nil {
		panic(err)
	}
	defer c.Close()
	rd := bufio.NewReader(c)
	msg, err := rd.ReadBytes('\n')
	if err != nil {
		panic(err)
	}
	if string(msg) != "sweetness\r\n" {
		panic("bad header")
	}
	duration := time.Duration((rand.Float64()*2 + 1) * float64(time.Second))
	start := time.Now()
	for time.Since(start) < duration {
		sz := rand.Int() % (1024 * 1024)
		data := make([]byte, sz)
		if _, err := rand.Read(data); err != nil {
			panic(err)
		}
		if _, err := c.Write(data); err != nil {
			panic(err)
		}
		data2 := make([]byte, sz)
		if _, err := io.ReadFull(rd, data2); err != nil {
			panic(err)
		}
		if string(data) != string(data2) {
			fmt.Printf("mismatch: %d bytes\n", len(data))
			//panic("mismatch")
		}
	}
}

func TestWake(t *testing.T) {
	var wg sync.WaitGroup
	wg.Add(1)
	go func() {
		defer wg.Done()
		testWake("tcp", ":9991", false)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testWake("tcp", ":9992", true)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testWake("unix", "socket1", false)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testWake("unix", "socket2", true)
	}()
	wg.Wait()
}
func testWake(network, addr string, stdlib bool) {
	var events Events
	var wake func(id int) bool
	events.Serving = func(wakefn func(id int) bool, addrs []net.Addr) (action Action) {
		wake = wakefn
		go func() {
			conn, err := net.Dial(network, addr)
			must(err)
			defer conn.Close()
			rd := bufio.NewReader(conn)
			for i := 0; i < 1000; i++ {
				line := []byte(fmt.Sprintf("msg%d\r\n", i))
				conn.Write(line)
				data, err := rd.ReadBytes('\n')
				must(err)
				if string(data) != string(line) {
					panic("msg mismatch")
				}
			}
		}()
		return
	}

	var cid int
	var cout []byte
	var cin []byte
	var cclosed bool
	var cond = sync.NewCond(&sync.Mutex{})
	events.Opened = func(id int, addr Addr) (out []byte, opts Options, action Action) {
		cid = id
		return
	}
	events.Closed = func(id int, err error) (action Action) {
		action = Shutdown
		cond.L.Lock()
		cclosed = true
		cond.Broadcast()
		cond.L.Unlock()
		return
	}
	go func() {
		cond.L.Lock()
		for !cclosed {
			if len(cin) > 0 {
				cout = append(cout, cin...)
				cin = nil
			}
			if len(cout) > 0 {
				wake(cid)
			}
			cond.Wait()
		}
		cond.L.Unlock()
	}()
	events.Data = func(id int, in []byte) (out []byte, action Action) {
		if in == nil {
			cond.L.Lock()
			out = cout
			cout = nil
			cond.L.Unlock()
		} else {
			cond.L.Lock()
			cin = append(cin, in...)
			cond.Broadcast()
			cond.L.Unlock()
		}
		return
	}
	if stdlib {
		must(Serve(events, network+"-net://"+addr))
	} else {
		must(Serve(events, network+"://"+addr))
	}
}
func must(err error) {
	if err != nil {
		panic(err)
	}
}
func TestTick(t *testing.T) {
	var wg sync.WaitGroup
	wg.Add(1)
	go func() {
		defer wg.Done()
		testTick("tcp", ":9991", false)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testTick("tcp", ":9992", true)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testTick("unix", "socket1", false)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testTick("unix", "socket2", true)
	}()
	wg.Wait()
}
func testTick(network, addr string, stdlib bool) {
	var events Events
	var count int
	start := time.Now()
	events.Tick = func() (delay time.Duration, action Action) {
		if count == 25 {
			action = Shutdown
			return
		}
		count++
		delay = time.Millisecond * 10
		return
	}
	if stdlib {
		must(Serve(events, network+"-net://"+addr))
	} else {
		must(Serve(events, network+"://"+addr))
	}
	dur := time.Since(start)
	if dur < 250&time.Millisecond || dur > time.Second {
		panic("bad ticker timing")
	}
}

func TestShutdown(t *testing.T) {
	var wg sync.WaitGroup
	wg.Add(1)
	go func() {
		defer wg.Done()
		testShutdown("tcp", ":9991", false)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testShutdown("tcp", ":9992", true)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testShutdown("unix", "socket1", false)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testShutdown("unix", "socket2", true)
	}()
	wg.Wait()
}
func testShutdown(network, addr string, stdlib bool) {
	var events Events
	var count int
	var clients int64
	var N = 10
	events.Opened = func(id int, addr Addr) (out []byte, opts Options, action Action) {
		atomic.AddInt64(&clients, 1)
		return
	}
	events.Closed = func(id int, err error) (action Action) {
		atomic.AddInt64(&clients, -1)
		return
	}
	events.Tick = func() (delay time.Duration, action Action) {
		if count == 0 {
			// start clients
			for i := 0; i < N; i++ {
				go func() {
					conn, err := net.Dial(network, addr)
					must(err)
					defer conn.Close()
					_, err = conn.Read([]byte{0})
					if err == nil {
						panic("expected error")
					}
				}()
			}
		} else {
			if int(atomic.LoadInt64(&clients)) == N {
				action = Shutdown
			}
		}
		count++
		delay = time.Second / 5
		return
	}
	if stdlib {
		must(Serve(events, network+"-net://"+addr))
	} else {
		must(Serve(events, network+"://"+addr))
	}
	if clients != 0 {
		panic("did not call close on all clients")
	}
}

func TestDetach(t *testing.T) {
	var wg sync.WaitGroup
	wg.Add(1)
	go func() {
		defer wg.Done()
		testDetach("tcp", ":9991", false)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testDetach("tcp", ":9992", true)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testDetach("unix", "socket1", false)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testDetach("unix", "socket2", true)
	}()
	wg.Wait()
}

func testDetach(network, addr string, stdlib bool) {
	// we will write a bunch of data with the text "--detached--" in the
	// middle followed by a bunch of data.
	rand.Seed(time.Now().UnixNano())
	rdat := make([]byte, 10*1024*1024)
	if _, err := rand.Read(rdat); err != nil {
		panic("random error: " + err.Error())
	}
	expected := []byte(string(rdat) + "--detached--" + string(rdat))
	var cin []byte
	var events Events
	events.Data = func(id int, in []byte) (out []byte, action Action) {
		cin = append(cin, in...)
		if len(cin) == len(expected) {
			if string(cin) != string(expected) {
				panic("mismatch client -> server")
			}
			return cin, Detach
		}
		return
	}

	//expected := "detached\r\n"
	var done int64
	events.Detached = func(id int, conn io.ReadWriteCloser) (action Action) {
		go func() {
			defer conn.Close()
			// detached connection
			n, err := conn.Write([]byte(expected))
			must(err)
			if n != len(expected) {
				panic("not enough data written")
			}
		}()
		return
	}
	events.Serving = func(_ func(id int) bool, addrs []net.Addr) (action Action) {
		go func() {
			// client connection
			conn, err := net.Dial(network, addr)
			must(err)
			defer conn.Close()
			_, err = conn.Write(expected)
			must(err)
			// read from the attached response
			packet := make([]byte, len(expected))
			time.Sleep(time.Second / 3)
			_, err = io.ReadFull(conn, packet)
			must(err)
			if string(packet) != string(expected) {
				panic("mismatch server -> client 1")
			}
			// read from the detached response
			time.Sleep(time.Second / 3)
			_, err = io.ReadFull(conn, packet)
			must(err)
			if string(packet) != string(expected) {
				panic("mismatch server -> client 2")
			}
			time.Sleep(time.Second / 3)
			_, err = conn.Read([]byte{0})

			if err == nil {
				panic("expected nil, got '" + err.Error() + "'")
			}
			atomic.StoreInt64(&done, 1)
		}()
		return
	}
	events.Tick = func() (delay time.Duration, action Action) {
		delay = time.Second / 5
		if atomic.LoadInt64(&done) == 1 {
			action = Shutdown
		}
		return
	}
	if stdlib {
		must(Serve(events, network+"-net://"+addr))
	} else {
		must(Serve(events, network+"://"+addr))
	}
}

func TestBadAddresses(t *testing.T) {
	var events Events
	events.Serving = func(wake func(id int) bool, addrs []net.Addr) (action Action) {
		return Shutdown
	}
	if err := Serve(events, "tulip://howdy"); err == nil {
		t.Fatalf("expected error")
	}
	if err := Serve(events, "howdy"); err == nil {
		t.Fatalf("expected error")
	}
	if err := Serve(events, "unix://"); err == nil {
		t.Fatalf("expected error")
	}
	if err := Serve(events, "tcp://"); err != nil {
		t.Fatalf("expected nil, got '%v'", err)
	}
}

func TestInputStream(t *testing.T) {
	var s InputStream
	in := []byte("HELLO")
	data := s.Begin(in)
	if string(data) != string(in) {
		t.Fatalf("expected '%v', got '%v'", in, data)
	}
	s.End(in[3:])
	data = s.Begin([]byte("WLY"))
	if string(data) != "LOWLY" {
		t.Fatalf("expected '%v', got '%v'", "LOWLY", data)
	}
	s.End(nil)
	data = s.Begin([]byte("PLAYER"))
	if string(data) != "PLAYER" {
		t.Fatalf("expected '%v', got '%v'", "PLAYER", data)
	}
}

func TestPrePostwrite(t *testing.T) {
	var wg sync.WaitGroup
	wg.Add(1)
	go func() {
		defer wg.Done()
		testPrePostwrite("tcp", ":9991", false)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testPrePostwrite("tcp", ":9992", true)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testPrePostwrite("unix", "socket1", false)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testPrePostwrite("unix", "socket2", true)
	}()
	wg.Wait()
}

func testPrePostwrite(network, addr string, stdlib bool) {
	var events Events
	var wake func(id int) bool
	var packets int
	var tout []byte
	events.Opened = func(id int, addr Addr) (out []byte, opts Options, action Action) {
		packets++
		out = []byte(fmt.Sprintf("hello %d\r\n", packets))
		tout = append(tout, out...)
		wake(id)
		return
	}
	events.Data = func(id int, in []byte) (out []byte, action Action) {
		packets++
		out = []byte(fmt.Sprintf("hello %d\r\n", packets))
		tout = append(tout, out...)
		wake(id)
		return
	}
	events.Prewrite = func(id int, amount int) (action Action) {
		if amount != len(tout) {
			panic("invalid prewrite amount")
		}
		return
	}
	events.Postwrite = func(id int, amount, remaining int) (action Action) {
		tout = tout[amount:]
		if remaining != len(tout) {
			panic("invalid postwrite amount")
		}
		return
	}
	events.Closed = func(id int, err error) (action Action) {
		action = Shutdown
		return
	}
	events.Serving = func(wakefn func(id int) bool, addrs []net.Addr) (action Action) {
		wake = wakefn
		go func() {
			conn, err := net.Dial(network, addr)
			must(err)
			defer conn.Close()
			rd := bufio.NewReader(conn)
			for i := 0; i < 1000; i++ {
				line, err := rd.ReadBytes('\n')
				must(err)
				ex := fmt.Sprintf("hello %d\r\n", i+1)
				if string(line) != ex {
					panic(fmt.Sprintf("expected '%v', got '%v'", ex, line))
				}
			}
		}()
		return
	}
	if stdlib {
		must(Serve(events, network+"-net://"+addr))
	} else {
		must(Serve(events, network+"://"+addr))
	}
}

func TestTranslate(t *testing.T) {
	var wg sync.WaitGroup
	wg.Add(1)
	go func() {
		defer wg.Done()
		testTranslate("tcp", ":9991", "passthrough", false)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testTranslate("tcp", ":9992", "passthrough", true)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testTranslate("unix", "socket1", "passthrough", false)
	}()
	wg.Add(1)
	go func() {
		defer wg.Done()
		testTranslate("unix", "socket2", "passthrough", true)
	}()
	wg.Wait()
}

func testTranslate(network, addr string, kind string, stdlib bool) {
	var events Events
	events.Data = func(id int, in []byte) (out []byte, action Action) {
		out = in
		return
	}
	events.Closed = func(id int, err error) (action Action) {
		action = Shutdown
		return
	}
	events.Opened = func(id int, addr Addr) (out []byte, opts Options, action Action) {
		out = []byte("sweetness\r\n")
		return
	}
	events.Serving = func(wakefn func(id int) bool, addrs []net.Addr) (action Action) {
		go func() {
			conn, err := net.Dial(network, addr)
			must(err)
			defer conn.Close()
			line := "sweetness\r\n"
			packet := make([]byte, len(line))
			n, err := io.ReadFull(conn, packet)
			must(err)
			if n != len(line) {
				panic("invalid amount")
			}
			if string(packet) != string(line) {
				panic(fmt.Sprintf("expected '%v', got '%v'\n", line, packet))
			}

			for i := 0; i < 100; i++ {
				line := fmt.Sprintf("hello %d\r\n", i)
				n, err := conn.Write([]byte(line))
				must(err)
				if n != len(line) {
					panic("invalid amount")
				}
				packet := make([]byte, len(line))
				n, err = io.ReadFull(conn, packet)
				must(err)
				if n != len(line) {
					panic("invalid amount")
				}
				if string(packet) != string(line) {
					panic(fmt.Sprintf("expected '%v', got '%v'\n", line, packet))
				}
			}
		}()
		return
	}

	tevents := Translate(events,
		func(id int, addr Addr) bool {
			return true
		},
		func(rw io.ReadWriter) io.ReadWriter {
			switch kind {
			case "passthrough":
				return rw
			}
			panic("invalid kind")
		},
	)

	if stdlib {
		must(Serve(tevents, network+"-net://"+addr))
	} else {
		must(Serve(tevents, network+"://"+addr))
	}

	// test with no shoulds
	tevents = Translate(events,
		func(id int, addr Addr) bool {
			return false
		},
		func(rw io.ReadWriter) io.ReadWriter {
			return rw
		},
	)
	if stdlib {
		must(Serve(tevents, network+"-net://"+addr))
	} else {
		must(Serve(tevents, network+"://"+addr))
	}
}

func TestVariousAddr(t *testing.T) {
	var events Events
	var kind string
	events.Serving = func(wake func(id int) bool, addrs []net.Addr) (action Action) {
		addr := addrs[0].(*net.TCPAddr)
		if (kind == "tcp4" && len(addr.IP) != 4) || (kind == "tcp6" && len(addr.IP) != 16) {
			println(len(addr.IP))
			panic("invalid ip")
		}
		go func(kind string) {
			conn, err := net.Dial(kind, ":9991")
			must(err)
			defer conn.Close()
		}(kind)
		return
	}
	events.Closed = func(id int, err error) (action Action) {
		return Shutdown
	}
	kind = "tcp4"
	must(Serve(events, "tcp4://:9991"))
	kind = "tcp6"
	must(Serve(events, "tcp6://:9991"))
}