From c4d493881d4f3741b1b0b58c58c369fd02a93a63 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?nils=20m=C3=A5s=C3=A9n?= <nils@piksel.se>
Date: Mon, 23 Oct 2023 10:31:07 +0200
Subject: [PATCH] fix: handle missing healthcheck keys in config (#1810)

---
 pkg/container/container.go      | 30 ++++++++++++++++--------------
 pkg/container/container_test.go | 30 ++++++++++++++++++++++++++++++
 2 files changed, 46 insertions(+), 14 deletions(-)

diff --git a/pkg/container/container.go b/pkg/container/container.go
index 0433c1e..10ed677 100644
--- a/pkg/container/container.go
+++ b/pkg/container/container.go
@@ -305,24 +305,26 @@ func (c Container) GetCreateConfig() *dockercontainer.Config {
 	}
 
 	// Clear HEALTHCHECK configuration (if default)
-	if util.SliceEqual(config.Healthcheck.Test, imageConfig.Healthcheck.Test) {
-		config.Healthcheck.Test = nil
-	}
+	if config.Healthcheck != nil && imageConfig.Healthcheck != nil {
+		if util.SliceEqual(config.Healthcheck.Test, imageConfig.Healthcheck.Test) {
+			config.Healthcheck.Test = nil
+		}
 
-	if config.Healthcheck.Retries == imageConfig.Healthcheck.Retries {
-		config.Healthcheck.Retries = 0
-	}
+		if config.Healthcheck.Retries == imageConfig.Healthcheck.Retries {
+			config.Healthcheck.Retries = 0
+		}
 
-	if config.Healthcheck.Interval == imageConfig.Healthcheck.Interval {
-		config.Healthcheck.Interval = 0
-	}
+		if config.Healthcheck.Interval == imageConfig.Healthcheck.Interval {
+			config.Healthcheck.Interval = 0
+		}
 
-	if config.Healthcheck.Timeout == imageConfig.Healthcheck.Timeout {
-		config.Healthcheck.Timeout = 0
-	}
+		if config.Healthcheck.Timeout == imageConfig.Healthcheck.Timeout {
+			config.Healthcheck.Timeout = 0
+		}
 
-	if config.Healthcheck.StartPeriod == imageConfig.Healthcheck.StartPeriod {
-		config.Healthcheck.StartPeriod = 0
+		if config.Healthcheck.StartPeriod == imageConfig.Healthcheck.StartPeriod {
+			config.Healthcheck.StartPeriod = 0
+		}
 	}
 
 	config.Env = util.SliceSubtract(config.Env, imageConfig.Env)
diff --git a/pkg/container/container_test.go b/pkg/container/container_test.go
index 8b9ed58..a129afe 100644
--- a/pkg/container/container_test.go
+++ b/pkg/container/container_test.go
@@ -124,6 +124,36 @@ var _ = Describe("the container", func() {
 				}))
 			})
 		})
+		When("container healthcheck config is empty", func() {
+			It("should not panic", func() {
+				c := MockContainer(WithImageHealthcheck(dc.HealthConfig{
+					Test:        []string{"/usr/bin/sleep", "10s"},
+					Interval:    10,
+					Timeout:     60,
+					StartPeriod: 30,
+					Retries:     10,
+				}))
+				Expect(c.GetCreateConfig().Healthcheck).To(BeNil())
+			})
+		})
+		When("container image healthcheck config is empty", func() {
+			It("should not panic", func() {
+				c := MockContainer(WithHealthcheck(dc.HealthConfig{
+					Test:        []string{"/usr/bin/sleep", "1s"},
+					Interval:    30,
+					Timeout:     30,
+					StartPeriod: 10,
+					Retries:     2,
+				}))
+				Expect(c.GetCreateConfig().Healthcheck).To(Equal(&dc.HealthConfig{
+					Test:        []string{"/usr/bin/sleep", "1s"},
+					Interval:    30,
+					Timeout:     30,
+					StartPeriod: 10,
+					Retries:     2,
+				}))
+			})
+		})
 	})
 	When("asked for metadata", func() {
 		var c *Container