From 9ec8a60eae8987f06d1971634c438586534d2987 Mon Sep 17 00:00:00 2001
From: zhangguanzhang <zhangguanzhang@qq.com>
Date: Mon, 31 Aug 2020 22:12:35 +0800
Subject: handle play kube with pod.spec.hostAliases

Signed-off-by: zhangguanzhang <zhangguanzhang@qq.com>
---
 test/e2e/play_kube_test.go | 49 +++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 48 insertions(+), 1 deletion(-)

(limited to 'test')

diff --git a/test/e2e/play_kube_test.go b/test/e2e/play_kube_test.go
index 1379cb35d..121cea017 100644
--- a/test/e2e/play_kube_test.go
+++ b/test/e2e/play_kube_test.go
@@ -42,6 +42,14 @@ metadata:
 
 spec:
   hostname: {{ .Hostname }}
+  hostAliases:
+{{ range .HostAliases }}
+  - hostnames:
+  {{ range .HostName }}
+    - {{ . }}
+  {{ end }}
+    ip: {{ .IP }}
+{{ end }}
   containers:
 {{ with .Ctrs }}
   {{ range . }}
@@ -249,16 +257,22 @@ func generateDeploymentKubeYaml(deployment *Deployment, fileName string) error {
 type Pod struct {
 	Name        string
 	Hostname    string
+	HostAliases []HostAlias
 	Ctrs        []*Ctr
 	Volumes     []*Volume
 	Annotations map[string]string
 }
 
+type HostAlias struct {
+	IP       string
+	HostName []string
+}
+
 // getPod takes a list of podOptions and returns a pod with sane defaults
 // and the configured options
 // if no containers are added, it will add the default container
 func getPod(options ...podOption) *Pod {
-	p := Pod{defaultPodName, "", make([]*Ctr, 0), make([]*Volume, 0), make(map[string]string)}
+	p := Pod{defaultPodName, "", nil, make([]*Ctr, 0), make([]*Volume, 0), make(map[string]string)}
 	for _, option := range options {
 		option(&p)
 	}
@@ -276,6 +290,15 @@ func withHostname(h string) podOption {
 	}
 }
 
+func withHostAliases(ip string, host []string) podOption {
+	return func(pod *Pod) {
+		pod.HostAliases = append(pod.HostAliases, HostAlias{
+			IP:       ip,
+			HostName: host,
+		})
+	}
+}
+
 func withCtr(c *Ctr) podOption {
 	return func(pod *Pod) {
 		pod.Ctrs = append(pod.Ctrs, c)
@@ -597,6 +620,30 @@ var _ = Describe("Podman generate kube", func() {
 		Expect(inspect.OutputToString()).To(Equal(hostname))
 	})
 
+	It("podman play kube test HostAliases", func() {
+		pod := getPod(withHostAliases("192.168.1.2", []string{
+			"test1.podman.io",
+			"test2.podman.io",
+		}),
+			withHostAliases("192.168.1.3", []string{
+				"test3.podman.io",
+				"test4.podman.io",
+			}),
+		)
+		err := generatePodKubeYaml(pod, kubeYaml)
+		Expect(err).To(BeNil())
+
+		kube := podmanTest.Podman([]string{"play", "kube", kubeYaml})
+		kube.WaitWithDefaultTimeout()
+		Expect(kube.ExitCode()).To(Equal(0))
+
+		inspect := podmanTest.Podman([]string{"inspect", getCtrNameInPod(pod), "--format", "{{ .HostConfig.ExtraHosts }}"})
+		inspect.WaitWithDefaultTimeout()
+		Expect(inspect.ExitCode()).To(Equal(0))
+		Expect(inspect.OutputToString()).
+			To(Equal("[test1.podman.io:192.168.1.2 test2.podman.io:192.168.1.2 test3.podman.io:192.168.1.3 test4.podman.io:192.168.1.3]"))
+	})
+
 	It("podman play kube cap add", func() {
 		capAdd := "CAP_SYS_ADMIN"
 		ctr := getCtr(withCapAdd([]string{capAdd}), withCmd([]string{"cat", "/proc/self/status"}), withArg(nil))
-- 
cgit v1.2.3-54-g00ecf