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

@@ -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