SMTP-Relay/internal/config/proto_test.go

package config

import (
	"testing"
)

func TestSplitProto(t *testing.T) {
	var tests = []struct {
		input string
		proto string
		addr  string
	}{
		{
			input: "localhost",
			proto: "",
			addr:  "localhost",
		},
		{
			input: "tls://my.local.domain",
			proto: "tls",
			addr:  "my.local.domain",
		},
		{
			input: "starttls://my.local.domain",
			proto: "starttls",
			addr:  "my.local.domain",
		},
	}

	for i, test := range tests {
		testName := test.input
		t.Run(
			testName, func(t *testing.T) {
				pa := splitProto(test.input)
				if pa.Protocol != test.proto {
					t.Errorf(
						"Testcase %d: Incorrect proto: expected %v, got %v",
						i, test.proto, pa.Protocol,
					)
				}
				if pa.Address != test.addr {
					t.Errorf(
						"Testcase %d: Incorrect addr: expected %v, got %v",
						i, test.addr, pa.Address,
					)
				}
			},
		)
	}
}