harness-drone/pkg/build/proxy/proxy_test.go
2014-02-07 03:10:01 -07:00

32 lines
741 B
Go

package proxy
import (
"testing"
)
func TestProxy(t *testing.T) {
// test creating a proxy with a few different
// addresses, and our ability to create the
// proxy shell script.
p := Proxy{}
p.Set("8080", "172.1.4.5")
p.Set("8000", "172.1.3.1")
b := p.Bytes()
expected := `#!/bin/bash
[ -x /usr/bin/socat ] && socat TCP-LISTEN:8080,fork TCP:172.1.4.5:8080 &
[ -x /usr/bin/socat ] && socat TCP-LISTEN:8000,fork TCP:172.1.3.1:8000 &
`
if string(b) != expected {
t.Errorf("Invalid proxy \n%s", expected)
}
// test creating a proxy script when there
// are no proxy addresses added to the map
p = Proxy{}
b = p.Bytes()
expected = "#!/bin/bash\n"
if string(b) != expected {
t.Errorf("Invalid proxy \n%s", expected)
}
}