FM / test

/usr/lib/python3/dist-packages/twisted/conch/test/

/usr/lib/python3/dist-packages/twisted/conch/test

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
keydata.py	35208	0644	edit dl rm
loopback.py	706	0644	edit dl rm
test_address.py	1550	0644	edit dl rm
test_agent.py	13219	0644	edit dl rm
test_cftp.py	50503	0644	edit dl rm
test_channel.py	12183	0644	edit dl rm
test_checkers.py	31340	0644	edit dl rm
test_ckeygen.py	24339	0644	edit dl rm
test_conch.py	25740	0644	edit dl rm
test_connection.py	29257	0644	edit dl rm
test_default.py	11588	0644	edit dl rm
test_endpoints.py	55888	0644	edit dl rm
test_filetransfer.py	33133	0644	edit dl rm
test_forwarding.py	2162	0644	edit dl rm
test_helper.py	20033	0644	edit dl rm
test_insults.py	32132	0644	edit dl rm
test_keys.py	62519	0644	edit dl rm
test_knownhosts.py	49497	0644	edit dl rm
test_manhole.py	13257	0644	edit dl rm
test_manhole_tap.py	4351	0644	edit dl rm
test_mixin.py	1054	0644	edit dl rm
test_openssh_compat.py	4876	0644	edit dl rm
test_recvline.py	25432	0644	edit dl rm
test_scripts.py	1781	0644	edit dl rm
test_session.py	44793	0644	edit dl rm
test_ssh.py	32938	0644	edit dl rm
test_tap.py	5037	0644	edit dl rm
test_telnet.py	26807	0644	edit dl rm
test_text.py	4021	0644	edit dl rm
test_transport.py	107271	0644	edit dl rm
test_unix.py	2600	0644	edit dl rm
test_userauth.py	33555	0644	edit dl rm
test_window.py	2113	0644	edit dl rm
__init__.py	14	0644	edit dl rm

Edit: /usr/lib/python3/dist-packages/twisted/conch/test/test_unix.py (2600B)

# -*- test-case-name: twisted.conch.test.test_unix -*-
# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.

from zope.interface import implementer

from twisted.internet.interfaces import IReactorProcess
from twisted.python.reflect import requireModule
from twisted.trial import unittest

cryptography = requireModule("cryptography")
unix = requireModule("twisted.conch.unix")

@implementer(IReactorProcess)
class MockProcessSpawner:
    """
    An L{IReactorProcess} that logs calls to C{spawnProcess}.
    """

def __init__(self):
        self._spawnProcessCalls = []

def spawnProcess(
        self,
        processProtocol,
        executable,
        args=(),
        env={},
        path=None,
        uid=None,
        gid=None,
        usePTY=0,
        childFDs=None,
    ):
        """
        Log a call to C{spawnProcess}. Do not actually spawn a process.
        """
        self._spawnProcessCalls.append(
            {
                "processProtocol": processProtocol,
                "executable": executable,
                "args": args,
                "env": env,
                "path": path,
                "uid": uid,
                "gid": gid,
                "usePTY": usePTY,
                "childFDs": childFDs,
            }
        )

class StubUnixConchUser:
    """
    Enough of UnixConchUser to exercise SSHSessionForUnixConchUser in the
    tests below.
    """

def __init__(self, homeDirectory):
        from .test_session import StubClient, StubConnection

self._homeDirectory = homeDirectory
        self.conn = StubConnection(transport=StubClient())

def getUserGroupId(self):
        return (None, None)

def getHomeDir(self):
        return self._homeDirectory

def getShell(self):
        pass

class TestSSHSessionForUnixConchUser(unittest.TestCase):

if cryptography is None:
        skip = "Cannot run without cryptography"
    elif unix is None:
        skip = "Unix system required"

def testExecCommandEnvironment(self):
        """
        C{execCommand} sets the C{HOME} environment variable to the avatar's home
        directory.
        """
        mockReactor = MockProcessSpawner()
        homeDirectory = "/made/up/path/"
        avatar = StubUnixConchUser(homeDirectory)
        session = unix.SSHSessionForUnixConchUser(avatar, reactor=mockReactor)
        protocol = None
        command = ["not-actually-executed"]
        session.execCommand(protocol, command)
        [call] = mockReactor._spawnProcessCalls
        self.assertEqual(homeDirectory, call["env"]["HOME"])

# -*- test-case-name: twisted.conch.test.test_unix -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. from zope.interface import implementer from twisted.internet.interfaces import IReactorProcess from twisted.python.reflect import requireModule from twisted.trial import unittest cryptography = requireModule("cryptography") unix = requireModule("twisted.conch.unix") @implementer(IReactorProcess) class MockProcessSpawner: """ An L{IReactorProcess} that logs calls to C{spawnProcess}. """ def __init__(self): self._spawnProcessCalls = [] def spawnProcess( self, processProtocol, executable, args=(), env={}, path=None, uid=None, gid=None, usePTY=0, childFDs=None, ): """ Log a call to C{spawnProcess}. Do not actually spawn a process. """ self._spawnProcessCalls.append( { "processProtocol": processProtocol, "executable": executable, "args": args, "env": env, "path": path, "uid": uid, "gid": gid, "usePTY": usePTY, "childFDs": childFDs, } ) class StubUnixConchUser: """ Enough of UnixConchUser to exercise SSHSessionForUnixConchUser in the tests below. """ def __init__(self, homeDirectory): from .test_session import StubClient, StubConnection self._homeDirectory = homeDirectory self.conn = StubConnection(transport=StubClient()) def getUserGroupId(self): return (None, None) def getHomeDir(self): return self._homeDirectory def getShell(self): pass class TestSSHSessionForUnixConchUser(unittest.TestCase): if cryptography is None: skip = "Cannot run without cryptography" elif unix is None: skip = "Unix system required" def testExecCommandEnvironment(self): """ C{execCommand} sets the C{HOME} environment variable to the avatar's home directory. """ mockReactor = MockProcessSpawner() homeDirectory = "/made/up/path/" avatar = StubUnixConchUser(homeDirectory) session = unix.SSHSessionForUnixConchUser(avatar, reactor=mockReactor) protocol = None command = ["not-actually-executed"] session.execCommand(protocol, command) [call] = mockReactor._spawnProcessCalls self.assertEqual(homeDirectory, call["env"]["HOME"])