github.com/platonnetwork/platon-go@v0.7.6/cases/environment/mock.py (about) 1 #!/usr/bin/env python3 2 # -*- coding: utf-8 -*- 3 # @Time : 2019/12/30 20:19 4 # @Author : PlatON-Developer 5 # @Site : https://github.com/PlatONnetwork/ 6 from common.log import log 7 8 class MockSftp: 9 def put(self, a, b): 10 pass 11 12 def get(self, a, b): 13 pass 14 15 16 class MockSsh: 17 def exec_command(self, cmd): 18 return MockStdin(), MockStdout(cmd), MockStderr() 19 20 21 class MockStdin: 22 23 def write(self, cmd): 24 pass 25 26 27 class MockStdout: 28 def __init__(self, cmd): 29 self.cmd = cmd 30 31 def readlines(self): 32 if "init" in self.cmd: 33 return [] 34 return [""] 35 36 37 class MockStderr: 38 pass 39 40 41 class MockT: 42 def close(self): 43 pass 44 45 46 def mock_connect_linux(): 47 log.info("mock server connect linux") 48 return MockSsh(), MockSftp(), MockT()