mirror of https://github.com/grpc/grpc-go.git
95 lines
2.7 KiB
Go
95 lines
2.7 KiB
Go
/*
|
|
*
|
|
* Copyright 2022 gRPC authors.
|
|
*
|
|
* 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 outlierdetection
|
|
|
|
import (
|
|
"sync/atomic"
|
|
"testing"
|
|
"unsafe"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
)
|
|
|
|
func (b1 *bucket) Equal(b2 *bucket) bool {
|
|
if b1 == nil && b2 == nil {
|
|
return true
|
|
}
|
|
if (b1 != nil) != (b2 != nil) {
|
|
return false
|
|
}
|
|
if b1.numSuccesses != b2.numSuccesses {
|
|
return false
|
|
}
|
|
return b1.numFailures == b2.numFailures
|
|
}
|
|
|
|
func (cc *callCounter) Equal(cc2 *callCounter) bool {
|
|
if cc == nil && cc2 == nil {
|
|
return true
|
|
}
|
|
if (cc != nil) != (cc2 != nil) {
|
|
return false
|
|
}
|
|
ab1 := (*bucket)(atomic.LoadPointer(&cc.activeBucket))
|
|
ab2 := (*bucket)(atomic.LoadPointer(&cc2.activeBucket))
|
|
if !ab1.Equal(ab2) {
|
|
return false
|
|
}
|
|
return cc.inactiveBucket.Equal(cc2.inactiveBucket)
|
|
}
|
|
|
|
// TestClear tests that clear on the call counter clears (everything set to 0)
|
|
// the active and inactive buckets.
|
|
func (s) TestClear(t *testing.T) {
|
|
cc := newCallCounter()
|
|
ab := (*bucket)(atomic.LoadPointer(&cc.activeBucket))
|
|
ab.numSuccesses = 1
|
|
ab.numFailures = 2
|
|
cc.inactiveBucket.numSuccesses = 4
|
|
cc.inactiveBucket.numFailures = 5
|
|
cc.clear()
|
|
// Both the active and inactive buckets should be cleared.
|
|
ccWant := newCallCounter()
|
|
if diff := cmp.Diff(cc, ccWant); diff != "" {
|
|
t.Fatalf("callCounter is different than expected, diff (-got +want): %v", diff)
|
|
}
|
|
}
|
|
|
|
// TestSwap tests that swap() on the callCounter successfully has the desired
|
|
// end result of inactive bucket containing the previous active buckets data,
|
|
// and the active bucket being cleared.
|
|
func (s) TestSwap(t *testing.T) {
|
|
cc := newCallCounter()
|
|
ab := (*bucket)(atomic.LoadPointer(&cc.activeBucket))
|
|
ab.numSuccesses = 1
|
|
ab.numFailures = 2
|
|
cc.inactiveBucket.numSuccesses = 4
|
|
cc.inactiveBucket.numFailures = 5
|
|
ib := cc.inactiveBucket
|
|
cc.swap()
|
|
// Inactive should pick up active's data, active should be swapped to zeroed
|
|
// inactive.
|
|
ccWant := newCallCounter()
|
|
ccWant.inactiveBucket.numSuccesses = 1
|
|
ccWant.inactiveBucket.numFailures = 2
|
|
atomic.StorePointer(&ccWant.activeBucket, unsafe.Pointer(ib))
|
|
if diff := cmp.Diff(cc, ccWant); diff != "" {
|
|
t.Fatalf("callCounter is different than expected, diff (-got +want): %v", diff)
|
|
}
|
|
}
|