Add mock client and testing code.