frp/test/func_test.go

64 lines
1.5 KiB
Go
Raw Normal View History

2016-05-17 19:13:37 +08:00
package test
import (
2017-03-09 02:03:47 +08:00
"bufio"
2016-05-17 19:13:37 +08:00
"fmt"
"io/ioutil"
"net/http"
"strings"
"testing"
"time"
2017-03-09 02:03:47 +08:00
"github.com/fatedier/frp/utils/net"
2016-05-17 19:13:37 +08:00
)
var (
ECHO_PORT int64 = 10711
HTTP_PORT int64 = 10710
ECHO_TEST_STR string = "Hello World\n"
HTTP_RES_STR string = "Hello World"
)
func TestEchoServer(t *testing.T) {
2017-03-09 02:03:47 +08:00
c, err := net.ConnectTcpServer(fmt.Sprintf("127.0.0.1:%d", ECHO_PORT))
2016-05-17 19:13:37 +08:00
if err != nil {
t.Fatalf("connect to echo server error: %v", err)
}
timer := time.Now().Add(time.Duration(5) * time.Second)
c.SetDeadline(timer)
2017-03-09 02:03:47 +08:00
c.Write([]byte(ECHO_TEST_STR))
c.Write('\n')
2016-05-17 19:13:37 +08:00
2017-03-09 02:03:47 +08:00
br := bufio.NewReader(c)
buf, err := br.ReadString('\n')
2016-05-17 19:13:37 +08:00
if err != nil {
t.Fatalf("read from echo server error: %v", err)
}
2017-03-09 02:03:47 +08:00
if ECHO_TEST_STR != buf {
t.Fatalf("content error, send [%s], get [%s]", strings.Trim(ECHO_TEST_STR, "\n"), strings.Trim(buf, "\n"))
2016-05-17 19:13:37 +08:00
}
}
func TestHttpServer(t *testing.T) {
client := &http.Client{}
req, _ := http.NewRequest("GET", fmt.Sprintf("http://127.0.0.1:%d", HTTP_PORT), nil)
res, err := client.Do(req)
if err != nil {
t.Fatalf("do http request error: %v", err)
}
if res.StatusCode == 200 {
body, err := ioutil.ReadAll(res.Body)
if err != nil {
t.Fatalf("read from http server error: %v", err)
}
bodystr := string(body)
if bodystr != HTTP_RES_STR {
t.Fatalf("content from http server error [%s], correct string is [%s]", bodystr, HTTP_RES_STR)
}
} else {
t.Fatalf("http code from http server error [%d]", res.StatusCode)
}
}