Add integration tests for country