Adding session unit tests