mirror of https://github.com/divan/expvarmon.git
122 lines
3.2 KiB
Go
122 lines
3.2 KiB
Go
package main
|
|
|
|
import "testing"
|
|
|
|
func TestUtils(t *testing.T) {
|
|
str := "memstats.Alloc,memstats.Sys"
|
|
|
|
vars, err := ParseVars(str)
|
|
if err != nil {
|
|
t.Fatalf("Err not nil: %v", err)
|
|
}
|
|
|
|
if len(vars) != 2 {
|
|
t.Fatalf("vars should contain 2 elements, but has %d", len(vars))
|
|
}
|
|
|
|
str = "memstats.Alloc,memstats.Sys,goroutines,Counter.A"
|
|
|
|
vars, err = ParseVars(str)
|
|
if err != nil {
|
|
t.Fatalf("Err not nil: %v", err)
|
|
}
|
|
|
|
if len(vars) != 4 {
|
|
t.Fatalf("vars should contain 4 elements, but has %d", len(vars))
|
|
}
|
|
}
|
|
|
|
func TestExtractUrlAndPorts(t *testing.T) {
|
|
var rawurl, ports string
|
|
rawurl, ports = extractURLAndPorts("40000-40002")
|
|
if rawurl != "http://localhost" || ports != "40000-40002" {
|
|
t.Fatalf("extract url and ports failed: %v, %v", rawurl, ports)
|
|
}
|
|
|
|
rawurl, ports = extractURLAndPorts("https://example.com:1234")
|
|
if rawurl != "https://example.com" || ports != "1234" {
|
|
t.Fatalf("extract url and ports failed: %v, %v", rawurl, ports)
|
|
}
|
|
|
|
rawurl, ports = extractURLAndPorts("http://user:passwd@example.com:1234-1256")
|
|
if rawurl != "http://user:passwd@example.com" || ports != "1234-1256" {
|
|
t.Fatalf("extract url and ports failed: %v, %v", rawurl, ports)
|
|
}
|
|
|
|
rawurl, ports = extractURLAndPorts("https://example.com:1234-1256/_endpoint")
|
|
if rawurl != "https://example.com/_endpoint" || ports != "1234-1256" {
|
|
t.Fatalf("extract url and ports failed: %v, %v", rawurl, ports)
|
|
}
|
|
}
|
|
|
|
func TestPorts(t *testing.T) {
|
|
arg := "1234,1235"
|
|
ports, err := ParsePorts(arg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(ports) != 2 || ports[0].Host != "localhost:1234" {
|
|
t.Fatalf("ParsePorts returns wrong data: %v", ports)
|
|
}
|
|
|
|
arg = "1234-1237,2000"
|
|
ports, err = ParsePorts(arg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(ports) != 5 || ports[0].Host != "localhost:1234" || ports[4].Host != "localhost:2000" {
|
|
t.Fatalf("ParsePorts returns wrong data: %v", ports)
|
|
}
|
|
|
|
arg = "40000-40002,localhost:2000-2002,remote:1234-1235,https://example.com:1234-1236"
|
|
ports, err = ParsePorts(arg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(ports) != 11 ||
|
|
ports[0].Host != "localhost:40000" ||
|
|
ports[3].Host != "localhost:2000" ||
|
|
ports[7].Host != "remote:1235" ||
|
|
ports[7].Path != "/debug/vars" ||
|
|
ports[10].Host != "example.com:1236" ||
|
|
ports[10].Scheme != "https" {
|
|
t.Fatalf("ParsePorts returns wrong data: %v", ports)
|
|
}
|
|
|
|
// Test Auth
|
|
arg = "http://user:pass@localhost:2000-2002"
|
|
ports, err = ParsePorts(arg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
pass, isSet := ports[0].User.Password()
|
|
if len(ports) != 3 ||
|
|
ports[0].User.Username() != "user" ||
|
|
pass != "pass" || !isSet ||
|
|
ports[0].Scheme != "http" {
|
|
t.Fatalf("ParsePorts returns wrong data: %v", ports)
|
|
}
|
|
|
|
arg = "localhost:2000-2002,remote:1234-1235,some:weird:1234-123input"
|
|
_, err = ParsePorts(arg)
|
|
if err == nil {
|
|
t.Fatalf("err shouldn't be nil")
|
|
}
|
|
|
|
arg = "string,sdasd"
|
|
_, err = ParsePorts(arg)
|
|
if err == nil {
|
|
t.Fatalf("err shouldn't be nil")
|
|
}
|
|
|
|
// Test endpoints
|
|
arg = "localhost:2000,https://example.com:1234/_custom_expvars"
|
|
ports, err = ParsePorts(arg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if ports[0].Path != "/debug/vars" || ports[1].Path != "/_custom_expvars" {
|
|
t.Fatalf("ParsePorts returns wrong data: %v", ports)
|
|
}
|
|
}
|