| ... | ... |
@@ -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 |
-} |