address comments

Azure · Sep 13, 2024 · 0f19b1b · 0f19b1b
1 parent e958a76
commit 0f19b1b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/ga/test_extension.py b/tests/ga/test_extension.py
@@ -29,6 +29,7 @@
 
 from azurelinuxagent.common import conf
 from azurelinuxagent.common.agent_supported_feature import get_agent_supported_features_list_for_crp
+from azurelinuxagent.common.future import ustr
 from azurelinuxagent.ga.cgroupconfigurator import CGroupConfigurator
 from azurelinuxagent.common.datacontract import get_properties
 from azurelinuxagent.common.event import WALAEventOperation
@@ -3534,9 +3535,9 @@ def test_unicode_values_for_boolean_fields_in_handler_manifest(self):
             "updateCommand": "update_cmd",
             "enableCommand": "enable_cmd",
             "disableCommand": "disable_cmd",
-            "reportHeartbeat": u"true",
-            "continueOnUpdateFailure": u"true",
-            "supportsMultipleExtensions": u"true"
+            "reportHeartbeat": ustr("true"),
+            "continueOnUpdateFailure": ustr("true"),
+            "supportsMultipleExtensions": ustr("true")
         }
 
         manifest = HandlerManifest({'handlerManifest': handler_json})