waf_test.go

// Copyright 2022 Juan Pablo Tosso and the OWASP Coraza contributors
// SPDX-License-Identifier: Apache-2.0

package coraza

import "testing"

func TestNewWAFLimits(t *testing.T) {
	testCases := map[string]struct {
		expectedErr string
		cfg         requestBodyConfig
	}{
		"empty limit": {
			cfg:         requestBodyConfig{},
			expectedErr: "request body limit should be bigger than 0",
		},
		"memory limit bigger than limit": {
			cfg: requestBodyConfig{
				limit:         5,
				inMemoryLimit: 9,
			},
			expectedErr: "request body limit should be at least the memory limit",
		},
	}

	for name, tCase := range testCases {
		t.Run(name, func(t *testing.T) {
			_, err := NewWAF(&wafConfig{
				requestBody: &tCase.cfg,
			})

			if tCase.expectedErr != "" {
				if err == nil {
					t.Fatal("expected error")
				}

				if want, have := tCase.expectedErr, err.Error(); want != have {
					t.Fatalf("unexpected error: want %q, have %q", want, have)
				}
			}
		})
	}
}