blob: efefcc5dc9db629aa0f109a8da4f4227c18504fa [file] [log] [blame]
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package kubernetes
import (
"context"
"strconv"
"sync"
"testing"
"time"
)
func TestWatchSet(t *testing.T) {
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()
s := newWatcherSet(ctx)
wg := sync.WaitGroup{}
for i := 0; i < 2; i++ {
wg.Add(1)
go func() {
defer wg.Done()
w, err := s.Watch("key-1", false)
if err != nil {
t.Error(err)
return
}
for {
select {
case e := <-w.ResultChan():
t.Logf("consumer %s got %s\n", w.ID(), e.Key)
case <-w.done():
t.Logf("consumer %s stopped", w.ID())
return
}
}
}()
}
for i := 2; i < 3; i++ {
wg.Add(1)
go func() {
defer wg.Done()
w, err := s.Watch("key", true)
if err != nil {
t.Error(err)
return
}
for {
select {
case e := <-w.ResultChan():
t.Logf("prefix consumer %s got %s\n", w.ID(), e.Key)
case <-w.done():
t.Logf("prefix consumer %s stopped", w.ID())
return
}
}
}()
}
for i := 0; i < 5; i++ {
go func(i int) {
if err := s.Put(&WatcherEvent{
Key: "key-" + strconv.Itoa(i),
Value: strconv.Itoa(i),
}); err != nil {
t.Error(err)
return
}
}(i)
}
wg.Wait()
}