Added mini testing framework and mock