Handle empty value for config set

Fixes: [#2925]
canonical · Mar 7, 2024 · c782dd6 · c782dd6
1 parent d4b79c6
commit c782dd6
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 15 deletions.
diff --git a/uaclient/cli/__init__.py b/uaclient/cli/__init__.py
@@ -690,6 +690,9 @@ def action_config_set(args, *, cfg, **kwargs):
         raise exceptions.InvalidArgChoice(
             arg="<key>", choices=", ".join(config.UA_CONFIGURABLE_KEYS)
         )
+    if not set_value.strip():
+        subparser.print_help()
+        raise exceptions.InvalidArgValue(arg=set_key)
     if set_key in ("http_proxy", "https_proxy"):
         protocol_type = set_key.split("_")[0]
         if protocol_type == "http":

diff --git a/uaclient/cli/tests/test_cli_config_set.py b/uaclient/cli/tests/test_cli_config_set.py
@@ -55,6 +55,14 @@ class TestMainConfigSet:
                 " global_apt_https_proxy, update_messaging_timer,"
                 " metering_timer",
             ),
+            (
+                "http_proxy=",
+                "Empty value provided for http_proxy.",
+            ),
+            (
+                "https_proxy=  ",
+                "Empty value provided for https_proxy.",
+            ),
         ),
     )
     @mock.patch("uaclient.cli.contract.get_available_resources")
@@ -274,13 +282,6 @@ def test_set_apt_http_proxy_and_apt_https_proxy_prints_warning(
                 "https://proxy",
                 "https://proxy",
             ),
-            (
-                "global_apt_http_proxy",
-                "",
-                apt.AptProxyScope.GLOBAL,
-                "https://proxy",
-                "https://proxy",
-            ),
         ),
     )
     @mock.patch("uaclient.cli.configure_apt_proxy")
@@ -384,13 +385,6 @@ def test_set_global_apt_http_and_global_apt_https_proxy(
                 "https://proxy",
                 "https://proxy",
             ),
-            (
-                "ua_apt_https_proxy",
-                "",
-                apt.AptProxyScope.UACLIENT,
-                "https://proxy",
-                "https://proxy",
-            ),
         ),
     )
     @mock.patch("uaclient.cli.configure_apt_proxy")

diff --git a/uaclient/exceptions.py b/uaclient/exceptions.py
@@ -335,7 +335,6 @@ class EntitlementNotFoundError(UbuntuProError):
 
 
 class EntitlementsNotEnabledError(UbuntuProError):
-
     exit_code = 4
     _msg = messages.E_ENTITLEMENTS_NOT_ENABLED_ERROR
 
@@ -496,6 +495,10 @@ class InvalidArgChoice(UbuntuProError):
     _formatted_msg = messages.E_CLI_VALID_CHOICES
 
 
+class InvalidArgValue(UbuntuProError):
+    _formatted_msg = messages.E_CLI_VALID_VALUE
+
+
 class GenericInvalidFormat(UbuntuProError):
     _formatted_msg = messages.E_CLI_EXPECTED_FORMAT
 

diff --git a/uaclient/messages/__init__.py b/uaclient/messages/__init__.py
@@ -2421,6 +2421,11 @@ def __repr__(self):
     "invalid-arg-choice", "\n" + t.gettext("{arg} must be one of: {choices}")
 )
 
+E_CLI_VALID_VALUE = FormattedNamedMessage(
+    "empty-value",
+    t.gettext("Empty value provided for {arg}."),
+)
+
 E_CLI_EXPECTED_FORMAT = FormattedNamedMessage(
     "generic-invalid-format",
     "\n" + t.gettext("Expected {expected} but found: {actual}"),