Updating tests and implementation to allow 3 parts to SRV string

Author: kyleburgess2025

lib/mongo/srv/result.rb

@@ -110,17 +110,28 @@ def normalize_hostname(host)
       # A hostname's domain name consists of each of the '.' delineated
       # parts after the first. For example, the hostname 'foo.bar.baz'
       # has the domain name 'bar.baz'.
+      # 
+      # If the hostname has less than three parts, its domain name is the hostname itself.
       #
       # @param [ String ] record_host The host of the SRV record.
       #
       # @raise [ Mongo::Error::MismatchedDomain ] If the record's domain name doesn't match that of
       #   the hostname.
       def validate_same_origin!(record_host)
-        domain_name ||= query_hostname.split('.')[1..-1]
-        host_parts = record_host.split('.')
+        srv_is_less_than_three_parts = query_hostname.split('.').length < 3
+        srv_host_domain = if srv_is_less_than_three_parts
+          query_hostname.split('.')
+        else
+          query_hostname.split('.')[1..-1]
+        end
+        record_host_parts = record_host.split('.')
+
+        if (srv_is_less_than_three_parts && record_host_parts.length <= srv_host_domain.length) 
+          raise Error::MismatchedDomain.new(MISMATCHED_DOMAINNAME % [record_host, srv_host_domain])
+        end
 
-        unless (host_parts.size > domain_name.size) && (domain_name == host_parts[-domain_name.length..-1])
-          raise Error::MismatchedDomain.new(MISMATCHED_DOMAINNAME % [record_host, domain_name])
+        unless (srv_host_domain == record_host_parts[-srv_host_domain.size..-1])
+          raise Error::MismatchedDomain.new(MISMATCHED_DOMAINNAME % [record_host, srv_host_domain])
         end
       end
     end

lib/mongo/uri/srv_protocol.rb

@@ -184,9 +184,6 @@ def validate_srv_hostname(hostname)
         if parts.any?(&:empty?)
           raise_invalid_error!("Hostname cannot have consecutive dots: #{hostname}")
         end
-        if parts.length < 1
-          raise_invalid_error!("Hostname must have a minimum of 1 component (tld): #{hostname}")
-        end
       end
 
       # Obtains the TXT options of a host.

spec/mongo/srv/result_spec.rb

@@ -23,6 +23,73 @@
         expect(result.address_strs).to eq(['foo.bar.com:42'])
       end
     end
+
+    exampleSrvName = ['i-love-rb', 'i-love-rb.mongodb', 'i-love-ruby.mongodb.io'];
+    exampleHostName = [
+      'rb-00.i-love-rb',
+      'rb-00.i-love-rb.mongodb',
+      'i-love-ruby-00.mongodb.io'
+    ];
+    exampleHostNameThatDoNotMatchParent = [
+      'rb-00.i-love-rb-a-little',
+      'rb-00.i-love-rb-a-little.mongodb',
+      'i-love-ruby-00.evil-mongodb.io'
+    ];
+
+    (0..2).each do |i|
+      context "when srvName has #{i+1} part#{i != 0 ? 's' : ''}" do
+        let(:srv_name) { exampleSrvName[i] }
+        let(:host_name) { exampleHostName[i] }
+        let(:mismatched_host_name) { exampleHostNameThatDoNotMatchParent[i] }
+        
+        context 'when address does not match parent domain' do
+          it 'raises MismatchedDomain error' do
+            record = double('record').tap do |record|
+              allow(record).to receive(:target).and_return(mismatched_host_name)
+              allow(record).to receive(:port).and_return(42)
+              allow(record).to receive(:ttl).and_return(1)
+            end
+
+            expect {
+              result = described_class.new(srv_name)
+              result.add_record(record)
+            }.to raise_error(Mongo::Error::MismatchedDomain)
+          end
+        end
+        
+        context 'when address matches parent domain' do
+          it 'adds the record' do
+            record = double('record').tap do |record|
+              allow(record).to receive(:target).and_return(host_name)
+              allow(record).to receive(:port).and_return(42)
+              allow(record).to receive(:ttl).and_return(1)
+            end
+
+            result = described_class.new(srv_name)
+            result.add_record(record)
+
+            expect(result.address_strs).to eq([host_name + ':42'])
+          end
+        end
+
+        if i < 2
+          context 'when the address is less than 3 parts' do
+            it 'does not accept address if it does not contain an extra domain level' do
+              record = double('record').tap do |record|
+                allow(record).to receive(:target).and_return(srv_name)
+                allow(record).to receive(:port).and_return(42)
+                allow(record).to receive(:ttl).and_return(1)
+              end
+
+              expect {
+                result = described_class.new(srv_name)
+                result.add_record(record)
+              }.to raise_error(Mongo::Error::MismatchedDomain)
+            end
+          end
+        end
+      end
+    end
   end
 
   describe '#normalize_hostname' do