Add unit tests

Author: JosueNina

lean/components/cloud/push_manager.py

@@ -254,13 +254,15 @@ def _push_metadata(self, project: Path, cloud_project: QCProject, encryption_act
                 encryption_key_id = get_project_key_hash(encryption_key)
                 update_args["encryption_key"] = encryption_key_id
 
+        if not force:
+            update_args["code_source_id"] = "cli"
+
         if update_args != {}:
             self._api_client.projects.update(cloud_project.projectId, **update_args)
 
             if "encryption_key" in update_args:
                 del update_args["encryption_key"]
-            if not force:
-                update_args["code_source_id"] = "cli"
+
             updated_keys = list(update_args)
             if len(updated_keys) == 1:
                 updated_keys_str = updated_keys[0]

tests/commands/cloud/test_push.py

@@ -15,6 +15,7 @@
 from unittest import mock
 from datetime import datetime
 
+import pytest
 from click.testing import CliRunner
 
 from lean.commands import lean
@@ -470,7 +471,11 @@ def _get_expected_encrypted_files_content() -> dict:
 bFWAAkX56MyDHwJefC1nkA=="""
 }
 
-def test_cloud_push_sets_code_source_id_to_cli() -> None:
+@pytest.mark.parametrize("force_flag, expected_code_source_id", [
+    (False, "cli"),
+    (True, None),
+])
+def test_cloud_push_code_source_id_behavior(force_flag: bool, expected_code_source_id: str | None) -> None:
     create_fake_lean_cli_directory()
     project_path = Path.cwd() / "Python Project"
 
@@ -483,15 +488,22 @@ def test_cloud_push_sets_code_source_id_to_cli() -> None:
 
     init_container(api_client_to_use=api_client)
 
-    result = CliRunner().invoke(lean, ["cloud", "push", "--project", project_path])
+    command = ["cloud", "push", "--project", project_path]
+    if force_flag:
+        command.append("--force")
 
+    result = CliRunner().invoke(lean, command)
     assert result.exit_code == 0
+
     expected_arguments = {
         "name": "Python Project",
         "description": "",
-        "files": mock.ANY,
+        "files": [{'name': 'main.py', 'content': '# region imports\nfrom AlgorithmImports import *\n# endregion\n\nclass PythonProject(QCAlgorithm):\n\n    def initialize(self):\n        # Locally Lean installs free sample data, to download more data please visit https://www.quantconnect.com/docs/v2/lean-cli/datasets/downloading-data\n        self.set_start_date(2013, 10, 7)  # Set Start Date\n        self.set_end_date(2013, 10, 11)  # Set End Date\n        self.set_cash(100000)  # Set Strategy Cash\n        self.add_equity("SPY", Resolution.MINUTE)\n\n    def on_data(self, data: Slice):\n        """on_data event is the primary entry point for your algorithm. Each new data point will be pumped in here.\n            Arguments:\n                data: Slice object keyed by symbol containing the stock data\n        """\n        if not self.portfolio.invested:\n            self.set_holdings("SPY", 1)\n            self.debug("Purchased Stock")\n'}, {'name': 'research.ipynb', 'content': '{\n    "cells": [\n        {\n            "cell_type": "markdown",\n            "metadata": {},\n            "source": [\n                "![QuantConnect Logo](https://cdn.quantconnect.com/web/i/icon.png)\\n",\n                "<hr>"\n            ]\n        },\n        {\n            "cell_type": "code",\n            "execution_count": null,\n            "metadata": {},\n            "outputs": [],\n            "source": [\n                "# QuantBook Analysis Tool \\n",\n                "# For more information see [https://www.quantconnect.com/docs/v2/our-platform/research/getting-started]\\n",\n                "qb = QuantBook()\\n",\n                "spy = qb.add_equity(\\"SPY\\")\\n",\n                "# Locally Lean installs free sample data, to download more data please visit https://www.quantconnect.com/docs/v2/lean-cli/datasets/downloading-data \\n",\n                "qb.set_start_date(2013, 10, 11)\\n",\n                "history = qb.history(qb.securities.keys(), 360, Resolution.DAILY)\\n",\n                "\\n",\n                "# Indicator Analysis\\n",\n                "bbdf = qb.indicator(BollingerBands(30, 2), spy.symbol, 360, Resolution.DAILY)\\n",\n                "bbdf.drop(\'standarddeviation\', axis=1).plot()"\n            ]\n        }\n    ],\n    "metadata": {\n        "kernelspec": {\n            "display_name": "Python 3",\n            "language": "python",\n            "name": "python3"\n        }\n    },\n    "nbformat": 4,\n    "nbformat_minor": 2\n}\n'}],
         "libraries": [],
         "encryption_key": "",
-        "codeSourceId": "cli"
     }
+
+    if expected_code_source_id is not None:
+        expected_arguments["code_source_id"] = expected_code_source_id
+
     api_client.projects.update.assert_called_once_with(1, **expected_arguments)