|
@@ -1,525 +0,0 @@
|
|
|
-package nat
|
|
|
-
|
|
|
-import (
|
|
|
- "testing"
|
|
|
-)
|
|
|
-
|
|
|
-func TestParsePort(t *testing.T) {
|
|
|
- var (
|
|
|
- p int
|
|
|
- err error
|
|
|
- )
|
|
|
-
|
|
|
- p, err = ParsePort("1234")
|
|
|
-
|
|
|
- if err != nil || p != 1234 {
|
|
|
- t.Fatal("Parsing '1234' did not succeed")
|
|
|
- }
|
|
|
-
|
|
|
- // FIXME currently this is a valid port. I don't think it should be.
|
|
|
- // I'm leaving this test commented out until we make a decision.
|
|
|
- // - erikh
|
|
|
-
|
|
|
- /*
|
|
|
- p, err = ParsePort("0123")
|
|
|
-
|
|
|
- if err != nil {
|
|
|
- t.Fatal("Successfully parsed port '0123' to '123'")
|
|
|
- }
|
|
|
- */
|
|
|
-
|
|
|
- p, err = ParsePort("asdf")
|
|
|
-
|
|
|
- if err == nil || p != 0 {
|
|
|
- t.Fatal("Parsing port 'asdf' succeeded")
|
|
|
- }
|
|
|
-
|
|
|
- p, err = ParsePort("1asdf")
|
|
|
-
|
|
|
- if err == nil || p != 0 {
|
|
|
- t.Fatal("Parsing port '1asdf' succeeded")
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func TestParsePortRange(t *testing.T) {
|
|
|
- var (
|
|
|
- begin int
|
|
|
- end int
|
|
|
- err error
|
|
|
- )
|
|
|
-
|
|
|
- type TestRange struct {
|
|
|
- Range string
|
|
|
- Begin int
|
|
|
- End int
|
|
|
- }
|
|
|
- validRanges := []TestRange{
|
|
|
- {"1234", 1234, 1234},
|
|
|
- {"1234-1234", 1234, 1234},
|
|
|
- {"1234-1235", 1234, 1235},
|
|
|
- {"8000-9000", 8000, 9000},
|
|
|
- {"0", 0, 0},
|
|
|
- {"0-0", 0, 0},
|
|
|
- }
|
|
|
-
|
|
|
- for _, r := range validRanges {
|
|
|
- begin, end, err = ParsePortRange(r.Range)
|
|
|
-
|
|
|
- if err != nil || begin != r.Begin {
|
|
|
- t.Fatalf("Parsing port range '%s' did not succeed. Expected begin %d, got %d", r.Range, r.Begin, begin)
|
|
|
- }
|
|
|
- if err != nil || end != r.End {
|
|
|
- t.Fatalf("Parsing port range '%s' did not succeed. Expected end %d, got %d", r.Range, r.End, end)
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- invalidRanges := []string{
|
|
|
- "asdf",
|
|
|
- "1asdf",
|
|
|
- "9000-8000",
|
|
|
- "9000-",
|
|
|
- "-8000",
|
|
|
- "-8000-",
|
|
|
- }
|
|
|
-
|
|
|
- for _, r := range invalidRanges {
|
|
|
- begin, end, err = ParsePortRange(r)
|
|
|
-
|
|
|
- if err == nil || begin != 0 || end != 0 {
|
|
|
- t.Fatalf("Parsing port range '%s' succeeded", r)
|
|
|
- }
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func TestPort(t *testing.T) {
|
|
|
- p, err := NewPort("tcp", "1234")
|
|
|
-
|
|
|
- if err != nil {
|
|
|
- t.Fatalf("tcp, 1234 had a parsing issue: %v", err)
|
|
|
- }
|
|
|
-
|
|
|
- if string(p) != "1234/tcp" {
|
|
|
- t.Fatal("tcp, 1234 did not result in the string 1234/tcp")
|
|
|
- }
|
|
|
-
|
|
|
- if p.Proto() != "tcp" {
|
|
|
- t.Fatal("protocol was not tcp")
|
|
|
- }
|
|
|
-
|
|
|
- if p.Port() != "1234" {
|
|
|
- t.Fatal("port string value was not 1234")
|
|
|
- }
|
|
|
-
|
|
|
- if p.Int() != 1234 {
|
|
|
- t.Fatal("port int value was not 1234")
|
|
|
- }
|
|
|
-
|
|
|
- p, err = NewPort("tcp", "asd1234")
|
|
|
- if err == nil {
|
|
|
- t.Fatal("tcp, asd1234 was supposed to fail")
|
|
|
- }
|
|
|
-
|
|
|
- p, err = NewPort("tcp", "1234-1230")
|
|
|
- if err == nil {
|
|
|
- t.Fatal("tcp, 1234-1230 was supposed to fail")
|
|
|
- }
|
|
|
-
|
|
|
- p, err = NewPort("tcp", "1234-1242")
|
|
|
- if err != nil {
|
|
|
- t.Fatalf("tcp, 1234-1242 had a parsing issue: %v", err)
|
|
|
- }
|
|
|
-
|
|
|
- if string(p) != "1234-1242/tcp" {
|
|
|
- t.Fatal("tcp, 1234-1242 did not result in the string 1234-1242/tcp")
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func TestSplitProtoPort(t *testing.T) {
|
|
|
- var (
|
|
|
- proto string
|
|
|
- port string
|
|
|
- )
|
|
|
-
|
|
|
- proto, port = SplitProtoPort("1234/tcp")
|
|
|
-
|
|
|
- if proto != "tcp" || port != "1234" {
|
|
|
- t.Fatal("Could not split 1234/tcp properly")
|
|
|
- }
|
|
|
-
|
|
|
- proto, port = SplitProtoPort("")
|
|
|
-
|
|
|
- if proto != "" || port != "" {
|
|
|
- t.Fatal("parsing an empty string yielded surprising results", proto, port)
|
|
|
- }
|
|
|
-
|
|
|
- proto, port = SplitProtoPort("1234")
|
|
|
-
|
|
|
- if proto != "tcp" || port != "1234" {
|
|
|
- t.Fatal("tcp is not the default protocol for portspec '1234'", proto, port)
|
|
|
- }
|
|
|
-
|
|
|
- proto, port = SplitProtoPort("1234/")
|
|
|
-
|
|
|
- if proto != "tcp" || port != "1234" {
|
|
|
- t.Fatal("parsing '1234/' yielded:" + port + "/" + proto)
|
|
|
- }
|
|
|
-
|
|
|
- proto, port = SplitProtoPort("/tcp")
|
|
|
-
|
|
|
- if proto != "" || port != "" {
|
|
|
- t.Fatal("parsing '/tcp' yielded:" + port + "/" + proto)
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func TestParsePortSpecs(t *testing.T) {
|
|
|
- var (
|
|
|
- portMap map[Port]struct{}
|
|
|
- bindingMap map[Port][]PortBinding
|
|
|
- err error
|
|
|
- )
|
|
|
-
|
|
|
- portMap, bindingMap, err = ParsePortSpecs([]string{"1234/tcp", "2345/udp"})
|
|
|
-
|
|
|
- if err != nil {
|
|
|
- t.Fatalf("Error while processing ParsePortSpecs: %s", err)
|
|
|
- }
|
|
|
-
|
|
|
- if _, ok := portMap[Port("1234/tcp")]; !ok {
|
|
|
- t.Fatal("1234/tcp was not parsed properly")
|
|
|
- }
|
|
|
-
|
|
|
- if _, ok := portMap[Port("2345/udp")]; !ok {
|
|
|
- t.Fatal("2345/udp was not parsed properly")
|
|
|
- }
|
|
|
-
|
|
|
- for portspec, bindings := range bindingMap {
|
|
|
- if len(bindings) != 1 {
|
|
|
- t.Fatalf("%s should have exactly one binding", portspec)
|
|
|
- }
|
|
|
-
|
|
|
- if bindings[0].HostIP != "" {
|
|
|
- t.Fatalf("HostIP should not be set for %s", portspec)
|
|
|
- }
|
|
|
-
|
|
|
- if bindings[0].HostPort != "" {
|
|
|
- t.Fatalf("HostPort should not be set for %s", portspec)
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- portMap, bindingMap, err = ParsePortSpecs([]string{"1234:1234/tcp", "2345:2345/udp"})
|
|
|
-
|
|
|
- if err != nil {
|
|
|
- t.Fatalf("Error while processing ParsePortSpecs: %s", err)
|
|
|
- }
|
|
|
-
|
|
|
- if _, ok := portMap[Port("1234/tcp")]; !ok {
|
|
|
- t.Fatal("1234/tcp was not parsed properly")
|
|
|
- }
|
|
|
-
|
|
|
- if _, ok := portMap[Port("2345/udp")]; !ok {
|
|
|
- t.Fatal("2345/udp was not parsed properly")
|
|
|
- }
|
|
|
-
|
|
|
- for portspec, bindings := range bindingMap {
|
|
|
- _, port := SplitProtoPort(string(portspec))
|
|
|
-
|
|
|
- if len(bindings) != 1 {
|
|
|
- t.Fatalf("%s should have exactly one binding", portspec)
|
|
|
- }
|
|
|
-
|
|
|
- if bindings[0].HostIP != "" {
|
|
|
- t.Fatalf("HostIP should not be set for %s", portspec)
|
|
|
- }
|
|
|
-
|
|
|
- if bindings[0].HostPort != port {
|
|
|
- t.Fatalf("HostPort should be %s for %s", port, portspec)
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- portMap, bindingMap, err = ParsePortSpecs([]string{"0.0.0.0:1234:1234/tcp", "0.0.0.0:2345:2345/udp"})
|
|
|
-
|
|
|
- if err != nil {
|
|
|
- t.Fatalf("Error while processing ParsePortSpecs: %s", err)
|
|
|
- }
|
|
|
-
|
|
|
- if _, ok := portMap[Port("1234/tcp")]; !ok {
|
|
|
- t.Fatal("1234/tcp was not parsed properly")
|
|
|
- }
|
|
|
-
|
|
|
- if _, ok := portMap[Port("2345/udp")]; !ok {
|
|
|
- t.Fatal("2345/udp was not parsed properly")
|
|
|
- }
|
|
|
-
|
|
|
- for portspec, bindings := range bindingMap {
|
|
|
- _, port := SplitProtoPort(string(portspec))
|
|
|
-
|
|
|
- if len(bindings) != 1 {
|
|
|
- t.Fatalf("%s should have exactly one binding", portspec)
|
|
|
- }
|
|
|
-
|
|
|
- if bindings[0].HostIP != "0.0.0.0" {
|
|
|
- t.Fatalf("HostIP is not 0.0.0.0 for %s", portspec)
|
|
|
- }
|
|
|
-
|
|
|
- if bindings[0].HostPort != port {
|
|
|
- t.Fatalf("HostPort should be %s for %s", port, portspec)
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- _, _, err = ParsePortSpecs([]string{"localhost:1234:1234/tcp"})
|
|
|
-
|
|
|
- if err == nil {
|
|
|
- t.Fatal("Received no error while trying to parse a hostname instead of ip")
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func TestParsePortSpecsWithRange(t *testing.T) {
|
|
|
- var (
|
|
|
- portMap map[Port]struct{}
|
|
|
- bindingMap map[Port][]PortBinding
|
|
|
- err error
|
|
|
- )
|
|
|
-
|
|
|
- portMap, bindingMap, err = ParsePortSpecs([]string{"1234-1236/tcp", "2345-2347/udp"})
|
|
|
-
|
|
|
- if err != nil {
|
|
|
- t.Fatalf("Error while processing ParsePortSpecs: %s", err)
|
|
|
- }
|
|
|
-
|
|
|
- if _, ok := portMap[Port("1235/tcp")]; !ok {
|
|
|
- t.Fatal("1234/tcp was not parsed properly")
|
|
|
- }
|
|
|
-
|
|
|
- if _, ok := portMap[Port("2346/udp")]; !ok {
|
|
|
- t.Fatal("2345/udp was not parsed properly")
|
|
|
- }
|
|
|
-
|
|
|
- for portspec, bindings := range bindingMap {
|
|
|
- if len(bindings) != 1 {
|
|
|
- t.Fatalf("%s should have exactly one binding", portspec)
|
|
|
- }
|
|
|
-
|
|
|
- if bindings[0].HostIP != "" {
|
|
|
- t.Fatalf("HostIP should not be set for %s", portspec)
|
|
|
- }
|
|
|
-
|
|
|
- if bindings[0].HostPort != "" {
|
|
|
- t.Fatalf("HostPort should not be set for %s", portspec)
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- portMap, bindingMap, err = ParsePortSpecs([]string{"1234-1236:1234-1236/tcp", "2345-2347:2345-2347/udp"})
|
|
|
-
|
|
|
- if err != nil {
|
|
|
- t.Fatalf("Error while processing ParsePortSpecs: %s", err)
|
|
|
- }
|
|
|
-
|
|
|
- if _, ok := portMap[Port("1235/tcp")]; !ok {
|
|
|
- t.Fatal("1234/tcp was not parsed properly")
|
|
|
- }
|
|
|
-
|
|
|
- if _, ok := portMap[Port("2346/udp")]; !ok {
|
|
|
- t.Fatal("2345/udp was not parsed properly")
|
|
|
- }
|
|
|
-
|
|
|
- for portspec, bindings := range bindingMap {
|
|
|
- _, port := SplitProtoPort(string(portspec))
|
|
|
- if len(bindings) != 1 {
|
|
|
- t.Fatalf("%s should have exactly one binding", portspec)
|
|
|
- }
|
|
|
-
|
|
|
- if bindings[0].HostIP != "" {
|
|
|
- t.Fatalf("HostIP should not be set for %s", portspec)
|
|
|
- }
|
|
|
-
|
|
|
- if bindings[0].HostPort != port {
|
|
|
- t.Fatalf("HostPort should be %s for %s", port, portspec)
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- portMap, bindingMap, err = ParsePortSpecs([]string{"0.0.0.0:1234-1236:1234-1236/tcp", "0.0.0.0:2345-2347:2345-2347/udp"})
|
|
|
-
|
|
|
- if err != nil {
|
|
|
- t.Fatalf("Error while processing ParsePortSpecs: %s", err)
|
|
|
- }
|
|
|
-
|
|
|
- if _, ok := portMap[Port("1235/tcp")]; !ok {
|
|
|
- t.Fatal("1234/tcp was not parsed properly")
|
|
|
- }
|
|
|
-
|
|
|
- if _, ok := portMap[Port("2346/udp")]; !ok {
|
|
|
- t.Fatal("2345/udp was not parsed properly")
|
|
|
- }
|
|
|
-
|
|
|
- for portspec, bindings := range bindingMap {
|
|
|
- _, port := SplitProtoPort(string(portspec))
|
|
|
- if len(bindings) != 1 || bindings[0].HostIP != "0.0.0.0" || bindings[0].HostPort != port {
|
|
|
- t.Fatalf("Expect single binding to port %s but found %s", port, bindings)
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- _, _, err = ParsePortSpecs([]string{"localhost:1234-1236:1234-1236/tcp"})
|
|
|
-
|
|
|
- if err == nil {
|
|
|
- t.Fatal("Received no error while trying to parse a hostname instead of ip")
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func TestParseNetworkOptsPrivateOnly(t *testing.T) {
|
|
|
- ports, bindings, err := ParsePortSpecs([]string{"192.168.1.100::80"})
|
|
|
- if err != nil {
|
|
|
- t.Fatal(err)
|
|
|
- }
|
|
|
- if len(ports) != 1 {
|
|
|
- t.Logf("Expected 1 got %d", len(ports))
|
|
|
- t.FailNow()
|
|
|
- }
|
|
|
- if len(bindings) != 1 {
|
|
|
- t.Logf("Expected 1 got %d", len(bindings))
|
|
|
- t.FailNow()
|
|
|
- }
|
|
|
- for k := range ports {
|
|
|
- if k.Proto() != "tcp" {
|
|
|
- t.Logf("Expected tcp got %s", k.Proto())
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- if k.Port() != "80" {
|
|
|
- t.Logf("Expected 80 got %s", k.Port())
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- b, exists := bindings[k]
|
|
|
- if !exists {
|
|
|
- t.Log("Binding does not exist")
|
|
|
- t.FailNow()
|
|
|
- }
|
|
|
- if len(b) != 1 {
|
|
|
- t.Logf("Expected 1 got %d", len(b))
|
|
|
- t.FailNow()
|
|
|
- }
|
|
|
- s := b[0]
|
|
|
- if s.HostPort != "" {
|
|
|
- t.Logf("Expected \"\" got %s", s.HostPort)
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- if s.HostIP != "192.168.1.100" {
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func TestParseNetworkOptsPublic(t *testing.T) {
|
|
|
- ports, bindings, err := ParsePortSpecs([]string{"192.168.1.100:8080:80"})
|
|
|
- if err != nil {
|
|
|
- t.Fatal(err)
|
|
|
- }
|
|
|
- if len(ports) != 1 {
|
|
|
- t.Logf("Expected 1 got %d", len(ports))
|
|
|
- t.FailNow()
|
|
|
- }
|
|
|
- if len(bindings) != 1 {
|
|
|
- t.Logf("Expected 1 got %d", len(bindings))
|
|
|
- t.FailNow()
|
|
|
- }
|
|
|
- for k := range ports {
|
|
|
- if k.Proto() != "tcp" {
|
|
|
- t.Logf("Expected tcp got %s", k.Proto())
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- if k.Port() != "80" {
|
|
|
- t.Logf("Expected 80 got %s", k.Port())
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- b, exists := bindings[k]
|
|
|
- if !exists {
|
|
|
- t.Log("Binding does not exist")
|
|
|
- t.FailNow()
|
|
|
- }
|
|
|
- if len(b) != 1 {
|
|
|
- t.Logf("Expected 1 got %d", len(b))
|
|
|
- t.FailNow()
|
|
|
- }
|
|
|
- s := b[0]
|
|
|
- if s.HostPort != "8080" {
|
|
|
- t.Logf("Expected 8080 got %s", s.HostPort)
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- if s.HostIP != "192.168.1.100" {
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func TestParseNetworkOptsPublicNoPort(t *testing.T) {
|
|
|
- ports, bindings, err := ParsePortSpecs([]string{"192.168.1.100"})
|
|
|
-
|
|
|
- if err == nil {
|
|
|
- t.Logf("Expected error Invalid containerPort")
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- if ports != nil {
|
|
|
- t.Logf("Expected nil got %s", ports)
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- if bindings != nil {
|
|
|
- t.Logf("Expected nil got %s", bindings)
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func TestParseNetworkOptsNegativePorts(t *testing.T) {
|
|
|
- ports, bindings, err := ParsePortSpecs([]string{"192.168.1.100:-1:-1"})
|
|
|
-
|
|
|
- if err == nil {
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- if len(ports) != 0 {
|
|
|
- t.Logf("Expected nil got %d", len(ports))
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- if len(bindings) != 0 {
|
|
|
- t.Logf("Expected 0 got %d", len(bindings))
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func TestParseNetworkOptsUdp(t *testing.T) {
|
|
|
- ports, bindings, err := ParsePortSpecs([]string{"192.168.1.100::6000/udp"})
|
|
|
- if err != nil {
|
|
|
- t.Fatal(err)
|
|
|
- }
|
|
|
- if len(ports) != 1 {
|
|
|
- t.Logf("Expected 1 got %d", len(ports))
|
|
|
- t.FailNow()
|
|
|
- }
|
|
|
- if len(bindings) != 1 {
|
|
|
- t.Logf("Expected 1 got %d", len(bindings))
|
|
|
- t.FailNow()
|
|
|
- }
|
|
|
- for k := range ports {
|
|
|
- if k.Proto() != "udp" {
|
|
|
- t.Logf("Expected udp got %s", k.Proto())
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- if k.Port() != "6000" {
|
|
|
- t.Logf("Expected 6000 got %s", k.Port())
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- b, exists := bindings[k]
|
|
|
- if !exists {
|
|
|
- t.Log("Binding does not exist")
|
|
|
- t.FailNow()
|
|
|
- }
|
|
|
- if len(b) != 1 {
|
|
|
- t.Logf("Expected 1 got %d", len(b))
|
|
|
- t.FailNow()
|
|
|
- }
|
|
|
- s := b[0]
|
|
|
- if s.HostPort != "" {
|
|
|
- t.Logf("Expected \"\" got %s", s.HostPort)
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- if s.HostIP != "192.168.1.100" {
|
|
|
- t.Fail()
|
|
|
- }
|
|
|
- }
|
|
|
-}
|