Test nextDynamicPort method

Author: everesio

proxy/proxy.go

@@ -156,6 +156,9 @@ func (p *Listeners) findListenerConfig(brokerId int32) *ListenerConfig {
 // Make sure all dynamically allocated ports are within the half open interval
 // [dynamicSequentialMinPort, dynamicSequentialMinPort + dynamicSequentialMaxPorts).
 func (p *Listeners) nextDynamicPort(portOffset uint64, brokerAddress string, brokerId int32) (uint16, error) {
+	if p.dynamicSequentialMaxPorts == 0 {
+		return 0, fmt.Errorf("dynamic sequential max ports is 0")
+	}
 	port := p.dynamicSequentialMinPort + uint16(portOffset%uint64(p.dynamicSequentialMaxPorts))
 	if port < p.dynamicSequentialMinPort {
 		return 0, fmt.Errorf("port assignment overflow %s %d: %d", brokerAddress, brokerId, port)
@@ -198,7 +201,7 @@ func (p *Listeners) ListenDynamicInstance(brokerAddress string, brokerId int32)
 			listenerAddress = net.JoinHostPort(p.defaultListenerIP, strconv.Itoa(0))
 		} else {
 			// Use sequentially allocated port.
-			port, err := p.nextDynamicPort(uint64(p.currentDynamicPortCounter), brokerAddress, brokerId)
+			port, err := p.nextDynamicPort(p.currentDynamicPortCounter, brokerAddress, brokerId)
 			if err != nil {
 				return "", 0, err
 			}

proxy/proxy_test.go

@@ -415,3 +415,70 @@ func TestGetDynamicAdvertisedAddress(t *testing.T) {
 		})
 	}
 }
+
+func TestNextDynamicPort(t *testing.T) {
+	tests := []struct {
+		name             string
+		portOffset       uint64
+		minPort          uint16
+		maxPorts         uint16
+		expectedNextPort uint16
+		expectError      bool
+	}{
+		{
+			name:        "should not happen (checked in config) - max ports is 0",
+			expectError: true,
+		},
+		{
+			name:             "broker 0",
+			portOffset:       0,
+			minPort:          30000,
+			maxPorts:         1,
+			expectedNextPort: 30000,
+		},
+		{
+			name:             "broker 999",
+			portOffset:       999,
+			minPort:          30000,
+			maxPorts:         1000,
+			expectedNextPort: 30999,
+		},
+		{
+			name:             "wrap-around : maxPorts 1, portOffsets 0 and 1 produce the same port 30000, listen will fail with bind: address already in use",
+			portOffset:       1,
+			minPort:          30000,
+			maxPorts:         1,
+			expectedNextPort: 30000,
+		},
+		{
+			name:             "wrap-around : broker 1000",
+			portOffset:       1000,
+			minPort:          30000,
+			maxPorts:         1000,
+			expectedNextPort: 30000,
+		},
+		{
+			name:        "port assignment overflow",
+			portOffset:  15,
+			minPort:     65530,
+			maxPorts:    100,
+			expectError: true,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			listeners := &Listeners{
+				dynamicSequentialMinPort:  tt.minPort,
+				dynamicSequentialMaxPorts: tt.maxPorts,
+			}
+			nextPort, err := listeners.nextDynamicPort(tt.portOffset, "", int32(tt.portOffset))
+			if tt.expectError {
+				assert.Error(t, err)
+			} else {
+				assert.NoError(t, err)
+				assert.Equal(t, tt.expectedNextPort, nextPort)
+			}
+		})
+	}
+}