Reject connection attempt when no AAAA or A records are resolved

Author: WyriHaximus

src/HappyEyeBallsConnectionBuilder.php

@@ -75,6 +75,10 @@ public function connect()
 
                     $that->mixIpsIntoConnectQueue($ips);
 
+                    if ($that->hasBeenResolved() && $that->ipsCount === 0) {
+                        $reject(new \RuntimeException($that->error()));
+                    }
+
                     // start next connection attempt if not already awaiting next
                     if ($that->nextAttemptTimer === null && $that->connectQueue) {
                         $that->check($resolve, $reject);
@@ -123,7 +127,19 @@ public function connect()
     public function resolve($type, $reject)
     {
         $that = $this;
-        return $that->resolver->resolveAll($that->host, $type)->then(null, function (\Exception $e) use ($type, $reject, $that) {
+        return $that->resolver->resolveAll($that->host, $type)->then(function ($ips) use ($that, $type) {
+            if (\count($ips) === 0) {
+                if ($type === Message::TYPE_A) {
+                    $that->lastError4 = 'no records found';
+                    $that->lastErrorFamily = 4;
+                } else {
+                    $that->lastError6 = 'no records found';
+                    $that->lastErrorFamily = 6;
+                }
+            }
+
+            return $ips;
+        }, function (\Exception $e) use ($type, $reject, $that) {
             unset($that->resolverPromises[$type]);
             $that->resolved[$type] = true;
 

tests/HappyEyeBallsConnectorTest.php

@@ -270,6 +270,21 @@ public function testCancelDuringTcpConnectionCancelsTcpConnectionIfGivenIp()
         $this->loop->run();
     }
 
+    public function testConnectionFailsWhenNoAAAOrARecordsAreResolved()
+    {
+        $this->resolver->expects($this->exactly(2))->method('resolveAll')->withConsecutive(
+            array('google.com', Message::TYPE_AAAA),
+            array('google.com', Message::TYPE_A)
+        )->willReturnOnConsecutiveCalls(
+            $this->returnValue(Promise\resolve(array())),
+            $this->returnValue(Promise\resolve(array()))
+        );
+        $this->tcp->expects($this->never())->method('connect');
+
+        $this->setExpectedException('RuntimeException', 'Connection to scheme://google.com:80/?hostname=google.com failed during DNS lookup: no records found');
+        Block\await($this->connector->connect('scheme://google.com:80/?hostname=google.com'), $this->loop, 3);
+    }
+
     /**
      * @internal
      */