Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests: Update assertAuthority to explain filtering put method requests

Browse Source
This commit is contained in:
Marek Siarkowicz 2022-02-01 16:40:23 +01:00
parent 43c1631f26
commit 902dc05014
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/integration/grpc_test.go
View File

@ -108,12 +108,13 @@ func TestAuthority(t *testing.T) {
kv := setupClient(t, tc.clientURLPattern, clus, tlsConfig) kv := setupClient(t, tc.clientURLPattern, clus, tlsConfig)
defer kv.Close() defer kv.Close()
putRequestMethod := "/etcdserverpb.KV/Put"
_, err := kv.Put(context.TODO(), "foo", "bar") _, err := kv.Put(context.TODO(), "foo", "bar")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
assertAuthority(t, templateAuthority(t, tc.expectAuthorityPattern, clus.Members[0]), clus) assertAuthority(t, templateAuthority(t, tc.expectAuthorityPattern, clus.Members[0]), clus, putRequestMethod)
}) })
} }
} }
@ -181,12 +182,12 @@ func templateAuthority(t *testing.T, pattern string, m *integration.Member) stri
return authority return authority
} }
func assertAuthority(t *testing.T, expectedAuthority string, clus *integration.Cluster) { func assertAuthority(t *testing.T, expectedAuthority string, clus *integration.Cluster, filterMethod string) {
t.Helper() t.Helper()
requestsFound := 0 requestsFound := 0
for _, m := range clus.Members { for _, m := range clus.Members {
for _, r := range m.RecordedRequests() { for _, r := range m.RecordedRequests() {
if r.FullMethod != "/etcdserverpb.KV/Put" { if filterMethod != "" && r.FullMethod != filterMethod {
continue continue
} }
requestsFound++ requestsFound++