10 "github.com/edwarnicke/exechelper"
14 logDir string = "/tmp/hs-test/"
18 workDir, _ = os.Getwd()
27 type Container struct {
32 extraRunningArgs string
33 volumes map[string]Volume
34 envVars map[string]string
35 vppInstance *VppInstance
38 func NewContainer(yamlInput ContainerConfig) (*Container, error) {
39 containerName := yamlInput["name"].(string)
40 if len(containerName) == 0 {
41 err := fmt.Errorf("container name must not be blank")
45 var container = new(Container)
46 container.volumes = make(map[string]Volume)
47 container.envVars = make(map[string]string)
48 container.name = containerName
50 if image, ok := yamlInput["image"]; ok {
51 container.image = image.(string)
53 container.image = "hs-test/vpp"
56 if args, ok := yamlInput["extra-args"]; ok {
57 container.extraRunningArgs = args.(string)
59 container.extraRunningArgs = ""
62 if isOptional, ok := yamlInput["is-optional"]; ok {
63 container.isOptional = isOptional.(bool)
65 container.isOptional = false
68 if _, ok := yamlInput["volumes"]; ok {
69 r := strings.NewReplacer("$HST_DIR", workDir)
70 for _, volu := range yamlInput["volumes"].([]interface{}) {
71 volumeMap := volu.(ContainerConfig)
72 hostDir := r.Replace(volumeMap["host-dir"].(string))
73 containerDir := volumeMap["container-dir"].(string)
74 isDefaultWorkDir := false
76 if isDefault, ok := volumeMap["is-default-work-dir"]; ok {
77 isDefaultWorkDir = isDefault.(bool)
80 container.addVolume(hostDir, containerDir, isDefaultWorkDir)
85 if _, ok := yamlInput["vars"]; ok {
86 for _, envVar := range yamlInput["vars"].([]interface{}) {
87 envVarMap := envVar.(ContainerConfig)
88 name := envVarMap["name"].(string)
89 value := envVarMap["value"].(string)
90 container.addEnvVar(name, value)
96 func (c *Container) Suite() *HstSuite {
100 func (c *Container) getWorkDirVolume() (res Volume, exists bool) {
101 for _, v := range c.volumes {
102 if v.isDefaultWorkDir {
111 func (c *Container) GetHostWorkDir() (res string) {
112 if v, ok := c.getWorkDirVolume(); ok {
118 func (c *Container) GetContainerWorkDir() (res string) {
119 if v, ok := c.getWorkDirVolume(); ok {
125 func (c *Container) getContainerArguments() string {
126 args := "--cap-add=all --privileged --network host --rm"
127 args += c.getVolumesAsCliOption()
128 args += c.getEnvVarsAsCliOption()
129 args += " --name " + c.name + " " + c.image
130 args += " " + c.extraRunningArgs
134 func (c *Container) create() error {
135 cmd := "docker create " + c.getContainerArguments()
137 return exechelper.Run(cmd)
140 func (c *Container) start() error {
141 cmd := "docker start " + c.name
143 return exechelper.Run(cmd)
146 func (c *Container) run() error {
148 return fmt.Errorf("run container failed: name is blank")
151 cmd := "docker run -d " + c.getContainerArguments()
153 err := exechelper.Run(cmd)
155 return fmt.Errorf("container run failed: %s", err)
161 func (c *Container) addVolume(hostDir string, containerDir string, isDefaultWorkDir bool) {
163 volume.hostDir = hostDir
164 volume.containerDir = containerDir
165 volume.isDefaultWorkDir = isDefaultWorkDir
166 c.volumes[hostDir] = volume
169 func (c *Container) getVolumesAsCliOption() string {
172 if len(c.volumes) > 0 {
173 for _, volume := range c.volumes {
174 cliOption += fmt.Sprintf(" -v %s:%s", volume.hostDir, volume.containerDir)
181 func (c *Container) addEnvVar(name string, value string) {
182 c.envVars[name] = value
185 func (c *Container) getEnvVarsAsCliOption() string {
187 if len(c.envVars) == 0 {
191 for name, value := range c.envVars {
192 cliOption += fmt.Sprintf(" -e %s=%s", name, value)
197 func (c *Container) newVppInstance(additionalConfig ...Stanza) (*VppInstance, error) {
198 vpp := new(VppInstance)
201 if len(additionalConfig) > 0 {
202 vpp.additionalConfig = additionalConfig[0]
210 func (c *Container) copy(sourceFileName string, targetFileName string) error {
211 cmd := exec.Command("docker", "cp", sourceFileName, c.name+":"+targetFileName)
215 func (c *Container) createFile(destFileName string, content string) error {
216 f, err := os.CreateTemp("/tmp", "hst-config")
220 defer os.Remove(f.Name())
222 if _, err := f.Write([]byte(content)); err != nil {
225 if err := f.Close(); err != nil {
228 c.copy(f.Name(), destFileName)
233 * Executes in detached mode so that the started application can continue to run
234 * without blocking execution of test
236 func (c *Container) execServer(command string, arguments ...any) {
237 serverCommand := fmt.Sprintf(command, arguments...)
238 containerExecCommand := "docker exec -d" + c.getEnvVarsAsCliOption() +
239 " " + c.name + " " + serverCommand
240 c.Suite().T().Helper()
241 c.Suite().log(containerExecCommand)
242 c.Suite().assertNil(exechelper.Run(containerExecCommand))
245 func (c *Container) exec(command string, arguments ...any) string {
246 cliCommand := fmt.Sprintf(command, arguments...)
247 containerExecCommand := "docker exec" + c.getEnvVarsAsCliOption() +
248 " " + c.name + " " + cliCommand
249 c.Suite().T().Helper()
250 c.Suite().log(containerExecCommand)
251 byteOutput, err := exechelper.CombinedOutput(containerExecCommand)
252 c.Suite().assertNil(err)
253 return string(byteOutput)
256 func (c *Container) getLogDirPath() string {
257 testId := c.Suite().getTestId()
258 testName := c.Suite().T().Name()
259 logDirPath := logDir + testName + "/" + testId + "/"
261 cmd := exec.Command("mkdir", "-p", logDirPath)
262 if err := cmd.Run(); err != nil {
263 c.Suite().T().Fatalf("mkdir error: %v", err)
269 func (c *Container) saveLogs() {
270 cmd := exec.Command("docker", "inspect", "--format='{{.State.Status}}'", c.name)
271 if output, _ := cmd.CombinedOutput(); !strings.Contains(string(output), "running") {
275 testLogFilePath := c.getLogDirPath() + "container-" + c.name + ".log"
277 cmd = exec.Command("docker", "logs", "--details", "-t", c.name)
278 output, err := cmd.CombinedOutput()
280 c.Suite().T().Fatalf("fetching logs error: %v", err)
283 f, err := os.Create(testLogFilePath)
285 c.Suite().T().Fatalf("file create error: %v", err)
287 fmt.Fprintf(f, string(output))
291 func (c *Container) log() string {
292 cmd := "docker logs " + c.name
294 o, err := exechelper.CombinedOutput(cmd)
295 c.Suite().assertNil(err)
299 func (c *Container) stop() error {
300 if c.vppInstance != nil && c.vppInstance.apiChannel != nil {
301 c.vppInstance.saveLogs()
302 c.vppInstance.disconnect()
306 return exechelper.Run("docker stop " + c.name + " -t 0")
309 func (c *Container) createConfig(targetConfigName string, templateName string, values any) {
310 template := template.Must(template.ParseFiles(templateName))
312 f, err := os.CreateTemp("/tmp/hs-test/", "hst-config")
313 c.Suite().assertNil(err)
314 defer os.Remove(f.Name())
316 err = template.Execute(f, values)
317 c.Suite().assertNil(err)
320 c.Suite().assertNil(err)
322 c.copy(f.Name(), targetConfigName)
326 cmd := exec.Command("mkdir", "-p", logDir)
327 if err := cmd.Run(); err != nil {