Browse code

Move test case TestGenerateGateway to common_test.go

xiangpengzhao authored on 2016/11/10 16:15:30
Showing 2 changed files
... ...
@@ -20,3 +20,23 @@ func TestConversion(t *testing.T) {
20 20
 		t.Fatal("Conversion back and forth failed")
21 21
 	}
22 22
 }
23
+
24
+func TestGenerateGateway(t *testing.T) {
25
+	sna, err := NewSubnetAllocator("10.1.0.0/16", 8, nil)
26
+	if err != nil {
27
+		t.Fatal("Failed to initialize IP allocator: ", err)
28
+	}
29
+
30
+	sn, err := sna.GetNetwork()
31
+	if err != nil {
32
+		t.Fatal("Failed to get network: ", err)
33
+	}
34
+	if sn.String() != "10.1.0.0/24" {
35
+		t.Fatalf("Did not get expected subnet (sn=%s)", sn.String())
36
+	}
37
+
38
+	gatewayIP := GenerateDefaultGateway(sn)
39
+	if gatewayIP.String() != "10.1.0.1" {
40
+		t.Fatalf("Did not get expected gateway IP Address (gatewayIP=%s)", gatewayIP.String())
41
+	}
42
+}
... ...
@@ -248,23 +248,3 @@ func TestAllocateReleaseSubnet(t *testing.T) {
248 248
 		t.Fatalf("Unexpectedly succeeded in getting network (sn=%s)", sn.String())
249 249
 	}
250 250
 }
251
-
252
-func TestGenerateGateway(t *testing.T) {
253
-	sna, err := NewSubnetAllocator("10.1.0.0/16", 8, nil)
254
-	if err != nil {
255
-		t.Fatal("Failed to initialize IP allocator: ", err)
256
-	}
257
-
258
-	sn, err := sna.GetNetwork()
259
-	if err != nil {
260
-		t.Fatal("Failed to get network: ", err)
261
-	}
262
-	if sn.String() != "10.1.0.0/24" {
263
-		t.Fatalf("Did not get expected subnet (sn=%s)", sn.String())
264
-	}
265
-
266
-	gatewayIP := GenerateDefaultGateway(sn)
267
-	if gatewayIP.String() != "10.1.0.1" {
268
-		t.Fatalf("Did not get expected gateway IP Address (gatewayIP=%s)", gatewayIP.String())
269
-	}
270
-}