Add JMock support