From 51da191b128063f136721785a7606be93e67d141 Mon Sep 17 00:00:00 2001 From: Shishir Mahajan Date: Wed, 17 Nov 2021 12:47:54 -0800 Subject: [PATCH] pids_limit: Add test. Signed-off-by: Shishir Mahajan --- drivers/docker/driver_linux_test.go | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/drivers/docker/driver_linux_test.go b/drivers/docker/driver_linux_test.go index eb526edfa700..45ae7587aa51 100644 --- a/drivers/docker/driver_linux_test.go +++ b/drivers/docker/driver_linux_test.go @@ -45,6 +45,27 @@ func TestDockerDriver_authFromHelper(t *testing.T) { require.Equal(t, "registry.local:5000", string(content)) } +func TestDockerDriver_PluginConfig_PidsLimit(t *testing.T) { + t.Parallel() + + dh := dockerDriverHarness(t, nil) + driver := dh.Impl().(*Driver) + driver.config.PidsLimit = 5 + + task, cfg, ports := dockerTask(t) + defer freeport.Return(ports) + require.NoError(t, task.EncodeConcreteDriverConfig(cfg)) + + c, err := driver.createContainerConfig(task, cfg, "org/repo:0.1") + require.NoError(t, err) + require.Equal(t, 5, c.HostConfig.PidsLimit) + + cfg.PidsLimit = 7 + _, err = driver.createContainerConfig(task, cfg, "org/repo:0.1") + require.Error(t, err) + require.Contains(t, err.Error(), `pids_limit cannot be greater than nomad plugin config pids_limit`) +} + func TestDockerDriver_PidsLimit(t *testing.T) { if !tu.IsCI() { t.Parallel()