github.com/anth0d/nomad@v0.0.0-20221214183521-ae3a0a2cad06/drivers/docker/driver_linux_test.go (about)

     1  //go:build linux
     2  
     3  package docker
     4  
     5  import (
     6  	"fmt"
     7  	"io/ioutil"
     8  	"os"
     9  	"path/filepath"
    10  	"strings"
    11  	"testing"
    12  
    13  	"github.com/hashicorp/nomad/ci"
    14  	"github.com/hashicorp/nomad/client/testutil"
    15  	"github.com/hashicorp/nomad/helper/freeport"
    16  	"github.com/hashicorp/nomad/helper/pointer"
    17  	tu "github.com/hashicorp/nomad/testutil"
    18  	"github.com/stretchr/testify/require"
    19  )
    20  
    21  func TestDockerDriver_authFromHelper(t *testing.T) {
    22  	testutil.DockerCompatible(t)
    23  
    24  	dir := t.TempDir()
    25  	helperPayload := "{\"Username\":\"hashi\",\"Secret\":\"nomad\"}"
    26  	helperContent := []byte(fmt.Sprintf("#!/bin/sh\ncat > %s/helper-$1.out;echo '%s'", dir, helperPayload))
    27  
    28  	helperFile := filepath.Join(dir, "docker-credential-testnomad")
    29  	err := ioutil.WriteFile(helperFile, helperContent, 0777)
    30  	require.NoError(t, err)
    31  
    32  	path := os.Getenv("PATH")
    33  	t.Setenv("PATH", fmt.Sprintf("%s:%s", path, dir))
    34  
    35  	authHelper := authFromHelper("testnomad")
    36  	creds, err := authHelper("registry.local:5000/repo/image")
    37  	require.NoError(t, err)
    38  	require.NotNil(t, creds)
    39  	require.Equal(t, "hashi", creds.Username)
    40  	require.Equal(t, "nomad", creds.Password)
    41  
    42  	if _, err := os.Stat(filepath.Join(dir, "helper-get.out")); os.IsNotExist(err) {
    43  		t.Fatalf("Expected helper-get.out to exist")
    44  	}
    45  	content, err := ioutil.ReadFile(filepath.Join(dir, "helper-get.out"))
    46  	require.NoError(t, err)
    47  	require.Equal(t, "registry.local:5000", string(content))
    48  }
    49  
    50  func TestDockerDriver_PluginConfig_PidsLimit(t *testing.T) {
    51  	ci.Parallel(t)
    52  	testutil.DockerCompatible(t)
    53  
    54  	dh := dockerDriverHarness(t, nil)
    55  	driver := dh.Impl().(*Driver)
    56  	driver.config.PidsLimit = 5
    57  
    58  	task, cfg, ports := dockerTask(t)
    59  	defer freeport.Return(ports)
    60  	require.NoError(t, task.EncodeConcreteDriverConfig(cfg))
    61  
    62  	cfg.PidsLimit = 7
    63  	_, err := driver.createContainerConfig(task, cfg, "org/repo:0.1")
    64  	require.Error(t, err)
    65  	require.Contains(t, err.Error(), `pids_limit cannot be greater than nomad plugin config pids_limit`)
    66  
    67  	// Task PidsLimit should override plugin PidsLimit.
    68  	cfg.PidsLimit = 3
    69  	opts, err := driver.createContainerConfig(task, cfg, "org/repo:0.1")
    70  	require.NoError(t, err)
    71  	require.Equal(t, pointer.Of(int64(3)), opts.HostConfig.PidsLimit)
    72  }
    73  
    74  func TestDockerDriver_PidsLimit(t *testing.T) {
    75  	ci.Parallel(t)
    76  	testutil.DockerCompatible(t)
    77  
    78  	task, cfg, ports := dockerTask(t)
    79  	defer freeport.Return(ports)
    80  
    81  	cfg.PidsLimit = 1
    82  	cfg.Command = "/bin/sh"
    83  	cfg.Args = []string{"-c", "sleep 5 & sleep 5 & sleep 5"}
    84  	require.NoError(t, task.EncodeConcreteDriverConfig(cfg))
    85  
    86  	_, _, _, cleanup := dockerSetup(t, task, nil)
    87  	defer cleanup()
    88  
    89  	// Check that data was written to the directory.
    90  	outputFile := filepath.Join(task.TaskDir().LogDir, "redis-demo.stderr.0")
    91  	exp := "can't fork"
    92  	tu.WaitForResult(func() (bool, error) {
    93  		act, err := ioutil.ReadFile(outputFile)
    94  		if err != nil {
    95  			return false, err
    96  		}
    97  		if !strings.Contains(string(act), exp) {
    98  			return false, fmt.Errorf("Expected %q in output %q", exp, string(act))
    99  		}
   100  		return true, nil
   101  	}, func(err error) {
   102  		require.NoError(t, err)
   103  	})
   104  }