Prune pdb

Shopify · Aug 9, 2018 · 84d87b3 · 84d87b3
1 parent b9aefc5
commit 84d87b3
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/lib/kubernetes-deploy/deploy_task.rb b/lib/kubernetes-deploy/deploy_task.rb
@@ -82,6 +82,7 @@ def prune_whitelist
         extensions/v1beta1/Ingress
         apps/v1beta1/StatefulSet
         autoscaling/v1/HorizontalPodAutoscaler
+        policy/v1beta1/PodDisruptionBudget
       )
       if server_version >= Gem::Version.new('1.8.0')
         wl << "batch/v1beta1/CronJob"

diff --git a/test/integration-serial/run_serial_test.rb b/test/integration-serial/run_serial_test.rb
@@ -158,13 +158,6 @@ def test_crd_can_fail
   end
 
   def test_crd_pruning
-    widget = Class.new(KubernetesDeploy::KubernetesResource) do
-      def deploy_method
-        :replace
-      end
-    end
-    KubernetesDeploy.const_set("Widget", widget)
-
     assert_deploy_success(deploy_fixtures("crd", subset: %w(mail.yml widgets.yml)))
     assert_logs_match_all([
       "Phase 1: Initializing deploy",

diff --git a/test/integration/kubernetes_deploy_test.rb b/test/integration/kubernetes_deploy_test.rb
@@ -1070,4 +1070,10 @@ def test_hpa_can_be_pruned
     assert_deploy_success(deploy_fixtures("hpa", subset: ["deployment.yml"]))
     assert_logs_match_all([/The following resources were pruned: horizontalpodautoscaler(.autoscaling)? "hello-hpa"/])
   end
+
+  def test_not_apply_resource_can_be_pruned
+    assert_deploy_success(deploy_fixtures("hello-cloud", subset: %w(disruption-budgets.yml configmap-data.yml)))
+    assert_deploy_success(deploy_fixtures("hello-cloud", subset: %w(configmap-data.yml)))
+    assert_logs_match_all([/The following resources were pruned: poddisruptionbudget(.policy)? "test"/])
+  end
 end