83 lines
1.7 KiB
Go
83 lines
1.7 KiB
Go
/*
|
|
Copyright 2014 Google Inc. All rights reserved.
|
|
|
|
Licensed 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 election
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type slowService struct {
|
|
t *testing.T
|
|
on bool
|
|
// We explicitly have no lock to prove that
|
|
// Start and Stop are not called concurrently.
|
|
changes chan<- bool
|
|
}
|
|
|
|
func (s *slowService) Start() {
|
|
if s.on {
|
|
s.t.Errorf("started already on service")
|
|
}
|
|
time.Sleep(2 * time.Millisecond)
|
|
s.on = true
|
|
s.changes <- true
|
|
}
|
|
|
|
func (s *slowService) Stop() {
|
|
if !s.on {
|
|
s.t.Errorf("stopped already off service")
|
|
}
|
|
time.Sleep(2 * time.Millisecond)
|
|
s.on = false
|
|
s.changes <- false
|
|
}
|
|
|
|
func Test(t *testing.T) {
|
|
m := NewFake()
|
|
changes := make(chan bool, 1500)
|
|
s := &slowService{t: t, changes: changes}
|
|
go Notify(m, "", "me", s)
|
|
|
|
done := make(chan struct{})
|
|
go func() {
|
|
for i := 0; i < 500; i++ {
|
|
for _, key := range []string{"me", "notme", "alsonotme"} {
|
|
m.ChangeMaster(Master(key))
|
|
}
|
|
}
|
|
close(done)
|
|
}()
|
|
|
|
<-done
|
|
time.Sleep(8 * time.Millisecond)
|
|
close(changes)
|
|
|
|
changeList := []bool{}
|
|
for {
|
|
change, ok := <-changes
|
|
if !ok {
|
|
break
|
|
}
|
|
changeList = append(changeList, change)
|
|
}
|
|
|
|
if len(changeList) > 1000 {
|
|
t.Errorf("unexpected number of changes: %v", len(changeList))
|
|
}
|
|
}
|