RUBY-1561 Handle mongos in server summary reporting (#1399)

Author: p-mongo

lib/mongo/server.rb

@@ -263,6 +263,8 @@ def summary
         'GHOST'
       when other?
         'OTHER'
+      when mongos?
+        'MONGOS'
       when unknown?
         'UNKNOWN'
       else

spec/mongo/server_spec.rb

@@ -380,5 +380,19 @@
         expect(server.summary).not_to include('replica_set')
       end
     end
+
+    context 'server is a mongos' do
+      let(:server) do
+        make_server(:mongos)
+      end
+
+      before do
+        expect(server).to be_mongos
+      end
+
+      it 'specifies the server is a mongos' do
+        expect(server.summary).to match(/MONGOS/)
+      end
+    end
   end
 end

spec/support/common_shortcuts.rb

@@ -46,18 +46,23 @@ def make_server(mode, options = {})
       tags = options[:tags] || {}
       average_round_trip_time = options[:average_round_trip_time] || 0
 
-      ismaster = {
-        'ismaster' => mode == :primary,
-        'secondary' => mode == :secondary,
-        'arbiterOnly' => mode == :arbiter,
-        'isreplicaset' => mode == :ghost,
-        'hidden' => mode == :other,
-        'tags' => tags,
-        'ok' => 1,
-        'minWireVersion' => 2, 'maxWireVersion' => 8,
-      }
-      if [:primary, :secondary, :arbiter, :other].include?(mode)
-        ismaster['setName'] = 'mongodb_set'
+      if mode == :unknown
+        ismaster = {}
+      else
+        ismaster = {
+          'ismaster' => mode == :primary,
+          'secondary' => mode == :secondary,
+          'arbiterOnly' => mode == :arbiter,
+          'isreplicaset' => mode == :ghost,
+          'hidden' => mode == :other,
+          'msg' => mode == :mongos ? 'isdbgrid' : nil,
+          'tags' => tags,
+          'ok' => 1,
+          'minWireVersion' => 2, 'maxWireVersion' => 8,
+        }
+        if [:primary, :secondary, :arbiter, :other].include?(mode)
+          ismaster['setName'] = 'mongodb_set'
+        end
       end
 
       listeners = Mongo::Event::Listeners.new