Add tests for cli module