github.com/tickoalcantara12/micro/v3@v3.0.0-20221007104245-9d75b9bcbab9/service/registry/memory/watcher_test.go (about)

     1  // Licensed under the Apache License, Version 2.0 (the "License");
     2  // you may not use this file except in compliance with the License.
     3  // You may obtain a copy of the License at
     4  //
     5  //     https://www.apache.org/licenses/LICENSE-2.0
     6  //
     7  // Unless required by applicable law or agreed to in writing, software
     8  // distributed under the License is distributed on an "AS IS" BASIS,
     9  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    10  // See the License for the specific language governing permissions and
    11  // limitations under the License.
    12  //
    13  // Original source: github.com/micro/go-micro/v3/registry/memory/watcher_test.go
    14  
    15  package memory
    16  
    17  import (
    18  	"testing"
    19  
    20  	"github.com/tickoalcantara12/micro/v3/service/registry"
    21  )
    22  
    23  func TestWatcher(t *testing.T) {
    24  	w := &Watcher{
    25  		id:   "test",
    26  		res:  make(chan *registry.Result),
    27  		exit: make(chan bool),
    28  		wo: registry.WatchOptions{
    29  			Domain: registry.WildcardDomain,
    30  		},
    31  	}
    32  
    33  	go func() {
    34  		w.res <- &registry.Result{
    35  			Service: &registry.Service{Name: "foo"},
    36  		}
    37  	}()
    38  
    39  	_, err := w.Next()
    40  	if err != nil {
    41  		t.Fatal("unexpected err", err)
    42  	}
    43  
    44  	w.Stop()
    45  
    46  	if _, err := w.Next(); err == nil {
    47  		t.Fatal("expected error on Next()")
    48  	}
    49  }