Merge pull request #9445 from jmguzik/no-header-info-for-systemd-generation

No header info for systemd generation

1 files changed, 50 insertions, 9 deletions

diff --git a/pkg/systemd/generate/containers_test.go b/pkg/systemd/generate/containers_test.go
index c007c6eb4..747c54699 100644
--- a/pkg/systemd/generate/containers_test.go
+++ b/pkg/systemd/generate/containers_test.go
@@ -37,9 +37,11 @@ func TestValidateRestartPolicyContainer(t *testing.T) {
 }
 
 func TestCreateContainerSystemdUnit(t *testing.T) {
-	goodID := `# container-639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401.service
-# autogenerated by Podman CI
-
+	serviceInfo := `# container-639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401.service
+`
+	headerInfo := `# autogenerated by Podman CI
+`
+	goodIDContent := `
 [Unit]
 Description=Podman container-639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401.service
 Documentation=man:podman-generate-systemd(1)
@@ -59,6 +61,8 @@ Type=forking
 [Install]
 WantedBy=multi-user.target default.target
 `
+	goodID := serviceInfo + headerInfo + goodIDContent
+	goodIDNoHeaderInfo := serviceInfo + goodIDContent
 
 	goodName := `# container-foobar.service
 # autogenerated by Podman CI
@@ -377,11 +381,12 @@ Type=forking
 WantedBy=multi-user.target default.target
 `
 	tests := []struct {
-		name    string
-		info    containerInfo
-		want    string
-		new     bool
-		wantErr bool
+		name     string
+		info     containerInfo
+		want     string
+		new      bool
+		noHeader bool
+		wantErr  bool
 	}{
 
 		{"good with id",
@@ -398,6 +403,23 @@ WantedBy=multi-user.target default.target
 			goodID,
 			false,
 			false,
+			false,
+		},
+		{"good with noHeader",
+			containerInfo{
+				Executable:        "/usr/bin/podman",
+				ServiceName:       "container-639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401",
+				ContainerNameOrID: "639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401",
+				RestartPolicy:     "always",
+				PIDFile:           "/run/containers/storage/overlay-containers/639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401/userdata/conmon.pid",
+				StopTimeout:       22,
+				PodmanVersion:     "CI",
+				EnvVariable:       EnvVariable,
+			},
+			goodIDNoHeaderInfo,
+			false,
+			true,
+			false,
 		},
 		{"good with name",
 			containerInfo{
@@ -413,6 +435,7 @@ WantedBy=multi-user.target default.target
 			goodName,
 			false,
 			false,
+			false,
 		},
 		{"good with name and bound to",
 			containerInfo{
@@ -429,6 +452,7 @@ WantedBy=multi-user.target default.target
 			goodNameBoundTo,
 			false,
 			false,
+			false,
 		},
 		{"bad restart policy",
 			containerInfo{
@@ -442,6 +466,7 @@ WantedBy=multi-user.target default.target
 			},
 			"",
 			false,
+			false,
 			true,
 		},
 		{"good with name and generic",
@@ -459,6 +484,7 @@ WantedBy=multi-user.target default.target
 			goodWithNameAndGeneric,
 			true,
 			false,
+			false,
 		},
 		{"good with explicit short detach param",
 			containerInfo{
@@ -475,6 +501,7 @@ WantedBy=multi-user.target default.target
 			goodWithExplicitShortDetachParam,
 			true,
 			false,
+			false,
 		},
 		{"good with explicit short detach param and podInfo",
 			containerInfo{
@@ -494,6 +521,7 @@ WantedBy=multi-user.target default.target
 			goodNameNewWithPodFile,
 			true,
 			false,
+			false,
 		},
 		{"good with explicit full detach param",
 			containerInfo{
@@ -510,6 +538,7 @@ WantedBy=multi-user.target default.target
 			goodNameNewDetach,
 			true,
 			false,
+			false,
 		},
 		{"good with id and no param",
 			containerInfo{
@@ -526,6 +555,7 @@ WantedBy=multi-user.target default.target
 			goodIDNew,
 			true,
 			false,
+			false,
 		},
 		{"good with explicit detach=true param",
 			containerInfo{
@@ -542,6 +572,7 @@ WantedBy=multi-user.target default.target
 			genGoodNewDetach("--detach=true"),
 			true,
 			false,
+			false,
 		},
 		{"good with explicit detach=false param",
 			containerInfo{
@@ -558,6 +589,7 @@ WantedBy=multi-user.target default.target
 			genGoodNewDetach("-d"),
 			true,
 			false,
+			false,
 		},
 		{"good with explicit detach=false param",
 			containerInfo{
@@ -574,6 +606,7 @@ WantedBy=multi-user.target default.target
 			goodNameNewDetachFalseWithCmd,
 			true,
 			false,
+			false,
 		},
 		{"good with multiple detach=false params",
 			containerInfo{
@@ -590,6 +623,7 @@ WantedBy=multi-user.target default.target
 			goodNameNewDetachFalseWithCmd,
 			true,
 			false,
+			false,
 		},
 		{"good with multiple shorthand params detach first",
 			containerInfo{
@@ -606,6 +640,7 @@ WantedBy=multi-user.target default.target
 			genGoodNewDetach("-dti"),
 			true,
 			false,
+			false,
 		},
 		{"good with multiple shorthand params detach last",
 			containerInfo{
@@ -622,6 +657,7 @@ WantedBy=multi-user.target default.target
 			genGoodNewDetach("-tid"),
 			true,
 			false,
+			false,
 		},
 		{"good with root flags",
 			containerInfo{
@@ -638,6 +674,7 @@ WantedBy=multi-user.target default.target
 			goodNewRootFlags,
 			true,
 			false,
+			false,
 		},
 		{"good with container create",
 			containerInfo{
@@ -654,6 +691,7 @@ WantedBy=multi-user.target default.target
 			goodContainerCreate,
 			true,
 			false,
+			false,
 		},
 		{"good with journald log tag (see #9034)",
 			containerInfo{
@@ -670,6 +708,7 @@ WantedBy=multi-user.target default.target
 			goodNewWithJournaldTag,
 			true,
 			false,
+			false,
 		},
 		{"good with special chars",
 			containerInfo{
@@ -686,13 +725,15 @@ WantedBy=multi-user.target default.target
 			goodNewWithSpecialChars,
 			true,
 			false,
+			false,
 		},
 	}
 	for _, tt := range tests {
 		test := tt
 		t.Run(tt.name, func(t *testing.T) {
 			opts := entities.GenerateSystemdOptions{
-				New: test.new,
+				New:      test.new,
+				NoHeader: test.noHeader,
 			}
 			got, err := executeContainerTemplate(&test.info, opts)
 			if (err != nil) != test.wantErr {