diff options
| author | Todd Leonhardt <todd.leonhardt@gmail.com> | 2018-09-30 11:38:00 -0400 |
|---|---|---|
| committer | Todd Leonhardt <todd.leonhardt@gmail.com> | 2018-09-30 11:38:00 -0400 |
| commit | 5a77702462a95de55cd70f5130e244db296abcd5 (patch) | |
| tree | 2065cc6519b10c6708fd58bf0e214d8f9aa0b516 /cmd2/transcript.py | |
| parent | bc49e71c73eb3593705a7332bb5352825def88b3 (diff) | |
| download | cmd2-git-5a77702462a95de55cd70f5130e244db296abcd5.tar.gz | |
Fixed transcript testing bug where last command in transcript has no expected output
Also:
- Added unit test for this specific case
Diffstat (limited to 'cmd2/transcript.py')
| -rw-r--r-- | cmd2/transcript.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/cmd2/transcript.py b/cmd2/transcript.py index 2d94f4e4..baaa6caf 100644 --- a/cmd2/transcript.py +++ b/cmd2/transcript.py @@ -67,7 +67,10 @@ class Cmd2TestCase(unittest.TestCase): break line_num += 1 command = [line[len(self.cmdapp.visible_prompt):]] - line = next(transcript) + try: + line = next(transcript) + except StopIteration: + line = '' # Read the entirety of a multi-line command while line.startswith(self.cmdapp.continuation_prompt): command.append(line[len(self.cmdapp.continuation_prompt):]) |