Add command tests

Add tests for the search command.