Implement wallet tests