@@ -30,7 +30,7 @@ import (
3030)
3131
3232func TestSetPositiveOomScoreAdjustment (t * testing.T ) {
33- adjustment , err := adjustOom (123 )
33+ _ , adjustment , err := adjustOom (123 )
3434 if err != nil {
3535 t .Error (err )
3636 return
@@ -44,7 +44,7 @@ func TestSetNegativeOomScoreAdjustmentWhenPrivileged(t *testing.T) {
4444 return
4545 }
4646
47- adjustment , err := adjustOom (- 123 )
47+ _ , adjustment , err := adjustOom (- 123 )
4848 if err != nil {
4949 t .Error (err )
5050 return
@@ -58,32 +58,37 @@ func TestSetNegativeOomScoreAdjustmentWhenUnprivilegedHasNoEffect(t *testing.T)
5858 return
5959 }
6060
61- adjustment , err := adjustOom (- 123 )
61+ initial , adjustment , err := adjustOom (- 123 )
6262 if err != nil {
6363 t .Error (err )
6464 return
6565 }
66- assert .Check (t , is .Equal (adjustment , 0 ))
66+ assert .Check (t , is .Equal (adjustment , initial ))
6767}
6868
69- func adjustOom (adjustment int ) (int , error ) {
69+ func adjustOom (adjustment int ) (int , int , error ) {
7070 cmd := exec .Command ("sleep" , "100" )
7171 if err := cmd .Start (); err != nil {
72- return 0 , err
72+ return 0 , 0 , err
7373 }
7474
7575 defer cmd .Process .Kill ()
7676
7777 pid , err := waitForPid (cmd .Process )
7878 if err != nil {
79- return 0 , err
79+ return 0 , 0 , err
80+ }
81+ initial , err := GetOOMScoreAdj (pid )
82+ if err != nil {
83+ return 0 , 0 , err
8084 }
8185
8286 if err := SetOOMScore (pid , adjustment ); err != nil {
83- return 0 , err
87+ return 0 , 0 , err
8488 }
8589
86- return GetOOMScoreAdj (pid )
90+ adj , err := GetOOMScoreAdj (pid )
91+ return initial , adj , err
8792}
8893
8994func waitForPid (process * os.Process ) (int , error ) {
0 commit comments