mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 17:27:53 +03:00
9dc21f8d0b
Summary: D13853115 adds `edenscm/` to `sys.path` and code still uses `import mercurial`. That has nasty problems if both `import mercurial` and `import edenscm.mercurial` are used, because Python would think `mercurial.foo` and `edenscm.mercurial.foo` are different modules so code like `try: ... except mercurial.error.Foo: ...`, or `isinstance(x, mercurial.foo.Bar)` would fail to handle the `edenscm.mercurial` version. There are also some module-level states (ex. `extensions._extensions`) that would cause trouble if they have multiple versions in a single process. Change imports to use the `edenscm` so ideally the `mercurial` is no longer imported at all. Add checks in extensions.py to catch unexpected extensions importing modules from the old (wrong) locations when running tests. Reviewed By: phillco Differential Revision: D13868981 fbshipit-source-id: f4e2513766957fd81d85407994f7521a08e4de48
48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
from __future__ import absolute_import, print_function
|
|
|
|
import io
|
|
import unittest
|
|
|
|
import silenttestrunner
|
|
from edenscm.mercurial import sshserver, wireproto
|
|
|
|
|
|
class SSHServerGetArgsTests(unittest.TestCase):
|
|
def testparseknown(self):
|
|
tests = [
|
|
("* 0\nnodes 0\n", ["", {}]),
|
|
(
|
|
"* 0\nnodes 40\n1111111111111111111111111111111111111111\n",
|
|
["1111111111111111111111111111111111111111", {}],
|
|
),
|
|
]
|
|
for input, expected in tests:
|
|
self.assertparse("known", input, expected)
|
|
|
|
def assertparse(self, cmd, input, expected):
|
|
server = mockserver(input)
|
|
_func, spec = wireproto.commands[cmd]
|
|
self.assertEqual(server.getargs(spec), expected)
|
|
|
|
|
|
def mockserver(inbytes):
|
|
ui = mockui(inbytes)
|
|
repo = mockrepo(ui)
|
|
return sshserver.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__":
|
|
silenttestrunner.main(__name__)
|