mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-12-18 14:36:30 +00:00
27 lines
658 B
Go
27 lines
658 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")
|
|
b := p.Bytes()
|
|
|
|
expected := header + "command -v socat >/dev/null && socat TCP-LISTEN:8080,fork TCP:172.1.4.5:8080 &\n"
|
|
if string(b) != expected {
|
|
t.Errorf("Invalid proxy got:\n%s\nwant:\n%s", string(b), expected)
|
|
}
|
|
|
|
// test creating a proxy script when there
|
|
// are no proxy addresses added to the map
|
|
p = Proxy{}
|
|
b = p.Bytes()
|
|
if string(b) != header {
|
|
t.Errorf("Invalid empty proxy file. Expected\n%s", header)
|
|
}
|
|
}
|