Переглянути джерело

added test, gofmtd

Signed-off-by: shuai-z <zs.broccoli@gmail.com>
shuai-z 10 роки тому
батько
коміт
2c2edabca5

+ 1 - 1
daemon/networkdriver/portallocator/portallocator.go

@@ -14,7 +14,7 @@ type portMap struct {
 
 func newPortMap() *portMap {
 	return &portMap{
-		p: map[int]struct{}{},
+		p:    map[int]struct{}{},
 		last: EndPortRange,
 	}
 }

+ 16 - 0
daemon/networkdriver/portallocator/portallocator_test.go

@@ -214,3 +214,19 @@ func TestPortAllocation(t *testing.T) {
 		t.Fatal("Requesting a dynamic port should never allocate a used port")
 	}
 }
+
+func TestNoDuplicateBPR(t *testing.T) {
+	defer reset()
+
+	if port, err := RequestPort(defaultIP, "tcp", BeginPortRange); err != nil {
+		t.Fatal(err)
+	} else if port != BeginPortRange {
+		t.Fatalf("Expected port %d got %d", BeginPortRange, port)
+	}
+
+	if port, err := RequestPort(defaultIP, "tcp", 0); err != nil {
+		t.Fatal(err)
+	} else if port == BeginPortRange {
+		t.Fatalf("Acquire(0) allocated the same port twice: %d", port)
+	}
+}