Add tests for new session methods