kraken

1
// Copyright (c) 2016-2019 Uber Technologies, Inc.
2
//
3
// Licensed under the Apache License, Version 2.0 (the "License");
4
// you may not use this file except in compliance with the License.
5
// You may obtain a copy of the License at
6
//
7
//     http://www.apache.org/licenses/LICENSE-2.0
8
//
9
// Unless required by applicable law or agreed to in writing, software
10
// distributed under the License is distributed on an "AS IS" BASIS,
11
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
// See the License for the specific language governing permissions and
13
// limitations under the License.
14
package hashring
15

16
 import (
17
	"testing"
18
	"time"
19

20
 	"github.com/andres-erbsen/clock"
21
	"github.com/stretchr/testify/require"
22

23
 	"github.com/uber/kraken/core"
24
	"github.com/uber/kraken/lib/healthcheck"
25
	"github.com/uber/kraken/lib/hostlist"
26
)
27

28
 func TestPassiveRingFailedAffectsRefreshFilter(t *testing.T) {
29
	require := require.New(t)
30

31
 	r := NewPassive(
32
		Config{MaxReplica: 3},
33
		hostlist.Fixture(addrsFixture(10)...),
34
		healthcheck.NewPassiveFilter(healthcheck.PassiveFilterConfig{
35
				Fails: 3,
36
				FailTimeout: 5 * time.Second,
37
			}, clock.New()))
38

39
 	d := core.DigestFixture()
40

41
 	replicas := r.Locations(d)
42
	require.Len(replicas, 3)
43

44
 	for i := 0; i < 4; i++ {
45
		r.Failed(replicas[0])
46
	}
47
	r.Refresh()
48

49
 	result := r.Locations(d)
50
	require.Equal(replicas[1:], result)
51
}
52