package/systemctl
1
0
Fork 0
mirror of https://github.com/taigrr/systemctl.git synced 2025-10-05 16:26:58 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add masked unit tests for enable and disable

Browse Source
This commit is contained in:
Tai Groot
2021-05-15 20:02:05 -07:00
parent 5865d30ada
commit 34f11bfcd1
4 changed files with 55 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
systemctl_test.go
View File

@@ -62,12 +62,13 @@ func TestEnable(t *testing.T) {
}
for _, tc := range testCases {
t.Run(fmt.Sprintf("%s as %s", tc.unit, userString), func(t *testing.T) {
t.Parallel()
if (userString == "root" || userString == "system") && tc.runAsUser {
t.Skip("skipping user test while running as superuser")
} else if (userString != "root" && userString != "system") && !tc.runAsUser {
t.Skip("skipping superuser test while running as user")
}
ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
err := Enable(ctx, tc.unit, tc.opts)
if err != tc.err {
@@ -75,6 +76,28 @@ func TestEnable(t *testing.T) {
}
})
}
t.Run(fmt.Sprintf(""), func(t *testing.T) {
if userString != "root" && userString != "system" {
t.Skip("skipping superuser test while running as user")
}
unit := "nginx"
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
err := Mask(ctx, unit, Options{usermode: false})
defer cancel()
if err != nil {
Unmask(ctx, unit, Options{usermode: false})
t.Errorf("Unable to mask %s", unit)
}
err = Enable(ctx, unit, Options{usermode: false})
if err != ErrMasked {
Unmask(ctx, unit, Options{usermode: false})
t.Errorf("error is %v, but should have been %v", err, ErrMasked)
}
err = Unmask(ctx, unit, Options{usermode: false})
if err != nil {
t.Errorf("Unable to unmask %s", unit)
}
})
}
@@ -120,6 +143,28 @@ func TestDisable(t *testing.T) {
}
})
}
t.Run(fmt.Sprintf(""), func(t *testing.T) {
if userString != "root" && userString != "system" {
t.Skip("skipping superuser test while running as user")
}
unit := "nginx"
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
err := Mask(ctx, unit, Options{usermode: false})
defer cancel()
if err != nil {
Unmask(ctx, unit, Options{usermode: false})
t.Errorf("Unable to mask %s", unit)
}
err = Disable(ctx, unit, Options{usermode: false})
if err != ErrMasked {
Unmask(ctx, unit, Options{usermode: false})
t.Errorf("error is %v, but should have been %v", err, ErrMasked)
}
err = Unmask(ctx, unit, Options{usermode: false})
if err != nil {
t.Errorf("Unable to unmask %s", unit)
}
})
}