package test

import (
	"testing"

	"encoding/hex"
	"strings"
)

func TestUpperHook(t *testing.T) {
	e := newEnv(t)
	defer e.Free()

	c := e.newInstance()

	c.Exec("add name T listen,addr=-:0 upper loop")

	conn := e.Dial("tcp", c.Get("tunnel.T.listen"))

	e.Write(conn, dummy)

	buf := make([]byte, len(dummy))
	e.ReadFull(conn, buf)

	if r := string(buf); r != strings.ToUpper(dummy) {
		t.Fatalf("wrong reply: send '%s', recv '%s'", dummy, r)
	}
}

func TestHexHook(t *testing.T) {
	e := newEnv(t)
	defer e.Free()

	c := e.newInstance()

	c.Exec("add name T listen,addr=-:0 hex dial,addr=@[addr]")

	listen := e.Listen("tcp", "127.0.0.1:0")
	c.Set("addr", listen.Addr())

	out := e.Dial("tcp", c.Get("tunnel.T.listen"))
	in := e.Accept(listen)

	e.Write(out, dummy)

	buf := make([]byte, 2*len(dummy))
	e.ReadFull(in, buf)

	if r := string(buf); r != hex.EncodeToString([]byte(dummy)) {
		t.Fatalf("wrong reply: send '%s', recv '%s'", dummy, r)
	}
}