kube_cluster 0.20.1 → 1.0.0

data/lib/kube/cluster/middleware/labels.rb

@@ -61,82 +61,82 @@ module Kube
   end
 end
 
-test do
-  Middleware = Kube::Cluster::Middleware
+__END__
 
-  it "adds_standard_labels" do
-    m = manifest(Kube::Cluster["ConfigMap"].new { metadata.name = "test" })
+Middleware = Kube::Cluster::Middleware
 
-    Middleware::Labels.new(app: "web", managed_by: "kube_cluster").call(m)
-    labels = m.resources.first.to_h.dig(:metadata, :labels)
+it "adds_standard_labels" do
+  m = manifest(Kube::Cluster["ConfigMap"].new { metadata.name = "test" })
 
-    labels[:"app.kubernetes.io/name"].should == "web"
-  end
-
-  it "maps_all_standard_keys" do
-    m = manifest(Kube::Cluster["ConfigMap"].new { metadata.name = "test" })
+  Middleware::Labels.new(app: "web", managed_by: "kube_cluster").call(m)
+  labels = m.resources.first.to_h.dig(:metadata, :labels)
 
-    Middleware::Labels.new(
-      app: "web",
-      instance: "my-release",
-      version: "1.0.0",
-      component: "frontend",
-      part_of: "platform",
-      managed_by: "kube_cluster",
-    ).call(m)
+  labels[:"app.kubernetes.io/name"].should == "web"
+end
 
-    labels = m.resources.first.to_h.dig(:metadata, :labels)
+it "maps_all_standard_keys" do
+  m = manifest(Kube::Cluster["ConfigMap"].new { metadata.name = "test" })
 
-    labels[:"app.kubernetes.io/managed-by"].should == "kube_cluster"
-  end
+  Middleware::Labels.new(
+    app: "web",
+    instance: "my-release",
+    version: "1.0.0",
+    component: "frontend",
+    part_of: "platform",
+    managed_by: "kube_cluster",
+  ).call(m)
 
-  it "resource_labels_override_middleware_defaults" do
-    m = manifest(Kube::Cluster["ConfigMap"].new {
-      metadata.name = "test"
-      metadata.labels = { "app.kubernetes.io/name": "override" }
-    })
+  labels = m.resources.first.to_h.dig(:metadata, :labels)
 
-    Middleware::Labels.new(app: "default").call(m)
-    labels = m.resources.first.to_h.dig(:metadata, :labels)
+  labels[:"app.kubernetes.io/managed-by"].should == "kube_cluster"
+end
 
-    labels[:"app.kubernetes.io/name"].should == "override"
-  end
+it "resource_labels_override_middleware_defaults" do
+  m = manifest(Kube::Cluster["ConfigMap"].new {
+    metadata.name = "test"
+    metadata.labels = { "app.kubernetes.io/name": "override" }
+  })
 
-  it "preserves_existing_labels" do
-    m = manifest(Kube::Cluster["ConfigMap"].new {
-      metadata.name = "test"
-      metadata.labels = { custom: "value" }
-    })
+  Middleware::Labels.new(app: "default").call(m)
+  labels = m.resources.first.to_h.dig(:metadata, :labels)
 
-    Middleware::Labels.new(app: "web").call(m)
-    labels = m.resources.first.to_h.dig(:metadata, :labels)
+  labels[:"app.kubernetes.io/name"].should == "override"
+end
 
-    labels[:custom].should == "value"
-  end
+it "preserves_existing_labels" do
+  m = manifest(Kube::Cluster["ConfigMap"].new {
+    metadata.name = "test"
+    metadata.labels = { custom: "value" }
+  })
 
-  it "passes_through_non_standard_keys" do
-    m = manifest(Kube::Cluster["ConfigMap"].new { metadata.name = "test" })
+  Middleware::Labels.new(app: "web").call(m)
+  labels = m.resources.first.to_h.dig(:metadata, :labels)
 
-    Middleware::Labels.new(:"team.io/name" => "platform").call(m)
-    labels = m.resources.first.to_h.dig(:metadata, :labels)
+  labels[:custom].should == "value"
+end
 
-    labels[:"team.io/name"].should == "platform"
-  end
+it "passes_through_non_standard_keys" do
+  m = manifest(Kube::Cluster["ConfigMap"].new { metadata.name = "test" })
 
-  it "converts_values_to_strings" do
-    m = manifest(Kube::Cluster["ConfigMap"].new { metadata.name = "test" })
+  Middleware::Labels.new(:"team.io/name" => "platform").call(m)
+  labels = m.resources.first.to_h.dig(:metadata, :labels)
 
-    Middleware::Labels.new(version: 2).call(m)
-    labels = m.resources.first.to_h.dig(:metadata, :labels)
+  labels[:"team.io/name"].should == "platform"
+end
 
