summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--armada/handlers/chart_deploy.py5
-rw-r--r--armada/tests/unit/handlers/test_armada.py5
2 files changed, 6 insertions, 4 deletions
diff --git a/armada/handlers/chart_deploy.py b/armada/handlers/chart_deploy.py
index 30671fd..203a145 100644
--- a/armada/handlers/chart_deploy.py
+++ b/armada/handlers/chart_deploy.py
@@ -121,8 +121,9 @@ class ChartDeploy(object):
121 121
122 upgrade = chart.get('upgrade', {}) 122 upgrade = chart.get('upgrade', {})
123 disable_hooks = upgrade.get('no_hooks', False) 123 disable_hooks = upgrade.get('no_hooks', False)
124 force = upgrade.get('force', False) 124 options = upgrade.get('options', {})
125 recreate_pods = upgrade.get('recreate_pods', False) 125 force = options.get('force', False)
126 recreate_pods = options.get('recreate_pods', False)
126 127
127 if upgrade: 128 if upgrade:
128 upgrade_pre = upgrade.get('pre', {}) 129 upgrade_pre = upgrade.get('pre', {})
diff --git a/armada/tests/unit/handlers/test_armada.py b/armada/tests/unit/handlers/test_armada.py
index 06321da..d1a35dc 100644
--- a/armada/tests/unit/handlers/test_armada.py
+++ b/armada/tests/unit/handlers/test_armada.py
@@ -437,8 +437,9 @@ class ArmadaHandlerTestCase(base.ArmadaTestCase):
437 if diff: 437 if diff:
438 upgrade = chart.get('upgrade', {}) 438 upgrade = chart.get('upgrade', {})
439 disable_hooks = upgrade.get('no_hooks', False) 439 disable_hooks = upgrade.get('no_hooks', False)
440 force = upgrade.get('force', False) 440 options = upgrade.get('options', {})
441 recreate_pods = upgrade.get( 441 force = options.get('force', False)
442 recreate_pods = options.get(
442 'recreate_pods', False) 443 'recreate_pods', False)
443 444
444 expected_update_release_calls.append( 445 expected_update_release_calls.append(