go http/websocket接口测试代码
Posted dy blog
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了go http/websocket接口测试代码相关的知识，希望对你有一定的参考价值。
//wsRouter_test.go
package Network import ( "encoding/json" "io/ioutil" "logs" "net/http" "net/url" "strings" "testing" "time" ) type testData struct { id string data interface{} code int msg string result interface{} expCode int } var websocketToken string var socket string = "127.0.0.1:8001" func TestWsRouter(test *testing.T) { publicFileDir = "../public" privateFileDir = "../private" go LaunchRouter() for len(dbQueue) <= 0 { time.Sleep(time.Millisecond * 1000) } dyTest([]testData{{data: ` { "Method":"/createGroup", "Args":"" }`, expCode: errAssertType}}) testAddGroupMemberList() testAddGroupMember() testCreateGroup() testDeleteGroupMember() /*---------------------http-----------------------*/ testWsLogin() testAlterIcon() } func dyTest(test []testData) { for idx, _ := range test { val := &test[idx] var req ClientRequest var res *ServerResponse err := json.Unmarshal([]byte(val.data.(string)), &req) if err != nil { logs.Print(err.Error()) panic(err) } res = dyRouter(req.Method, val.id, req.Args) val.code = res.Code val.msg = res.Message val.result = res.Result if val.code != val.expCode { logs.Print(val) panic(val) } } } func httpTest(test []testData) { var err error for idx, _ := range test { val := &test[idx] var req *http.Request switch tmp := val.data.(type) { case *http.Request: req = tmp case url.Values: req, err = http.NewRequest("POST", val.msg, strings.NewReader(tmp.Encode())) if err != nil { panic(err) } default: logs.Print(val) panic(val) } resp, err := http.DefaultClient.Do(req) if err != nil { panic(err) } dealHttpResp(resp, val) } } func dealHttpResp(resp *http.Response, test *testData) { defer resp.Body.Close() contentType := resp.Header["Content-Type"][0] if strings.Index(contentType, "json") != -1 { body, err := ioutil.ReadAll(resp.Body) if err != nil { panic(err) } var res HTTPResult err = json.Unmarshal(body, &res) if err != nil { panic(err) } test.code = res.Code test.msg = res.Message test.result = res.Result if test.code != test.expCode { logs.Print(test) panic(test) } } }
//alterIcon_test.go
package Network
import (
"bytes"
"io"
"mime/multipart"
"net/http"
"net/url"
"os"
"path/filepath"
)
func testAlterIcon() {
test := []testData{
{
data: map[string]string{
"token": websocketToken,
"objectID": "100000001",
},
msg: "icon",
expCode: Success,
},
{
data: map[string]string{
"token": websocketToken,
"objectID": "100000000",
},
msg: "icon",
expCode: errNoPermission,
},
{
data: map[string]string{
"token": websocketToken,
"objectID": "100000005",
},
msg: "icon",
expCode: errNoPermission,
},
{
data: map[string]string{
"token": websocketToken,
},
msg: "icon",
expCode: Success,
},
{
data: map[string]string{
"objectID": "100000005",
},
msg: "icon",
expCode: errInvalidToken,
},
{
data: map[string]string{
"token": websocketToken,
},
msg: "file",
expCode: errParseFileFailed,
},
}
for idx, _ := range test {
data, ok := test[idx].data.(map[string]string)
if !ok {
panic(test[idx].data)
}
req, err := newfileUploadRequest("http://"+socket+"/alterIcon", data, test[idx].msg, "../public/icon/default.ico")
if err != nil {
panic(err)
}
test[idx].data = req
}
test = append(test, testData{
data: url.Values{
"token": {websocketToken},
"objectID": {"100000001"},
"icon": {""},
},
msg: "http://" + socket + "/alterIcon",
expCode: errParseForm,
})
httpTest(test)
}
func newfileUploadRequest(uri string, form map[string]string, formFileName, path string) (*http.Request, error) {
file, err := os.Open(path)
if err != nil {
return nil, err
}
defer file.Close()
body := &bytes.Buffer{}
writer := multipart.NewWriter(body)
part, err := writer.CreateFormFile(formFileName, filepath.Base(path))
if err != nil {
return nil, err
}
_, err = io.Copy(part, file)
if err != nil {
return nil, err
}
for key, val := range form {
err = writer.WriteField(key, val)
if err != nil {
return nil, err
}
}
err = writer.Close()
if err != nil {
return nil, err
}
req, err := http.NewRequest("POST", uri, body)
req.Header.Add("Content-Type", writer.FormDataContentType())
if err != nil {
return nil, err
}
return req, nil
}
//addGroupMember_test.go
package Network
func testAddGroupMember() {
var test []testData = []testData{
{
id: "100000000",
data: `
{
"Method":"/addGroupMember",
"Args":{
"groupID":"100000006",
"memberIDs":[
"100000004",
"100000007",
"100000008"
]
}
}
`, expCode: Success},
{
id: "100000000",
data: `
{
"Method":"/addGroupMember",
"Args":{
"groupID":"1006",
"memberIDs":[
"100000004"
]
}
}
`, expCode: errInvalidID},
{
id: "100000000",
data: `
{
"Method":"/addGroupMember",
"Args":{
"groupID":"100000000",
"memberIDs":[
"100000004"
]
}
}
`, expCode: errInvalidID},
{
id: "100000000",
data: `
{
"Method":"/addGroupMember",
"Args":{
"groupID":"10000000",
"memberIDs":[
"100000004"
]
}
}
`, expCode: errNotExistID},
{
id: "100000000",
data: `
{
"Method":"/addGroupMember",
"Args":{
"groupID":100000006,
"memberIDs":[
"100000004"
]
}
}
`, expCode: errAssertType},
{
id: "100000001",
data: `
{
"Method":"/addGroupMember",
"Args":{
"groupID":"100000006",
"memberIDs":[
"100000004"
]
}
}
`, expCode: errNoPermission},
{
id: "1000000010",
data: `
{
"Method":"/addGroupMember",
"Args":{
"groupID":"100000006",
"memberIDs":[
"100000004"
]
}
}
`, expCode: errNoPermission},
{
id: "100000000",
data: `
{
"Method":"/addGroupMember",
"Args":{
"groupID":"100000006"
}
}
`, expCode: errAssertType},
{
id: "100000000",
data: `
{
"Method":"/addGroupMember",
"Args":{
"groupID":"100000006",
"memberIDs":[
"100000006",
"10000001"
]
}
}
`, expCode: errInvalidID},
{
id: "100000000",
data: `
{
"Method":"/addGroupMember",
"Args":{
"groupID":"100000006",
"memberIDs":[
"100000007",
"100000004",
"100000007",
"100000008"
]
}
}
`, expCode: errInvalidID},
{
id: "100000000",
data: `
{
"Method":"/addGroupMember",
"Args":{
"groupID":"100000006",
"memberIDs":[
"100000009",
"100000007",
"100000008"
]
}
}
`, expCode: errInvalidID},
{
id: "100000000",
data: `
{
"Method":"/addGroupMember",
"Args":{
"groupID":"100000006",
"memberIDs":[
"100000a",
"100000007",
"100000008"
]
}
}
`, expCode: errInvalidID},
//num memberIDs > 200
}
dyTest(test)
}
以上是关于go http/websocket接口测试代码的主要内容，如果未能解决你的问题，请参考以下文章