cache-config/t3c-preprocess/t3c-preprocess_test.go

package main

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

import (
	"testing"

	"github.com/apache/trafficcontrol/lib/go-atscfg"
	"github.com/apache/trafficcontrol/lib/go-tc"
	"github.com/apache/trafficcontrol/lib/go-util"
)

func TestPreprocessConfigFile(t *testing.T) {
	// the TCP port replacement is fundamentally different for 80 vs non-80, so test both
	t.Run("verify port 80 replace", func(t *testing.T) {
		server := &atscfg.Server{}
		server.TCPPort = util.IntPtr(8080)
		server.Interfaces = []tc.ServerInterfaceInfoV40{}
		{
			ssi := tc.ServerInterfaceInfoV40{}
			ssi.IPAddresses = []tc.ServerIPAddress{
				tc.ServerIPAddress{
					Address:        "127.0.2.1",
					ServiceAddress: true,
				},
			}
			server.Interfaces = append(server.Interfaces, ssi)
		}
		server.HostName = util.StrPtr("my-edge")
		server.DomainName = util.StrPtr("example.net")
		cfgFile := "abc__SERVER_TCP_PORT__def__CACHE_IPV4__ghi __RETURN__  \t __HOSTNAME__ jkl __FULL_HOSTNAME__ \n__SOMETHING__ __ELSE__\nmno\r\n"

		actual := PreprocessConfigFile(server, cfgFile)

		expected := "abc8080def127.0.2.1ghi\nmy-edge jkl my-edge.example.net \n__SOMETHING__ __ELSE__\nmno\r\n"

		if expected != actual {
			t.Errorf("PreprocessConfigFile expected '%v' actual '%v'", expected, actual)
		}
	})

	t.Run("verify nonstandard port replace", func(t *testing.T) {
		server := &atscfg.Server{}
		server.TCPPort = util.IntPtr(80)
		server.Interfaces = []tc.ServerInterfaceInfoV40{}
		{
			si := tc.ServerInterfaceInfoV40{}
			si.IPAddresses = []tc.ServerIPAddress{
				tc.ServerIPAddress{
					Address:        "127.0.2.1",
					ServiceAddress: true,
				},
			}
			server.Interfaces = append(server.Interfaces, si)
		}
		server.HostName = util.StrPtr("my-edge")
		server.DomainName = util.StrPtr("example.net")

		cfgFile := "abc__SERVER_TCP_PORT__def__CACHE_IPV4__ghi __RETURN__  \t __HOSTNAME__ jkl __FULL_HOSTNAME__ \n__SOMETHING__ __ELSE__\nmno:__SERVER_TCP_PORT__\r\n"

		actual := PreprocessConfigFile(server, cfgFile)

		expected := "abc__SERVER_TCP_PORT__def127.0.2.1ghi\nmy-edge jkl my-edge.example.net \n__SOMETHING__ __ELSE__\nmno\r\n"

		if expected != actual {
			t.Errorf("PreprocessConfigFile expected '%v' actual '%v'", expected, actual)
		}
	})

	t.Run("verify cachegroup replace", func(t *testing.T) {
		server := &atscfg.Server{}
		server.TCPPort = util.IntPtr(80)
		server.Cachegroup = util.StrPtr("mycachegroup")
		server.HostName = util.StrPtr("my-edge")
		server.DomainName = util.StrPtr("example.net")

		cfgFile := "abc__CACHEGROUP__defghi __RETURN__  \t __HOSTNAME__ jkl __FULL_HOSTNAME__ \n__SOMETHING__ __ELSE__\nmno\r\n"

		actual := PreprocessConfigFile(server, cfgFile)

		expected := "abcmycachegroupdefghi\nmy-edge jkl my-edge.example.net \n__SOMETHING__ __ELSE__\nmno\r\n"

		if expected != actual {
			t.Errorf("PreprocessConfigFile expected '%v' actual '%v'", expected, actual)
		}
	})
}