fix broken resolver specs

Author: jamis

spec/integration/reconnect_spec.rb

@@ -181,8 +181,6 @@
       end
 
       around do |example|
-        require 'support/dns'
-
         rules = [
           ['_mongodb._tcp.test-fake.test.build.10gen.cc', :srv,
             [0, 0, 2799, 'localhost.test.build.10gen.cc'],

spec/integration/srv_monitoring_spec.rb

@@ -76,10 +76,6 @@
     # NotImplementedError: recvmsg_nonblock is not implemented
     fails_on_jruby
 
-    before(:all) do
-      require 'support/dns'
-    end
-
     around do |example|
       # Speed up the tests by listening on the fake ports we are using.
       done = false

spec/integration/srv_spec.rb

@@ -12,10 +12,6 @@
     # NotImplementedError: recvmsg_nonblock is not implemented
     fails_on_jruby
 
-    before(:all) do
-      require 'support/dns'
-    end
-
     let(:uri) do
       "mongodb+srv://test-fake.test.build.10gen.cc/?tls=#{SpecConfig.instance.ssl?}&tlsInsecure=true"
     end

spec/support/common_shortcuts.rb

@@ -337,51 +337,33 @@ def stop_monitoring(*clients)
       [:tcp, "0.0.0.0", 5300],
     ]
 
-    def mock_dns(config)
-      semaphore = Mongo::Semaphore.new
-
-      thread = Thread.new do
-        RubyDNS::run_server(DNS_INTERFACES) do
-          config.each do |(query, type, *answers)|
-
-            resource_cls = Resolv::DNS::Resource::IN.const_get(type.to_s.upcase)
-            resources = answers.map do |answer|
-              resource_cls.new(*answer)
-            end
-            match(query, resource_cls) do |req|
-              req.add(resources)
-            end
+    # Starts the DNS server and returns it; should be run from within an
+    # Async block. Prefer #mock_dns instead, which does the setup for you.
+    def start_dns_server(config)
+      RubyDNS::run_server(DNS_INTERFACES) do
+        config.each do |(query, type, *answers)|
+          resource_cls = Resolv::DNS::Resource::IN.const_get(type.to_s.upcase)
+          resources = answers.map do |answer|
+            resource_cls.new(*answer)
           end
 
-          semaphore.signal
+          match(query, resource_cls) do |req|
+            req.add(resources)
+          end
         end
       end
+    end
 
-      semaphore.wait
+    # Starts and runs a DNS server, then yields to the attached block.
+    def mock_dns(config)
+      # only require rubydns when we need it; it's MRI-only.
+      require 'rubydns'
 
-      begin
+      Async do |task|
+        server = start_dns_server(config)
         yield
       ensure
-        10.times do
-          if $last_async_task
-            break
-          end
-          sleep 0.5
-        end
-
-        # Hack to stop the server - https://github.com/socketry/rubydns/issues/75
-        if $last_async_task.nil?
-          STDERR.puts "No async task - server never started?"
-        else
-          begin
-            $last_async_task.stop
-          rescue NoMethodError => e
-            STDERR.puts "Error stopping async task: #{e}"
-          end
-        end
-
-        thread.kill
-        thread.join
+        server.stop
       end
     end