Add missing tests for `set_session`
Fix `set_session.to(nil)` to no longer be a no-op