upstream/mercurial-mirror Files · tests/test-sshserver.py

bookmarks: cache reverse mapping (issue5868)...

bookmarks: cache reverse mapping (issue5868) I chose a simpler implementation. If the initial cost of building reverse mapping is significant, we'll have to move it under @propertycache. The nodemap could be a dict of sets, but I think keeping a sorted list is better since each node is likely to have zero/one bookmark. Micro-benchmark with 1001 bookmarks and 1001 revisions: $ for n in `seq 0 1000`; do touch $n; hg book book$n; hg ci -qAm$n; done $ hg bookmarks --time > /dev/null (orig) time: real 0.040 secs (user 0.050+0.000 sys 0.000+0.000) (new) time: real 0.040 secs (user 0.040+0.000 sys 0.010+0.000) $ hg log -T '{bookmarks}\n' --time > /dev/null (orig) time: real 0.160 secs (user 0.160+0.000 sys 0.000+0.000) (new) time: real 0.090 secs (user 0.100+0.000 sys 0.000+0.000)

Gregory Szorc - - Load All Authors

File last commit:

r37803:b4d85bc1 default


                r37869:04ceb267

@26 default

Download file

             test-sshserver.py
        
                    53 lines
            
             | 1.5 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / tests / test-sshserver.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      from __future__ import absolute_import, print_function

      import io

      import unittest

      import silenttestrunner

      from mercurial import (

          wireprotoserver,

          wireprotov1server,

      )

      from mercurial.utils import (

          procutil,

      )

      class SSHServerGetArgsTests(unittest.TestCase):

          def testparseknown(self):

              tests = [

                  (b'* 0\nnodes 0\n', [b'', {}]),

                  (b'* 0\nnodes 40\n1111111111111111111111111111111111111111\n',

                   [b'1111111111111111111111111111111111111111', {}]),

              ]

              for input, expected in tests:

                  self.assertparse(b'known', input, expected)

          def assertparse(self, cmd, input, expected):

              server = mockserver(input)

              proto = wireprotoserver.sshv1protocolhandler(server._ui,

                                                           server._fin,

                                                           server._fout)

              _func, spec = wireprotov1server.commands[cmd]

              self.assertEqual(proto.getargs(spec), expected)

      def mockserver(inbytes):

          ui = mockui(inbytes)

          repo = mockrepo(ui)

          return wireprotoserver.sshserver(ui, repo)

      class mockrepo(object):

          def __init__(self, ui):

              self.ui = ui

      class mockui(object):

          def __init__(self, inbytes):

              self.fin = io.BytesIO(inbytes)

              self.fout = io.BytesIO()

              self.ferr = io.BytesIO()

      if __name__ == '__main__':

          # Don't call into msvcrt to set BytesIO to binary mode

          procutil.setbinary = lambda fp: True

          silenttestrunner.main(__name__)

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				from __future__ import absolute_import, print_function

				import io
				import unittest

				import silenttestrunner

				from mercurial import (
				wireprotoserver,
				wireprotov1server,
				)

				from mercurial.utils import (
				procutil,
				)

				class SSHServerGetArgsTests(unittest.TestCase):
				def testparseknown(self):
				tests = [
				(b'* 0\nnodes 0\n', [b'', {}]),
				(b'* 0\nnodes 40\n1111111111111111111111111111111111111111\n',
				[b'1111111111111111111111111111111111111111', {}]),
				]
				for input, expected in tests:
				self.assertparse(b'known', input, expected)

				def assertparse(self, cmd, input, expected):
				server = mockserver(input)
				proto = wireprotoserver.sshv1protocolhandler(server._ui,
				server._fin,
				server._fout)
				_func, spec = wireprotov1server.commands[cmd]
				self.assertEqual(proto.getargs(spec), expected)

				def mockserver(inbytes):
				ui = mockui(inbytes)
				repo = mockrepo(ui)
				return wireprotoserver.sshserver(ui, repo)

				class mockrepo(object):
				def __init__(self, ui):
				self.ui = ui

				class mockui(object):
				def __init__(self, inbytes):
				self.fin = io.BytesIO(inbytes)
				self.fout = io.BytesIO()
				self.ferr = io.BytesIO()

				if __name__ == '__main__':
				# Don't call into msvcrt to set BytesIO to binary mode
				procutil.setbinary = lambda fp: True
				silenttestrunner.main(__name__)