Add mock framework for testing locally