From 1bb954649b942e8968ca0ae726eb4d47dda17887 Mon Sep 17 00:00:00 2001
From: Jim Madge <jmadge@turing.ac.uk>
Date: Fri, 20 Sep 2024 14:45:19 +0100
Subject: [PATCH] Add test for force upload

---
 tests/commands/test_config_sre.py | 41 ++++++++++++++++++++++++++-----
 1 file changed, 35 insertions(+), 6 deletions(-)

diff --git a/tests/commands/test_config_sre.py b/tests/commands/test_config_sre.py
index 8771129c9e..134705ae20 100644
--- a/tests/commands/test_config_sre.py
+++ b/tests/commands/test_config_sre.py
@@ -284,7 +284,9 @@ def test_upload_file_does_not_exist(self, mocker, runner):
         assert result.exit_code == 1
         assert "Configuration file 'fake_config.yaml' not found." in result.stdout
 
-    def test_upload_invalid_config(self, mocker, runner, context, sre_config_file, sre_config_yaml):
+    def test_upload_invalid_config(
+        self, mocker, runner, context, sre_config_file, sre_config_yaml
+    ):
         sre_name = "SandBox"
         sre_filename = sre_config_name(sre_name)
 
@@ -292,13 +294,40 @@ def test_upload_invalid_config(self, mocker, runner, context, sre_config_file, s
         mocker.patch.object(
             SREConfig, "remote_yaml_diff", side_effect=DataSafeHavenTypeError(" ")
         )
-        mocker.patch.object(
-            AzureSdk, "download_blob", return_value=sre_config_yaml
-        )
+        mocker.patch.object(AzureSdk, "download_blob", return_value=sre_config_yaml)
 
         result = runner.invoke(config_command_group, ["upload", str(sre_config_file)])
 
-        mock_exists.assert_called_once_with(context, filename=sre_filename)
         assert result.exit_code == 1
+
+        mock_exists.assert_called_once_with(context, filename=sre_filename)
         assert sre_config_yaml in result.stdout
-        assert "To overwrite the remote config, use `dsh config upload --force`" in result.stdout
+        assert (
+            "To overwrite the remote config, use `dsh config upload --force`"
+            in result.stdout
+        )
+
+    def test_upload_invalid_config_force(
+        self, mocker, runner, context, sre_config_file, sre_config_yaml
+    ):
+        sre_name = "SandBox"
+        sre_filename = sre_config_name(sre_name)
+
+        mocker.patch.object(
+            SREConfig, "remote_yaml_diff", side_effect=DataSafeHavenTypeError(" ")
+        )
+        mock_upload = mocker.patch.object(AzureSdk, "upload_blob", return_value=None)
+
+        result = runner.invoke(
+            config_command_group, ["upload", "--force", str(sre_config_file)]
+        )
+
+        assert result.exit_code == 0
+
+        mock_upload.assert_called_once_with(
+            sre_config_yaml,
+            sre_filename,
+            context.resource_group_name,
+            context.storage_account_name,
+            context.storage_container_name,
+        )