"""
Tests for CLI module.
"""

import pytest
from dcos.utils.cli import main


class TestCLI:
    def test_version(self):
        result = main(["--version"])
        assert result == 0

    def test_status(self):
        result = main(["status"])
        assert result == 0

    def test_info(self):
        result = main(["info"])
        assert result == 0

    def test_unknown_command(self):
        with pytest.raises(SystemExit) as exc:
            main(["badcommand"])
        # argparse exits with code 2 for invalid choices
        assert exc.value.code == 2