Adding test for session begin/end