github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-10 12:32:03 +00:00
Code Issues Projects Releases Wiki Activity

add ut for audit useragent

Browse Source
This commit is contained in:
xuzhonghu 2018-06-08 17:06:39 +08:00
parent b5990b78cb
commit a8b0ccc70c
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
staging/src/k8s.io/apiserver/pkg/endpoints/audit_test.go
View File

@ -63,6 +63,7 @@ func TestAudit(t *testing.T) {
Other: "bla", Other: "bla",
} }
simpleCPrimeJSON, _ := runtime.Encode(testCodec, simpleCPrime) simpleCPrimeJSON, _ := runtime.Encode(testCodec, simpleCPrime)
userAgent := "audit-test"
// event checks // event checks
noRequestBody := func(i int) eventCheck { noRequestBody := func(i int) eventCheck {
@ -111,6 +112,16 @@ func TestAudit(t *testing.T) {
return nil return nil
} }
} }
requestUserAgentMatches := func(userAgent string) eventCheck {
return func(events []*auditinternal.Event) error {
for i := range events {
if events[i].UserAgent != userAgent {
return fmt.Errorf("expected request user agent to match %q, but got: %q", userAgent, events[i].UserAgent)
}
}
return nil
}
}
for _, test := range []struct { for _, test := range []struct {
desc string desc string
@ -295,6 +306,8 @@ func TestAudit(t *testing.T) {
t.Errorf("[%s] error creating the request: %v", test.desc, err) t.Errorf("[%s] error creating the request: %v", test.desc, err)
} }
req.Header.Set("User-Agent", userAgent)
response, err := client.Do(req) response, err := client.Do(req)
if err != nil { if err != nil {
t.Errorf("[%s] error: %v", test.desc, err) t.Errorf("[%s] error: %v", test.desc, err)
@ -326,6 +339,10 @@ func TestAudit(t *testing.T) {
t.Errorf("[%s,%d] %v", test.desc, i, err) t.Errorf("[%s,%d] %v", test.desc, i, err)
} }
} }
if err := requestUserAgentMatches(userAgent)(events); err != nil {
t.Errorf("[%s] %v", test.desc, err)
}
} }
if len(events) > 0 { if len(events) > 0 {