diff --git a/brian_polling_manager/cli.py b/brian_polling_manager/main.py similarity index 98% rename from brian_polling_manager/cli.py rename to brian_polling_manager/main.py index 5be7ba2c3f73e6a971ea17b4dc583222204535e0..1b8999721a7f25a587ceff8ab17ee20a583d5c7a 100644 --- a/brian_polling_manager/cli.py +++ b/brian_polling_manager/main.py @@ -61,7 +61,7 @@ def _validate_config(_ctx, _param, file): '--force/--no-force', default=False, help="refresh inventory data even if it hasn't been updated") -def main(config, force): +def cli(config, force): """ Update BRIAN snmp checks based on Inventory Provider data. """ @@ -85,4 +85,4 @@ def main(config, force): if __name__ == '__main__': - main() + cli() diff --git a/docs/source/index.rst b/docs/source/index.rst index d87a4b83d0b48564ca2897eb4d64af5692b64444..4ced46d553c7bfc2978b28ca97fa62a94d276141 100644 --- a/docs/source/index.rst +++ b/docs/source/index.rst @@ -16,4 +16,4 @@ Sensu checks for polling the data required by BRIAN. :maxdepth: 2 :caption: Contents: - cli + main diff --git a/docs/source/cli.rst b/docs/source/main.rst similarity index 64% rename from docs/source/cli.rst rename to docs/source/main.rst index 128c1a0d70a7d1bd43ed6b1e2991dae613a4d20a..4ef198ec2450507e2c450a9ce8c6c13bc548b8c3 100644 --- a/docs/source/cli.rst +++ b/docs/source/main.rst @@ -2,5 +2,5 @@ brian-polling-manager =================================================== -.. automodule:: brian_polling_manager.cli +.. automodule:: brian_polling_manager.main diff --git a/test/test_e2e.py b/test/test_e2e.py index 61d48375e9cd69f2a74ab9e99fbfc6f27c61f4b9..11b6d741cbc43f5249be9e1b0b870422ca7dbf6b 100644 --- a/test/test_e2e.py +++ b/test/test_e2e.py @@ -4,7 +4,7 @@ import tempfile from click.testing import CliRunner import responses -from brian_polling_manager import cli +from brian_polling_manager import main @responses.activate @@ -16,7 +16,7 @@ def test_run_flashtest(config, mocked_sensu, mocked_inventory): runner = CliRunner() result = runner.invoke( - cli.main, + main.cli, ['--config', f.name, '--force'] ) assert result.exit_code == 0