scottk/vault
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/vault.git synced 2025-10-30 02:02:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

headers test tweaks (#26532)

Browse Source
This commit is contained in:
Peter Wilson
2024-04-19 09:34:20 +01:00
committed by GitHub
parent 1e36019f1c
commit 7c9e4f39b6
1 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
audit/headers_test.go
View File

@@ -86,6 +86,8 @@ func TestAuditedHeadersConfig_CRUD(t *testing.T) {
} }
func testAddHeaders(t *testing.T, conf *HeadersConfig) { func testAddHeaders(t *testing.T, conf *HeadersConfig) {
t.Helper()
err := conf.Add(context.Background(), "X-Test-Header", false) err := conf.Add(context.Background(), "X-Test-Header", false)
if err != nil { if err != nil {
t.Fatalf("Error when adding header to config: %s", err) t.Fatalf("Error when adding header to config: %s", err)
@@ -162,6 +164,8 @@ func testAddHeaders(t *testing.T, conf *HeadersConfig) {
} }
func testRemoveHeaders(t *testing.T, conf *HeadersConfig) { func testRemoveHeaders(t *testing.T, conf *HeadersConfig) {
t.Helper()
err := conf.Remove(context.Background(), "X-Test-Header") err := conf.Remove(context.Background(), "X-Test-Header")
if err != nil { if err != nil {
t.Fatalf("Error when adding header to config: %s", err) t.Fatalf("Error when adding header to config: %s", err)
@@ -232,8 +236,10 @@ func TestAuditedHeadersConfig_ApplyConfig(t *testing.T) {
conf := mockAuditedHeadersConfig(t) conf := mockAuditedHeadersConfig(t)
conf.Add(context.Background(), "X-TesT-Header", false) err := conf.Add(context.Background(), "X-TesT-Header", false)
conf.Add(context.Background(), "X-Vault-HeAdEr", true) require.NoError(t, err)
err = conf.Add(context.Background(), "X-Vault-HeAdEr", true)
require.NoError(t, err)
reqHeaders := map[string][]string{ reqHeaders := map[string][]string{
"X-Test-Header": {"foo"}, "X-Test-Header": {"foo"},
@@ -364,8 +370,10 @@ func TestAuditedHeadersConfig_ApplyConfig_HashStringError(t *testing.T) {
conf := mockAuditedHeadersConfig(t) conf := mockAuditedHeadersConfig(t)
conf.Add(context.Background(), "X-TesT-Header", false) err := conf.Add(context.Background(), "X-TesT-Header", false)
conf.Add(context.Background(), "X-Vault-HeAdEr", true) require.NoError(t, err)
err = conf.Add(context.Background(), "X-Vault-HeAdEr", true)
require.NoError(t, err)
reqHeaders := map[string][]string{ reqHeaders := map[string][]string{
"X-Test-Header": {"foo"}, "X-Test-Header": {"foo"},
@@ -375,7 +383,7 @@ func TestAuditedHeadersConfig_ApplyConfig_HashStringError(t *testing.T) {
salter := &FailingSalter{} salter := &FailingSalter{}
_, err := conf.ApplyConfig(context.Background(), reqHeaders, salter) _, err = conf.ApplyConfig(context.Background(), reqHeaders, salter)
if err == nil { if err == nil {
t.Fatal("expected error from ApplyConfig") t.Fatal("expected error from ApplyConfig")
} }
@@ -403,7 +411,8 @@ func BenchmarkAuditedHeaderConfig_ApplyConfig(b *testing.B) {
// Reset the timer since we did a lot above // Reset the timer since we did a lot above
b.ResetTimer() b.ResetTimer()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
conf.ApplyConfig(context.Background(), reqHeaders, salter) _, err := conf.ApplyConfig(context.Background(), reqHeaders, salter)
require.NoError(b, err)
} }
} }