Set session in corresponding test.

Only one test needs to set the country in session, so there is no need
to allow the country in session in init method.