chore: add more tests (#3010)

This commit is contained in:
Kevin Wan
2023-03-10 21:56:19 +08:00
committed by GitHub
parent c8a17a97be
commit fbf129d535
6 changed files with 109 additions and 13 deletions

View File

@@ -72,7 +72,7 @@ type p2cPicker struct {
lock sync.Mutex
}
func (p *p2cPicker) Pick(info balancer.PickInfo) (balancer.PickResult, error) {
func (p *p2cPicker) Pick(_ balancer.PickInfo) (balancer.PickResult, error) {
p.lock.Lock()
defer p.lock.Unlock()

View File

@@ -123,6 +123,15 @@ func TestP2cPicker_Pick(t *testing.T) {
}
}
func TestPickerWithEmptyConns(t *testing.T) {
var picker p2cPicker
_, err := picker.Pick(balancer.PickInfo{
FullMethodName: "/",
Ctx: context.Background(),
})
assert.ErrorIs(t, err, balancer.ErrNoSubConnAvailable)
}
type mockClientConn struct {
// add random string member to avoid map key equality.
id string