Could we move the `MockSession` to a separate file?