zishuo/stun
1
0
Fork 0
mirror of https://github.com/pion/stun.git synced 2025-10-25 00:30:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

client: add initialization checks

Browse Source
This commit is contained in:
Aleksandr Razumov
2018-04-29 14:20:09 +03:00
parent bb835f8153
commit f539368a80
2 changed files with 32 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
client_test.go
View File

@@ -447,6 +447,15 @@ func TestClientGC(t *testing.T) {
}
}
func TestClientCheckInit(t *testing.T) {
if err := (&Client{}).Indicate(nil); err != ErrClientNotInitialized {
t.Error("unexpected error")
}
if err := (&Client{}).Do(nil, time.Time{}, nil); err != ErrClientNotInitialized {
t.Error("unexpected error")
}
}
func TestClientFinalizer(t *testing.T) {
var buf bytes.Buffer
log.SetOutput(&buf)
@@ -466,9 +475,7 @@ func TestClientFinalizer(t *testing.T) {
if err != nil {
log.Fatal(err)
}
if err := c.Close(); err != nil {
t.Error(err)
}
clientFinalizer(c)
clientFinalizer(c)
response := MustBuild(TransactionID, BindingSuccess)
response.Encode()
@@ -497,7 +504,7 @@ func TestClientFinalizer(t *testing.T) {
if reader.Err() != nil {
t.Error(err)
}
if lines != 2 {
if lines != 3 {
t.Error("incorrect count of log lines:", lines)
}
}