-    labels[:"app.kubernetes.io/version"].should == "2"
-  end
+it "converts_values_to_strings" do
+  m = manifest(Kube::Cluster["ConfigMap"].new { metadata.name = "test" })
 
-  private
+  Middleware::Labels.new(version: 2).call(m)
+  labels = m.resources.first.to_h.dig(:metadata, :labels)
 
-    def manifest(*resources)
-      m = Kube::Cluster::Manifest.new
-      resources.each { |r| m << r }
-      m
-    end
+  labels[:"app.kubernetes.io/version"].should == "2"
 end
+
+private
+
+  def manifest(*resources)
+    m = Kube::Cluster::Manifest.new
+    resources.each { |r| m << r }
+    m
+  end

data/lib/kube/cluster/middleware/pod_anti_affinity.rb

@@ -63,125 +63,125 @@ module Kube
   end
 end
 
-test do
-  Middleware = Kube::Cluster::Middleware
-
-  it "injects_soft_anti_affinity_on_deployment" do
-    m = manifest(Kube::Cluster["Deployment"].new {
-      metadata.name = "web"
-      spec.selector.matchLabels = { app: "web", instance: "prod" }
-      spec.template.metadata.labels = { app: "web" }
-      spec.template.spec.containers = [
-        { name: "web", image: "nginx:latest" },
-      ]
-    })
-
-    Middleware::PodAntiAffinity.new.call(m)
-    affinity = m.resources.first.to_h.dig(:spec, :template, :spec, :affinity)
-
-    paa = affinity.dig(:podAntiAffinity, :preferredDuringSchedulingIgnoredDuringExecution)
-    paa.size.should == 1
-  end
+__END__
 
-  it "custom_topology_key" do
-    m = manifest(Kube::Cluster["Deployment"].new {
-      metadata.name = "web"
-      spec.selector.matchLabels = { app: "web" }
-      spec.template.metadata.labels = { app: "web" }
-      spec.template.spec.containers = [
-        { name: "web", image: "nginx:latest" },
-      ]
-    })
+Middleware = Kube::Cluster::Middleware
 
-    Middleware::PodAntiAffinity.new(
-      topology_key: "topology.kubernetes.io/zone",
-    ).call(m)
+it "injects_soft_anti_affinity_on_deployment" do
+  m = manifest(Kube::Cluster["Deployment"].new {
+    metadata.name = "web"
+    spec.selector.matchLabels = { app: "web", instance: "prod" }
+    spec.template.metadata.labels = { app: "web" }
+    spec.template.spec.containers = [
+      { name: "web", image: "nginx:latest" },
+    ]
+  })
 
-    affinity = m.resources.first.to_h.dig(:spec, :template, :spec, :affinity)
-    term = affinity.dig(:podAntiAffinity, :preferredDuringSchedulingIgnoredDuringExecution, 0)
+  Middleware::PodAntiAffinity.new.call(m)
+  affinity = m.resources.first.to_h.dig(:spec, :template, :spec, :affinity)
 
-    term.dig(:podAffinityTerm, :topologyKey).should == "topology.kubernetes.io/zone"
-  end
+  paa = affinity.dig(:podAntiAffinity, :preferredDuringSchedulingIgnoredDuringExecution)
+  paa.size.should == 1
+end
 
-  it "custom_weight" do
-    m = manifest(Kube::Cluster["Deployment"].new {
-      metadata.name = "web"
-      spec.selector.matchLabels = { app: "web" }
-      spec.template.metadata.labels = { app: "web" }
-      spec.template.spec.containers = [
-        { name: "web", image: "nginx:latest" },
-      ]
-    })
-
-    Middleware::PodAntiAffinity.new(weight: 100).call(m)
-    term = m.resources.first.to_h.dig(:spec, :template, :spec, :affinity,
-      :podAntiAffinity, :preferredDuringSchedulingIgnoredDuringExecution, 0)
-
-    term[:weight].should == 100
-  end
+it "custom_topology_key" do
+  m = manifest(Kube::Cluster["Deployment"].new {
+    metadata.name = "web"
+    spec.selector.matchLabels = { app: "web" }
+    spec.template.metadata.labels = { app: "web" }
+    spec.template.spec.containers = [
+      { name: "web", image: "nginx:latest" },
+    ]
+  })
 
-  it "skips_resources_with_existing_affinity" do
-    m = manifest(Kube::Cluster["Deployment"].new {
-      metadata.name = "web"
-      spec.selector.matchLabels = { app: "web" }
-      spec.template.metadata.labels = { app: "web" }
-      spec.template.spec.affinity = { nodeAffinity: { custom: true } }
-      spec.template.spec.containers = [
-        { name: "web", image: "nginx:latest" },
-      ]
-    })
-
-    Middleware::PodAntiAffinity.new.call(m)
-    affinity = m.resources.first.to_h.dig(:spec, :template, :spec, :affinity)
-
-    affinity.should == { nodeAffinity: { custom: true } }
-  end
+  Middleware::PodAntiAffinity.new(
+    topology_key: "topology.kubernetes.io/zone",
+  ).call(m)
 
