Add unit test for DisableCgroup, RestrictOOMScoreAdj.

Signed-off-by: Lantao Liu <lantaol@google.com>
This commit is contained in:
Lantao Liu
2019-01-03 10:47:34 -08:00
parent 0fa8668aa4
commit b1ad4ee9b6
3 changed files with 47 additions and 0 deletions

View File

@@ -304,3 +304,20 @@ systemd_cgroup = true
})
}
}
func TestRestrictOOMScoreAdj(t *testing.T) {
current, err := getCurrentOOMScoreAdj()
require.NoError(t, err)
got, err := restrictOOMScoreAdj(current - 1)
require.NoError(t, err)
assert.Equal(t, got, current)
got, err = restrictOOMScoreAdj(current)
require.NoError(t, err)
assert.Equal(t, got, current)
got, err = restrictOOMScoreAdj(current + 1)
require.NoError(t, err)
assert.Equal(t, got, current+1)
}