updating unit test for session.