FilesExpand file tree

 main

/

client_wrapper.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

54 lines (42 loc) · 1.58 KB

 main

/

client_wrapper.py

Copy path

Top

File metadata and controls

• Code
• Blame

54 lines (42 loc) · 1.58 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

import pexpect

PROMPT = "> "

class MockClientWrapper:

def __init__(self):

# Start the CLI program as a child process using pexpect

self.proc = pexpect.spawn("./mock_terminal_client") # Adjust the path if needed

self.proc.expect(PROMPT)

return

def send_command(self, command):

if command == "exit":

self.exit()

# Send a command to the CLI program and return the output

self.proc.sendline(command)

self.proc.expect(PROMPT) # we expect the command to finish, and get the prompt back [in a terminal session]

output: bytes = self.proc.before

output_decoded: str = output.decode()

output_lines = output_decoded.splitlines()

output_lines.pop(0) # first line is the command that was sent

return output_lines

def help(self):

return self.send_command("help")

def hello(self):

return self.send_command("hello")

def print(self):

return self.send_command("print")

def exit(self):

# exit does not use send_command, as we don't expect the prompt to return to us

self.proc.sendline("exit")

self.proc.expect(pexpect.EOF)

self.proc.close()

if __name__ == "__main__":

client = MockClientWrapper()

print("calling help")

client_output = client.help()

print(client_output)

print("calling hello")

client_output = client.hello()

print(client_output)

print("calling print")

client_output = client.print()

print(client_output)

client.exit() # Optional, to exit the CLI program gracefully