diff --git a/reactive/kubernetes_worker.py b/reactive/kubernetes_worker.py index c28aacd..46a822a 100644 --- a/reactive/kubernetes_worker.py +++ b/reactive/kubernetes_worker.py @@ -37,6 +37,7 @@ from charms.reactive import is_state, is_flag_set, any_flags_set from charms.reactive import when, when_any, when_not, when_none from charms.reactive import data_changed, is_data_changed +from charms.reactive import register_trigger from charmhelpers.core import hookenv, unitdata from charmhelpers.core.host import service_stop, service_restart @@ -102,6 +103,11 @@ db = unitdata.kv() +register_trigger( + when="endpoint.azure.ready.changed", clear_flag="kubernetes-worker.cloud.ready" +) + + @hook("upgrade-charm") def upgrade_charm(): # migrate to new flags @@ -1265,6 +1271,7 @@ def cloud_ready(): write_gcp_snap_config("kubelet") elif is_state("endpoint.azure.ready"): write_azure_snap_config("kubelet") + clear_flag("endpoint.azure.ready.changed") set_state("kubernetes-worker.cloud.ready") set_state("kubernetes-worker.restart-needed") # force restart diff --git a/tox.ini b/tox.ini index 410c35e..4586b8f 100644 --- a/tox.ini +++ b/tox.ini @@ -44,3 +44,9 @@ deps = commands = flake8 {toxinidir}/reactive {toxinidir}/tests {toxinidir}/actions black --check {toxinidir}/reactive {toxinidir}/tests {toxinidir}/actions + +[testenv:format] +deps = + black +commands = + black {toxinidir}/reactive {toxinidir}/tests {toxinidir}/actions