testutil: set -500 OOM score for test daemons

Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
JunGuoCS · Oct 16, 2020 · 561b801 · 561b801
1 parent 2b8e68e
commit 561b801
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/testutil/daemon/daemon.go b/testutil/daemon/daemon.go
@@ -88,6 +88,7 @@ type Daemon struct {
 	DefaultAddrPool []string
 	SubnetSize      uint32
 	DataPathPort    uint32
+	OOMScoreAdjust  int
 	// cached information
 	CachedInfo types.Info
 }
@@ -206,6 +207,7 @@ func New(t testing.TB, ops ...Option) *Daemon {
 		}
 		ops = append(ops, WithRootlessUser("unprivilegeduser"))
 	}
+	ops = append(ops, WithOOMScoreAdjust(-500))
 
 	d, err := NewDaemon(dest, ops...)
 	assert.NilError(t, err, "could not create daemon at %q", dest)

diff --git a/testutil/daemon/ops.go b/testutil/daemon/ops.go
@@ -115,3 +115,10 @@ func WithRootlessUser(username string) Option {
 		d.rootlessUser = u
 	}
 }
+
+// WithOOMScoreAdjust sets OOM score for the daemon
+func WithOOMScoreAdjust(score int) Option {
+	return func(d *Daemon) {
+		d.OOMScoreAdjust = score
+	}
+}