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}, } for _, test := range testCases { addr := Address{ ma: buildMultiaddr(test.input, t), } require.Equal(t, test.wantTLS, addr.TLSEnabled(), test.input) } } func TestAddress_AddTLS(t *testing.T) { input, tls := "/dns4/localhost/tcp/8080", tls.String() testCases := [...]struct { input string want string }{ {input, input + tls}, {input + tls, input + tls}, } for _, test := range testCases { addr := Address{ ma: buildMultiaddr(test.input, t), } addr.AddTLS() var netAddr Address err := netAddr.FromString(test.want) require.NoError(t, err) require.True(t, netAddr.equal(addr), test.input) } }