Add session support to testing infrastructure