Simplify TopologyDescription.reset_server.

ajdavis · ajdavis · commit fcc06e52d5ce · 2015-03-05T13:57:47.000-05:00
Same effect, much more directly expressed.
diff --git a/pymongo/topology_description.py b/pymongo/topology_description.py
@@ -83,17 +83,7 @@ def has_server(self, address):
 
     def reset_server(self, address):
         """A copy of this description, with one server marked Unknown."""
-        sds = self.server_descriptions()
-
-        # The default ServerDescription's type is Unknown.
-        sds[address] = ServerDescription(address)
-
-        if self._topology_type == TOPOLOGY_TYPE.ReplicaSetWithPrimary:
-            topology_type = _check_has_primary(sds)
-        else:
-            topology_type = self._topology_type
-
-        return TopologyDescription(topology_type, sds, self._replica_set_name)
+        return updated_topology_description(self, ServerDescription(address))
 
     def reset(self):
         """A copy of this description, with all servers marked Unknown."""
