Merge pull request #116 from quickstrom/handle-end-after-error

Handle End message after Error message
quickstrom · Nov 9, 2022 · b8352cd · b8352cd
2 parents 19ffd6d + dcc15da
commit b8352cd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/quickstrom/executor.py b/quickstrom/executor.py
@@ -278,6 +278,9 @@ def run_sessions() -> List[result.PlainResult]:
                             raise e
                         except Exception as e:
                             send(Error(str(e)))
+                            msg = receive() 
+                            if not isinstance(msg, End):
+                                raise Exception(f"Expected End after Error but got: {msg}")
                     elif isinstance(msg, Done):
                         return [
                             attach_screenshots(result.from_protocol_result(r))