Add CLI tests