# HG changeset patch
# User Pierre-Yves David <pierre-yves.david@fb.com>
# Date 1420592864 28800
# Node ID 31e75a362d44b1826a8d269a0f2a38d2d50918c3
# Parent  6a5877a73141448e6d91bc74e53808b428357e41
setdiscovery: drop '_setupsample' usage in '_takequicksample'

For '_takefullsample' we can just retrieve the list of head directly and
ignore the rest of the complex return values. This was the last call to the
infamous '_updatesample' function.

diff -r 6a5877a73141 -r 31e75a362d44 mercurial/setdiscovery.py
--- a/mercurial/setdiscovery.py	Wed Jan 07 10:32:17 2015 -0800
+++ b/mercurial/setdiscovery.py	Tue Jan 06 17:07:44 2015 -0800
@@ -95,10 +95,9 @@
     return always, set(), desiredlen
 
 def _takequicksample(dag, nodes, size):
-    always, sample, desiredlen = _setupsample(dag, nodes, size)
-    if sample is None:
-        return always
-    sample = always
+    sample = dag.headsetofconnecteds(nodes)
+    if size <= len(sample):
+        return _limitsample(sample, size)
     _updatesample(dag, None, sample, quicksamplesize=size)
     return sample