Revert "Fixes #331 - Removes expectation methods from test session_helpers"

This reverts commit 78234582ee2a05f36f21fabfe70fb7a7890eaaa4.