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)
	}
}