utils_test.go 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. package ipamutils
  2. import (
  3. "net"
  4. "testing"
  5. "gotest.tools/v3/assert"
  6. is "gotest.tools/v3/assert/cmp"
  7. )
  8. func initBroadPredefinedNetworks() []*net.IPNet {
  9. pl := make([]*net.IPNet, 0, 31)
  10. mask := []byte{255, 255, 0, 0}
  11. for i := 17; i < 32; i++ {
  12. pl = append(pl, &net.IPNet{IP: []byte{172, byte(i), 0, 0}, Mask: mask})
  13. }
  14. mask20 := []byte{255, 255, 240, 0}
  15. for i := 0; i < 16; i++ {
  16. pl = append(pl, &net.IPNet{IP: []byte{192, 168, byte(i << 4), 0}, Mask: mask20})
  17. }
  18. return pl
  19. }
  20. func initGranularPredefinedNetworks() []*net.IPNet {
  21. pl := make([]*net.IPNet, 0, 256*256)
  22. mask := []byte{255, 255, 255, 0}
  23. for i := 0; i < 256; i++ {
  24. for j := 0; j < 256; j++ {
  25. pl = append(pl, &net.IPNet{IP: []byte{10, byte(i), byte(j), 0}, Mask: mask})
  26. }
  27. }
  28. return pl
  29. }
  30. func initGlobalScopeNetworks() []*net.IPNet {
  31. pl := make([]*net.IPNet, 0, 256*256)
  32. mask := []byte{255, 255, 255, 0}
  33. for i := 0; i < 256; i++ {
  34. for j := 0; j < 256; j++ {
  35. pl = append(pl, &net.IPNet{IP: []byte{30, byte(i), byte(j), 0}, Mask: mask})
  36. }
  37. }
  38. return pl
  39. }
  40. func TestDefaultNetwork(t *testing.T) {
  41. for _, nw := range GetGlobalScopeDefaultNetworks() {
  42. if ones, bits := nw.Mask.Size(); bits != 32 || ones != 24 {
  43. t.Fatalf("Unexpected size for network in granular list: %v", nw)
  44. }
  45. }
  46. for _, nw := range GetLocalScopeDefaultNetworks() {
  47. if ones, bits := nw.Mask.Size(); bits != 32 || (ones != 20 && ones != 16) {
  48. t.Fatalf("Unexpected size for network in broad list: %v", nw)
  49. }
  50. }
  51. originalBroadNets := initBroadPredefinedNetworks()
  52. m := make(map[string]bool)
  53. for _, v := range originalBroadNets {
  54. m[v.String()] = true
  55. }
  56. for _, nw := range GetLocalScopeDefaultNetworks() {
  57. _, ok := m[nw.String()]
  58. assert.Check(t, ok)
  59. delete(m, nw.String())
  60. }
  61. assert.Check(t, is.Len(m, 0))
  62. originalGranularNets := initGranularPredefinedNetworks()
  63. m = make(map[string]bool)
  64. for _, v := range originalGranularNets {
  65. m[v.String()] = true
  66. }
  67. for _, nw := range GetGlobalScopeDefaultNetworks() {
  68. _, ok := m[nw.String()]
  69. assert.Check(t, ok)
  70. delete(m, nw.String())
  71. }
  72. assert.Check(t, is.Len(m, 0))
  73. }
  74. func TestConfigGlobalScopeDefaultNetworks(t *testing.T) {
  75. err := ConfigGlobalScopeDefaultNetworks([]*NetworkToSplit{{"30.0.0.0/8", 24}})
  76. assert.NilError(t, err)
  77. originalGlobalScopeNetworks := initGlobalScopeNetworks()
  78. m := make(map[string]bool)
  79. for _, v := range originalGlobalScopeNetworks {
  80. m[v.String()] = true
  81. }
  82. for _, nw := range GetGlobalScopeDefaultNetworks() {
  83. _, ok := m[nw.String()]
  84. assert.Check(t, ok)
  85. delete(m, nw.String())
  86. }
  87. assert.Check(t, is.Len(m, 0))
  88. }