Add tests around cli