test/integration: update TestWebsocketQuery

Because 6517bbc392 introduced automatic
status update message upon connection.

Change-Id: I17383930d97b38d2a4abf7a646aa291bf84d7014
This commit is contained in:
mpl 2014-09-02 00:57:18 +02:00
parent 9ddc3988aa
commit 638c660227
1 changed files with 9 additions and 0 deletions

View File

@ -90,6 +90,15 @@ func TestWebsocketQuery(t *testing.T) {
errc <- err
return
}
if !strings.HasPrefix(string(inMsg), `{"tag":"_status"`) {
errc <- fmt.Errorf("unexpected message type=%d msg=%q, wanted status update", inType, inMsg)
return
}
inType, inMsg, err = wc.ReadMessage()
if err != nil {
errc <- err
return
}
if strings.Contains(string(inMsg), pn.String()) {
errc <- nil
return