test_main.py

import os
import sys
import unittest
from unittest.mock import patch
from click.testing import CliRunner

from mesa.main import cli


class TestCli(unittest.TestCase):
    """
    Test CLI commands
    """

    def setUp(self):
        self.old_sys_path = sys.path[:]
        self.runner = CliRunner()

    def tearDown(self):
        sys.path[:] = self.old_sys_path

    def test_run(self):
        with patch("mesa.visualization.ModularServer") as ModularServer:
            example_dir = os.path.abspath(
                os.path.join(os.path.dirname(__file__), "../examples/wolf_sheep")
            )
            with self.runner.isolated_filesystem():
                result = self.runner.invoke(cli, ["runserver", example_dir])
                assert result.exit_code == 0, result.output
                assert ModularServer().launch.call_count == 1