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()