corrections

psh/tools/README.md

@@ -8,7 +8,7 @@ The python module `psh` makes writing harness tests easier. It provides the foll
 
 * `assert_prompt_after_cmd(pexpect_proc, cmd, result='success', msg=None)` - Sends a command and asserts only `EOL` and next prompt. This method can be useful, when expecting command's output is not needed. Sent command isn't also asserted, so this function can be used for sending some unprintable characters. If it does not fail, the output caught before prompt will be returned. If the `msg` argument is not passed, the default error message will be used. The function also checks the exit code of a passed command in the same way as `assert_cmd()`.
 
-* `assert_prompt(pexpect_proc, msg='', timeout=-1, catch_timeout=True)` - Asserts psh prompt. There is also a possibility to pass a message to print if the assertion fails and set timeout arguments.
+* `assert_prompt(pexpect_proc, msg='', timeout=-1, catch_timeout=True)` - Asserts psh prompt by searching only for `'(psh)% '` in buffer (without checking an escape sequence). There is also a possibility to pass a message to print if the assertion fails and set timeout arguments.
 
 * `assert_prompt_fail(pexpect_proc, msg='', timeout=-1)` - Asserts that there is no psh prompt in the read buffer.
 

psh/tools/psh.py

@@ -90,18 +90,18 @@ def assert_prompt_after_cmd(pexpect_proc, cmd, result='success', msg=None):
 
 
 def assert_prompt(pexpect_proc, msg='', timeout=-1, catch_timeout=True):
-    patterns = [PROMPT]
+    patterns = ['(psh)% ']
     if catch_timeout:
         patterns.append(pexpect.TIMEOUT)
 
-    idx = pexpect_proc.expect(patterns, timeout=timeout)
+    idx = pexpect_proc.expect_exact(patterns, timeout=timeout)
     # if catch_timeout is false then pyexpect exception is raised
     assert idx == 0, msg
 
 
 def assert_prompt_fail(pexpect_proc, msg='', timeout=-1):
-    patterns = [PROMPT, pexpect.TIMEOUT, pexpect.EOF]
-    idx = pexpect_proc.expect(patterns, timeout=timeout)
+    patterns = ['(psh)% ', pexpect.TIMEOUT, pexpect.EOF]
+    idx = pexpect_proc.expect_exact(patterns, timeout=timeout)
     assert idx != 0, msg