-  it "skips_non_pod_bearing_resources" do
-    resource = Kube::Cluster["ConfigMap"].new { metadata.name = "config" }
-    m = manifest(resource)
+  affinity = m.resources.first.to_h.dig(:spec, :template, :spec, :affinity)
+  term = affinity.dig(:podAntiAffinity, :preferredDuringSchedulingIgnoredDuringExecution, 0)
 
-    Middleware::PodAntiAffinity.new.call(m)
+  term.dig(:podAffinityTerm, :topologyKey).should == "topology.kubernetes.io/zone"
+end
 
-    m.resources.first.to_h.should == resource.to_h
-  end
+it "custom_weight" do
+  m = manifest(Kube::Cluster["Deployment"].new {
+    metadata.name = "web"
+    spec.selector.matchLabels = { app: "web" }
+    spec.template.metadata.labels = { app: "web" }
+    spec.template.spec.containers = [
+      { name: "web", image: "nginx:latest" },
+    ]
+  })
+
+  Middleware::PodAntiAffinity.new(weight: 100).call(m)
+  term = m.resources.first.to_h.dig(:spec, :template, :spec, :affinity,
+    :podAntiAffinity, :preferredDuringSchedulingIgnoredDuringExecution, 0)
+
+  term[:weight].should == 100
+end
 
-  it "skips_resources_without_match_labels" do
-    m = manifest(Kube::Cluster["Deployment"].new {
-      metadata.name = "web"
-      spec.template.metadata.labels = { app: "web" }
-      spec.template.spec.containers = [
-        { name: "web", image: "nginx:latest" },
-      ]
-    })
+it "skips_resources_with_existing_affinity" do
+  m = manifest(Kube::Cluster["Deployment"].new {
+    metadata.name = "web"
+    spec.selector.matchLabels = { app: "web" }
+    spec.template.metadata.labels = { app: "web" }
+    spec.template.spec.affinity = { nodeAffinity: { custom: true } }
+    spec.template.spec.containers = [
+      { name: "web", image: "nginx:latest" },
+    ]
+  })
+
+  Middleware::PodAntiAffinity.new.call(m)
+  affinity = m.resources.first.to_h.dig(:spec, :template, :spec, :affinity)
+
+  affinity.should == { nodeAffinity: { custom: true } }
+end
 
-    Middleware::PodAntiAffinity.new.call(m)
-    affinity = m.resources.first.to_h.dig(:spec, :template, :spec, :affinity)
+it "skips_non_pod_bearing_resources" do
+  resource = Kube::Cluster["ConfigMap"].new { metadata.name = "config" }
+  m = manifest(resource)
 
-    affinity.should.be.nil
-  end
+  Middleware::PodAntiAffinity.new.call(m)
 
-  it "applies_to_statefulset" do
-    m = manifest(Kube::Cluster["StatefulSet"].new {
-      metadata.name = "db"
-      spec.selector.matchLabels = { app: "db" }
-      spec.template.metadata.labels = { app: "db" }
-      spec.template.spec.containers = [
-        { name: "postgres", image: "postgres:16" },
-      ]
-    })
+  m.resources.first.to_h.should == resource.to_h
+end
 
-    Middleware::PodAntiAffinity.new.call(m)
-    affinity = m.resources.first.to_h.dig(:spec, :template, :spec, :affinity)
+it "skips_resources_without_match_labels" do
+  m = manifest(Kube::Cluster["Deployment"].new {
+    metadata.name = "web"
+    spec.template.metadata.labels = { app: "web" }
+    spec.template.spec.containers = [
+      { name: "web", image: "nginx:latest" },
+    ]
+  })
 
-    affinity.dig(:podAntiAffinity).should.not.be.nil
-  end
+  Middleware::PodAntiAffinity.new.call(m)
+  affinity = m.resources.first.to_h.dig(:spec, :template, :spec, :affinity)
 
-  private
+  affinity.should.be.nil
+end
 
-    def manifest(*resources)
-      m = Kube::Cluster::Manifest.new
-      resources.each { |r| m << r }
-      m
-    end
+it "applies_to_statefulset" do
+  m = manifest(Kube::Cluster["StatefulSet"].new {
+    metadata.name = "db"
+    spec.selector.matchLabels = { app: "db" }
+    spec.template.metadata.labels = { app: "db" }
+    spec.template.spec.containers = [
+      { name: "postgres", image: "postgres:16" },
+    ]
+  })
+
+  Middleware::PodAntiAffinity.new.call(m)
+  affinity = m.resources.first.to_h.dig(:spec, :template, :spec, :affinity)
+
+  affinity.dig(:podAntiAffinity).should.not.be.nil
 end
+
+private
+
+  def manifest(*resources)
+    m = Kube::Cluster::Manifest.new
+    resources.each { |r| m << r }
+    m
+  end