116 lines
2.2 KiB
Go
116 lines
2.2 KiB
Go
// Copyright (C) 2017 Michał Matczuk
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestNormalizeAddress(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
tests := []struct {
|
|
addr string
|
|
expected string
|
|
error string
|
|
}{
|
|
{
|
|
addr: "22",
|
|
expected: "127.0.0.1:22",
|
|
},
|
|
{
|
|
addr: ":22",
|
|
expected: "127.0.0.1:22",
|
|
},
|
|
{
|
|
addr: "0.0.0.0:22",
|
|
expected: "0.0.0.0:22",
|
|
},
|
|
{
|
|
addr: "0.0.0.0",
|
|
error: "missing port",
|
|
},
|
|
{
|
|
addr: "",
|
|
error: "missing port",
|
|
},
|
|
}
|
|
|
|
for i, tt := range tests {
|
|
actual, err := normalizeAddress(tt.addr)
|
|
if actual != tt.expected {
|
|
t.Errorf("[%d] expected %q got %q err: %s", i, tt.expected, actual, err)
|
|
}
|
|
if tt.error != "" && err == nil {
|
|
t.Errorf("[%d] expected error", i)
|
|
}
|
|
if err != nil && (tt.error == "" || !strings.Contains(err.Error(), tt.error)) {
|
|
t.Errorf("[%d] expected error contains %q, got %q", i, tt.error, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestNormalizeURL(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
tests := []struct {
|
|
rawurl string
|
|
expected string
|
|
error string
|
|
}{
|
|
{
|
|
rawurl: "localhost",
|
|
expected: "http://localhost",
|
|
},
|
|
{
|
|
rawurl: "localhost:80",
|
|
expected: "http://localhost:80",
|
|
},
|
|
{
|
|
rawurl: "localhost:80/path/",
|
|
expected: "http://localhost:80/path/",
|
|
},
|
|
{
|
|
rawurl: "localhost:80/path",
|
|
error: "/",
|
|
},
|
|
{
|
|
rawurl: "https://localhost",
|
|
expected: "https://localhost",
|
|
},
|
|
{
|
|
rawurl: "https://localhost:443",
|
|
expected: "https://localhost:443",
|
|
},
|
|
{
|
|
rawurl: "https://localhost:443/path/",
|
|
expected: "https://localhost:443/path/",
|
|
},
|
|
{
|
|
rawurl: "https://localhost:443/path",
|
|
error: "/",
|
|
},
|
|
{
|
|
rawurl: "ftp://localhost",
|
|
error: "unsupported url schema",
|
|
},
|
|
}
|
|
|
|
for i, tt := range tests {
|
|
actual, err := normalizeURL(tt.rawurl)
|
|
if actual != tt.expected {
|
|
t.Errorf("[%d] expected %q got %q, err: %s", i, tt.expected, actual, err)
|
|
}
|
|
if tt.error != "" && err == nil {
|
|
t.Errorf("[%d] expected error", i)
|
|
}
|
|
if err != nil && (tt.error == "" || !strings.Contains(err.Error(), tt.error)) {
|
|
t.Errorf("[%d] expected error contains %q, got %q", i, tt.error, err)
|
|
}
|
|
}
|
|
|
|
}
|