Adds Wallet tests, extracts FooWalletExtension for testing.