unistack-org/micro-register-mdns
4
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases 3 Wiki Activity

fix test

Browse Source 
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2021-02-20 14:23:07 +03:00
parent 9c94a3e6a8
commit 21928ee740
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
mdns_test.go
View File

@ -1,6 +1,7 @@
package mdns
import (
"context"
"os"
"testing"
"time"
@ -9,6 +10,8 @@ import (
)
func TestMDNS(t *testing.T) {
ctx := context.Background()
// skip test in travis because of sendto: operation not permitted error
if travis := os.Getenv("TRAVIS"); travis == "true" {
t.Skip()
@ -69,12 +72,12 @@ func TestMDNS(t *testing.T) {
for _, service := range testData {
// register service
if err := r.Register(service); err != nil {
if err := r.Register(ctx, service); err != nil {
t.Fatal(err)
}
// get registered service
s, err := r.LookupService(service.Name)
s, err := r.LookupService(ctx, service.Name)
if err != nil {
t.Fatal(err)
}
@ -106,7 +109,7 @@ func TestMDNS(t *testing.T) {
}
}
services, err := r.ListServices()
services, err := r.ListServices(ctx)
if err != nil {
t.Fatal(err)
}
@ -124,14 +127,14 @@ func TestMDNS(t *testing.T) {
}
// deregister
if err := r.Deregister(service); err != nil {
if err := r.Deregister(ctx, service); err != nil {
t.Fatal(err)
}
time.Sleep(time.Millisecond * 5)
// check its gone
s, _ := r.GetService(service.Name)
s, _ := r.LookupService(ctx, service.Name)
if len(s) > 0 {
t.Fatalf("Expected nothing got %+v", s[0])
}
@ -200,6 +203,8 @@ func TestEncoding(t *testing.T) {
}
func TestWatcher(t *testing.T) {
ctx := context.Background()
if travis := os.Getenv("TRAVIS"); travis == "true" {
t.Skip()
}
@ -286,7 +291,7 @@ func TestWatcher(t *testing.T) {
// new registry
r := NewRegister(opts...)
w, err := r.Watch()
w, err := r.Watch(ctx)
if err != nil {
t.Fatal(err)
}
@ -294,7 +299,7 @@ func TestWatcher(t *testing.T) {
for _, service := range testData {
// register service
if err := r.Register(service); err != nil {
if err := r.Register(ctx, service); err != nil {
t.Fatal(err)
}
@ -317,7 +322,7 @@ func TestWatcher(t *testing.T) {
}
// deregister
if err := r.Deregister(service); err != nil {
if err := r.Deregister(ctx, service); err != nil {
t.Fatal(err)
}