96 lines
1.8 KiB
Go
96 lines
1.8 KiB
Go
package integrationutil
|
|
|
|
import (
|
|
"archive/tar"
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"path/filepath"
|
|
|
|
"github.com/juanfont/headscale/integration/dockertestutil"
|
|
"github.com/ory/dockertest/v3"
|
|
"github.com/ory/dockertest/v3/docker"
|
|
)
|
|
|
|
func WriteFileToContainer(
|
|
pool *dockertest.Pool,
|
|
container *dockertest.Resource,
|
|
path string,
|
|
data []byte,
|
|
) error {
|
|
dirPath, fileName := filepath.Split(path)
|
|
|
|
file := bytes.NewReader(data)
|
|
|
|
buf := bytes.NewBuffer([]byte{})
|
|
|
|
tarWriter := tar.NewWriter(buf)
|
|
|
|
header := &tar.Header{
|
|
Name: fileName,
|
|
Size: file.Size(),
|
|
// Mode: int64(stat.Mode()),
|
|
// ModTime: stat.ModTime(),
|
|
}
|
|
|
|
err := tarWriter.WriteHeader(header)
|
|
if err != nil {
|
|
return fmt.Errorf("failed write file header to tar: %w", err)
|
|
}
|
|
|
|
_, err = io.Copy(tarWriter, file)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to copy file to tar: %w", err)
|
|
}
|
|
|
|
err = tarWriter.Close()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to close tar: %w", err)
|
|
}
|
|
|
|
// Ensure the directory is present inside the container
|
|
_, _, err = dockertestutil.ExecuteCommand(
|
|
container,
|
|
[]string{"mkdir", "-p", dirPath},
|
|
[]string{},
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to ensure directory: %w", err)
|
|
}
|
|
|
|
err = pool.Client.UploadToContainer(
|
|
container.Container.ID,
|
|
docker.UploadToContainerOptions{
|
|
NoOverwriteDirNonDir: false,
|
|
Path: dirPath,
|
|
InputStream: bytes.NewReader(buf.Bytes()),
|
|
},
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func FetchPathFromContainer(
|
|
pool *dockertest.Pool,
|
|
container *dockertest.Resource,
|
|
path string,
|
|
) ([]byte, error) {
|
|
buf := bytes.NewBuffer([]byte{})
|
|
|
|
err := pool.Client.DownloadFromContainer(
|
|
container.Container.ID,
|
|
docker.DownloadFromContainerOptions{
|
|
OutputStream: buf,
|
|
Path: path,
|
|
},
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return buf.Bytes(), nil
|
|
}
|