allow disabling elementary version check

elementary/cli/upgrade.py

@@ -3,10 +3,16 @@
 import click
 from packaging import version
 
+from elementary.config.config import Config
 from elementary.utils import package
 
 
 def recommend_version_upgrade():
+    config = Config()
+
+    if config.disable_elementary_version_check:
+        return
+
     try:
         latest_version = package.get_latest_package_version()
         current_version = package.get_package_version()

elementary/config/config.py

@@ -190,6 +190,10 @@ def __init__(
             run_dbt_deps_if_needed, config.get("run_dbt_deps_if_needed"), True
         )
 
+        self.disable_elementary_version_check = config.get(
+            "disable_elementary_version_check", False
+        )
+
     def _load_configuration(self) -> dict:
         if not os.path.exists(self.config_dir):
             os.makedirs(self.config_dir)

tests/unit/config/test_config.py

@@ -14,7 +14,8 @@
         "token": "test_slack_token",
         "channel_name": "test_channel_name",
         "workflows": False,
-    }
+    },
+    "disable_elementary_version_check": True,
 }
 
 WORKFLOWS_CONFIG = {
@@ -54,6 +55,14 @@ def test_config_get_slack_notification_channel_name(config: Config):
     assert config.slack_channel_name == CONFIG["slack"]["channel_name"]
 
 
+@pytest.mark.parametrize("config", [CONFIG], indirect=["config"])
+def test_disable_elementary_version_check(config: Config):
+    assert (
+        config.disable_elementary_version_check
+        == CONFIG["disable_elementary_version_check"]
+    )
+
+
 @pytest.mark.parametrize("config", [WORKFLOWS_CONFIG], indirect=["config"])
 def test_slack_workflows_config_get_workflows(config: Config):
     assert config.is_slack_workflow == WORKFLOWS_CONFIG["slack"]["workflows"]