# HG changeset patch
# User Philippe Pepiot <philippe.pepiot@logilab.fr>
# Date 1521732833 -3600
#      Thu Mar 22 16:33:53 2018 +0100
# Node ID 540974ee64e621c7a94de319bdcdc140367b9ce6
# Parent  51c3f36bd42e6b3eceae7de7e7441bf5d01ff69d
Fix debugdiscovery benchmarks

* debugdiscovery does not support revset argument, so move it to its own class
  with a single 'revset' param combination.
* expected return code is always 0

diff --git a/benchmarks/basic_commands.py b/benchmarks/basic_commands.py
--- a/benchmarks/basic_commands.py
+++ b/benchmarks/basic_commands.py
@@ -152,11 +152,17 @@
             cmd.extend(['-r', self.rev])
         self._single_execute(cmd, expected_return_code=1 if strip == "same" else 0)
 
+
+class DiscoveryTimeSuite(BaseExchangeTimeSuite):
+    # debugdiscovery does not support revset argument
+    params = BaseTimeTestSuite.params + [
+        ['local', 'ssh'],
+        ['same', 'last(all(), 10)', 'last(all(), 100)', 'last(all(), 1000)'],
+        [None]]
+
     def time_debugdiscovery(self, repo_name, strip, revset):
         cmd = [self.hgpath, '--cwd', self.clone_path, 'debugdiscovery', self.repo_path]
-        # if self.rev:
-        #     cmd.extend(['--rev', self.rev])
-        self._single_execute(cmd, expected_return_code=1 if strip == "same" else 0)
+        self._single_execute(cmd)
 
 
 class PushPullTimeSuite(BaseExchangeTimeSuite):