summaryrefslogtreecommitdiffstats
path: root/extras/hs-test/hst_suite.go
diff options
context:
space:
mode:
authorMaros Ondrejicka <mondreji@cisco.com>2023-02-21 10:53:20 +0100
committerFlorin Coras <florin.coras@gmail.com>2023-02-21 18:40:49 +0000
commit300f70d3a3b09ed5d9371cb2936e72e1c01d042a (patch)
tree45a807706b009f800e4a8c4b6edd65fec8d99792 /extras/hs-test/hst_suite.go
parent7c06b5790d879c216ad27f8e281789b35858db43 (diff)
hs-test: clean-up ip address generation
Type: test Signed-off-by: Maros Ondrejicka <mondreji@cisco.com> Change-Id: I74c505920d1363d0ff2b3213fd831c181b70a173
Diffstat (limited to 'extras/hs-test/hst_suite.go')
-rw-r--r--extras/hs-test/hst_suite.go58
1 files changed, 29 insertions, 29 deletions
diff --git a/extras/hs-test/hst_suite.go b/extras/hs-test/hst_suite.go
index 9cd9aea5f24..409ab1688c6 100644
--- a/extras/hs-test/hst_suite.go
+++ b/extras/hs-test/hst_suite.go
@@ -12,7 +12,7 @@ import (
)
const (
- defaultNamespaceName string = "default"
+ defaultNetworkNumber int = 1
)
func IsPersistent() bool {
@@ -173,7 +173,6 @@ func (s *HstSuite) loadContainerTopology(topologyName string) {
if err != nil {
s.T().Fatalf("container config error: %v", err)
}
- s.log(newContainer.getRunCommand())
s.containers[newContainer.name] = newContainer
}
}
@@ -250,47 +249,48 @@ func (s *HstSuite) unconfigureNetworkTopology() {
}
}
-type NamespaceAddresses struct {
- namespace string
+type NetworkAddresses struct {
+ network int
numberOfAddresses int
}
+type AddressCounter = int
+
type Addresser struct {
- namespaces []*NamespaceAddresses
- suite *HstSuite
+ networks map[int]AddressCounter
+ suite *HstSuite
}
-func (a *Addresser) AddNamespace(name string) {
- var newNamespace = &NamespaceAddresses{
- namespace: name,
- numberOfAddresses: 0,
- }
- a.namespaces = append(a.namespaces, newNamespace)
+func (a *Addresser) AddNetwork(networkNumber int) {
+ a.networks[networkNumber] = 1
}
-func (a *Addresser) NewIp4Address() (string, error) {
- return a.NewIp4AddressWithNamespace(defaultNamespaceName)
-}
+func (a *Addresser) NewIp4Address(inputNetworkNumber ...int) (string, error) {
+ var networkNumber int = 0
+ if len(inputNetworkNumber) > 0 {
+ networkNumber = inputNetworkNumber[0]
+ }
-func (a *Addresser) NewIp4AddressWithNamespace(namespace string) (string, error) {
- for i, val := range a.namespaces {
- if val.namespace != namespace {
- continue
- }
- if val.numberOfAddresses == 255 {
- return "", fmt.Errorf("no available IPv4 addresses")
- }
- address := fmt.Sprintf("10.10.%v.%v/24", i, val.numberOfAddresses+1)
- val.numberOfAddresses++
- return address, nil
+ if _, ok := a.networks[networkNumber]; !ok {
+ a.AddNetwork(networkNumber)
}
- a.AddNamespace(namespace)
- return a.NewIp4AddressWithNamespace(namespace)
+
+ numberOfAddresses := a.networks[networkNumber]
+
+ if numberOfAddresses == 254 {
+ return "", fmt.Errorf("no available IPv4 addresses")
+ }
+
+ address := fmt.Sprintf("10.10.%v.%v/24", networkNumber, numberOfAddresses)
+ a.networks[networkNumber] = numberOfAddresses + 1
+
+ return address, nil
}
func NewAddresser(suite *HstSuite) *Addresser {
var addresser = new(Addresser)
addresser.suite = suite
- addresser.AddNamespace(defaultNamespaceName)
+ addresser.networks = make(map[int]AddressCounter)
+ addresser.AddNetwork(0)
return addresser
}