headerRewrite_test.go

// Copyright (c) 2021 Red Hat, Inc.
// This program and the accompanying materials are made
// available under the terms of the Eclipse Public License 2.0
// which is available at https://www.eclipse.org/legal/epl-2.0/
//
// SPDX-License-Identifier: EPL-2.0
//
// Contributors:
//   Red Hat, Inc. - initial API and implementation
//
package header_rewrite_traefik_plugin

import (
	"net/http"
	"testing"
)

func TestHeaderRewritten(t *testing.T) {
	headers := &http.Header{}
	headers.Add("hello", "world")

	rewriteHeaders(headers, &Config{
		From:         "hello",
		To:           "goodbye",
		Prefix:       "",
		KeepOriginal: true,
	})

	if headers.Get("goodbye") != "world" {
		t.Errorf("header 'hello' should be rewritten to new key 'goodbye'")
	}

	if headers.Get("hello") != "world" {
		t.Errorf("Original header should not be rewritten.")
	}
}

func TestKeepOriginal(t *testing.T) {
	headers := &http.Header{}
	headers.Add("hello", "world")

	rewriteHeaders(headers, &Config{
		From:         "hello",
		To:           "goodbye",
		Prefix:       "",
		KeepOriginal: false,
	})

	if headers.Get("goodbye") != "world" {
		t.Errorf("header 'hello' should be rewritten to new key 'goodbye'")
	}

	if headers.Get("hello") != "" {
		t.Errorf("original header should be removed")
	}
}

func TestPrefix(t *testing.T) {
	headers := &http.Header{}
	headers.Add("hello", "world")

	rewriteHeaders(headers, &Config{
		From:         "hello",
		To:           "goodbye",
		Prefix:       "prefix ",
		KeepOriginal: true,
	})

	if headers.Get("goodbye") != "prefix world" {
		t.Errorf("header 'hello' should be rewritten to new key 'goodbye'")
	}

	if headers.Get("hello") != "world" {
		t.Errorf("original header should not be removed or modified")
	}
}

func TestMultipleValuesUnderSameKey(t *testing.T) {
	headers := &http.Header{}
	headers.Add("hello", "world")
	headers.Add("hello", "there")

	rewriteHeaders(headers, &Config{
		From:         "hello",
		To:           "goodbye",
		Prefix:       "prefix ",
		KeepOriginal: false,
	})

	headersResult := headers.Values("goodbye")

	if len(headersResult) != 2 {
		t.Errorf("all header values under same key must be rewritten. '%+v'", headersResult)
	}

	if headersResult[0] != "prefix world" {
		t.Errorf("all header values under same key must be rewritten. '%+v'", headersResult)
	}

	if headersResult[1] != "prefix there" {
		t.Errorf("all header values under same key must be rewritten. '%+v'", headersResult)
	}

	if headers.Get("hello") != "" {
		t.Errorf("original header must not be removed when keepOriginal: 'false'")
	}
}

func TestKeepOriginalTarget(t *testing.T) {
	headers := &http.Header{}
	headers.Add("hello", "world")
	headers.Add("goodbye", "there")

	rewriteHeaders(headers, &Config{
		From:               "hello",
		To:                 "goodbye",
		KeepOriginalTarget: true,
	})

	headersResult := headers.Values("goodbye")

	if len(headersResult) != 2 {
		t.Errorf("There should be 2 target headers. '%+v'", headersResult)
	}

	if headersResult[0] != "there" {
		t.Errorf("First target header should be the original one. '%+v'", headersResult)
	}

	if headersResult[1] != "world" {
		t.Errorf("Second target header should be the new one. '%+v'", headersResult)
	}
}

func TestNotKeepOriginalTarget(t *testing.T) {
	headers := &http.Header{}
	headers.Add("hello", "world")
	headers.Add("goodbye", "there")

	rewriteHeaders(headers, &Config{
		From:               "hello",
		To:                 "goodbye",
		KeepOriginalTarget: false,
	})

	headersResult := headers.Values("goodbye")

	if len(headersResult) != 1 {
		t.Errorf("Target header should be cleaned first. '%+v'", headersResult)
	}

	if headersResult[0] != "world" {
		t.Errorf("The target header value should be the rewritten one. '%+v'", headersResult)
	}
}