oss/FrankerFaceZ
oss/FrankerFaceZ
1
0
Fork 0
mirror of https://github.com/FrankerFaceZ/FrankerFaceZ.git synced 2025-09-16 10:06:54 +00:00
Code Wiki Activity

More test infrastructure

Browse source
This commit is contained in:
Kane York 2015-10-26 11:25:17 -07:00
parent 0be3693c99
commit 812c6f2557
1 changed files with 20 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

28
socketserver/internal/server/publisher_test.go
View file

@ -53,6 +53,21 @@ func TSendMessage(tb testing.TB, conn *websocket.Conn, messageId int, command Co
return err == nil return err == nil
} }
type TURLs struct {
Websocket string
Origin string
PubMsg string
}
func TGetUrls(testserver httptest.Server) TURLs {
addr := testserver.Listener.Addr().String()
return TURLs{
Websocket: fmt.Sprintf("ws://%s/", addr),
Origin: fmt.Sprintf("http://%s"),
PubMsg: fmt.Sprintf("http://%s/pub_msg"),
}
}
func TestSubscriptionAndPublish(t *testing.T) { func TestSubscriptionAndPublish(t *testing.T) {
var doneWg sync.WaitGroup var doneWg sync.WaitGroup
var readyWg sync.WaitGroup var readyWg sync.WaitGroup
@ -74,11 +89,9 @@ func TestSubscriptionAndPublish(t *testing.T) {
server := httptest.NewUnstartedServer(serveMux) server := httptest.NewUnstartedServer(serveMux)
server.Start() server.Start()
wsUrl := fmt.Sprintf("ws://%s/", server.Listener.Addr().String()) urls := TGetUrls(server)
originUrl := fmt.Sprintf("http://%s", server.Listener.Addr().String())
publishUrl := fmt.Sprintf("http://%s/pub_msg", server.Listener.Addr().String())
conn, err := websocket.Dial(wsUrl, "", originUrl) conn, err := websocket.Dial(urls.Websocket, "", urls.Origin)
if err != nil { if err != nil {
t.Error(err) t.Error(err)
return return
@ -116,7 +129,7 @@ func TestSubscriptionAndPublish(t *testing.T) {
panic("halting test") panic("halting test")
} }
resp, err := http.PostForm(publishUrl, sealedForm) resp, err := http.PostForm(urls.PubMsg, sealedForm)
if err != nil { if err != nil {
t.Error(err) t.Error(err)
server.CloseClientConnections() server.CloseClientConnections()
@ -158,8 +171,7 @@ func BenchmarkThousandUserSubscription(b *testing.B) {
server := httptest.NewUnstartedServer(serveMux) server := httptest.NewUnstartedServer(serveMux)
server.Start() server.Start()
wsUrl := fmt.Sprintf("ws://%s/", server.Listener.Addr().String()) urls := TGetUrls(server)
originUrl := fmt.Sprintf("http://%s", server.Listener.Addr().String())
message := ClientMessage{MessageID: -1, Command: "testdata", Arguments: TestData} message := ClientMessage{MessageID: -1, Command: "testdata", Arguments: TestData}
@ -180,7 +192,7 @@ func BenchmarkThousandUserSubscription(b *testing.B) {
b.ResetTimer() b.ResetTimer()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
conn, err := websocket.Dial(wsUrl, "", originUrl) conn, err := websocket.Dial(urls.Websocket, "", urls.Origin)
if err != nil { if err != nil {
b.Error(err) b.Error(err)
break break