contract: send activityInfo after cli enable and disable

canonical · Jun 21, 2023 · ae06224 · ae06224
1 parent e1c796a
commit ae06224
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/uaclient/cli.py b/uaclient/cli.py
@@ -1262,6 +1262,9 @@ def action_disable(args, *, cfg, **kwargs):
             service_msg=service_msg,
         )
 
+    contract_client = contract.UAContractClient(cfg)
+    contract_client.update_activity_token()
+
     event.process_events()
     return 0 if ret else 1
 
@@ -1344,6 +1347,9 @@ def action_enable(args, *, cfg, **kwargs):
         event.services_failed(entitlements_not_found)
         raise exceptions.UserFacingError(msg=msg.msg, msg_code=msg.name)
 
+    contract_client = contract.UAContractClient(cfg)
+    contract_client.update_activity_token()
+
     event.process_events()
     return 0 if ret else 1