From 7a7ded6b7e644cac0107ac7f3d1b8614e480ee4c Mon Sep 17 00:00:00 2001
From: bmc-msft <41130664+bmc-msft@users.noreply.github.com>
Date: Tue, 13 Jul 2021 12:08:07 -0400
Subject: [PATCH] force upgrade custom script extensions (#1059)

---
 src/api-service/__app__/onefuzzlib/extension.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/api-service/__app__/onefuzzlib/extension.py b/src/api-service/__app__/onefuzzlib/extension.py
index 9a6ec2308..0b0abb945 100644
--- a/src/api-service/__app__/onefuzzlib/extension.py
+++ b/src/api-service/__app__/onefuzzlib/extension.py
@@ -5,7 +5,7 @@
 
 import os
 from typing import List, Optional
-from uuid import UUID
+from uuid import UUID, uuid4
 
 from onefuzztypes.enums import OS, AgentMode
 from onefuzztypes.models import AgentConfig, Pool, ReproConfig, Scaleset
@@ -211,6 +211,7 @@ def agent_config(
             "type": "CustomScriptExtension",
             "publisher": "Microsoft.Compute",
             "location": region,
+            "force_update_tag": uuid4(),
             "type_handler_version": "1.9",
             "auto_upgrade_minor_version": True,
             "settings": {"commandToExecute": to_execute_cmd, "fileUris": urls},
@@ -247,6 +248,7 @@ def agent_config(
             "typeHandlerVersion": "2.1",
             "location": region,
             "autoUpgradeMinorVersion": True,
+            "force_update_tag": uuid4(),
             "settings": {"commandToExecute": to_execute_cmd, "fileUris": urls},
             "protectedSettings": {},
         }