frostfs-node/pkg/network/tls_test.go

package network

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestAddress_TLSEnabled(t *testing.T) {
	testCases := [...]struct {
		input   string
		wantTLS bool
	}{
		{"/dns4/localhost/tcp/8080", false},
		{"/dns4/localhost/tcp/8080/tls", true},
		{"/tls/dns4/localhost/tcp/8080", true},
		{"grpc://localhost:8080", false},
		{"grpcs://localhost:8080", true},
	}

	var addr Address

	for _, test := range testCases {
		err := addr.FromString(test.input)
		require.NoError(t, err)

		require.Equal(t, test.wantTLS, addr.IsTLSEnabled(), test.input)
	}
}

func BenchmarkAddressTLSEnabled(b *testing.B) {
	var addr Address
	err := addr.FromString("/dns4/localhost/tcp/8080/tls")
	require.NoError(b, err)

	b.ResetTimer()
	b.ReportAllocs()

	var enabled bool
	for range b.N {
		enabled = addr.IsTLSEnabled()
	}
	require.True(b, enabled)
}