extra test

Author: pjfanning

management-cluster-bootstrap/src/test/scala/org/apache/pekko/management/cluster/bootstrap/internal/BootstrapCoordinatorSpec.scala

@@ -14,22 +14,23 @@
 package org.apache.pekko.management.cluster.bootstrap.internal
 
 import java.util.concurrent.atomic.AtomicReference
+
 import org.apache.pekko
 import pekko.actor.{ ActorRef, ActorSystem, Props }
 import pekko.discovery.ServiceDiscovery.{ Resolved, ResolvedTarget }
 import pekko.discovery.{ Lookup, MockDiscovery }
 import pekko.http.scaladsl.model.Uri
-import com.typesafe.config.ConfigFactory
 import pekko.management.cluster.bootstrap.internal.BootstrapCoordinator.Protocol.InitiateBootstrapping
 import pekko.management.cluster.bootstrap.{ ClusterBootstrapSettings, LowestAddressJoinDecider }
-import org.scalatest.concurrent.Eventually
+import com.typesafe.config.ConfigFactory
 import org.scalatest.BeforeAndAfterAll
+import org.scalatest.concurrent.Eventually
+import org.scalatest.matchers.should.Matchers
 import org.scalatest.time.{ Millis, Seconds, Span }
+import org.scalatest.wordspec.AnyWordSpec
 
 import scala.concurrent.{ Await, Future }
 import scala.concurrent.duration._
-import org.scalatest.matchers.should.Matchers
-import org.scalatest.wordspec.AnyWordSpec
 
 class BootstrapCoordinatorSpec extends AnyWordSpec with Matchers with BeforeAndAfterAll with Eventually {
   val serviceName = "bootstrap-coordinator-test-service"
@@ -156,6 +157,39 @@ class BootstrapCoordinatorSpec extends AnyWordSpec with Matchers with BeforeAndA
         targetsToCheck.flatMap(_.port).toSet shouldBe empty
       }
     }
+
+    "probe all hosts with fallback port (https)" in {
+
+      MockDiscovery.set(
+        Lookup(serviceName, portName = None, protocol = Some("tcp")),
+        () =>
+          Future.successful(
+            Resolved(
+              serviceName,
+              List(
+                ResolvedTarget("host1", None, None),
+                ResolvedTarget("host1", None, None),
+                ResolvedTarget("host2", None, None),
+                ResolvedTarget("host2", None, None)))))
+
+      val targets = new AtomicReference[List[ResolvedTarget]](Nil)
+      val coordinator = system.actorOf(Props(new BootstrapCoordinator(discovery, joinDecider, settings) {
+        override def ensureProbing(selfContactPointScheme: String, contactPoint: ResolvedTarget): Option[ActorRef] = {
+          println(s"Resolving $contactPoint")
+          val targetsSoFar = targets.get
+          targets.compareAndSet(targetsSoFar, contactPoint +: targetsSoFar)
+          None
+        }
+      }))
+      coordinator ! InitiateBootstrapping(secureSelfUri)
+      eventually {
+        val targetsToCheck = targets.get
+        targetsToCheck.length should be >= 2
+        targetsToCheck.map(_.host) should contain("host1")
+        targetsToCheck.map(_.host) should contain("host2")
+        targetsToCheck.flatMap(_.port).toSet shouldBe empty
+      }
+    }
   }
 
   "BootstrapCoordinator target filtering" should {