Add the getMockSession method from TestCase class