Test disable function

This commit is contained in:
Tai Groot
2021-05-15 19:14:53 -07:00
parent b1346b7992
commit 184bbc4936
2 changed files with 47 additions and 2 deletions

View File

@@ -78,6 +78,51 @@ func TestEnable(t *testing.T) {
}
func TestDisable(t *testing.T) {
testCases := []struct {
unit string
err error
opts Options
runAsUser bool
}{
// Run these tests only as a user
//try nonexistant unit in user mode as user
{"nonexistant", ErrDoesNotExist, Options{usermode: true}, true},
// try existing unit in user mode as user
{"syncthing", nil, Options{usermode: true}, true},
// try nonexisting unit in system mode as user
{"nonexistant", ErrInsufficientPermissions, Options{usermode: false}, true},
// try existing unit in system mode as user
{"nginx", ErrInsufficientPermissions, Options{usermode: false}, true},
// Run these tests only as a superuser
// try nonexistant unit in system mode as system
{"nonexistant", ErrDoesNotExist, Options{usermode: false}, false},
// try existing unit in system mode as system
{"nginx", ErrBusFailure, Options{usermode: true}, false},
// try existing unit in system mode as system
{"nginx", nil, Options{usermode: false}, false},
}
for _, tc := range testCases {
t.Run(fmt.Sprintf("%s as %s", tc.unit, userString), func(t *testing.T) {
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)
defer cancel()
err := Disable(ctx, tc.unit, tc.opts)
if err != tc.err {
t.Errorf("error is %v, but should have been %v", err, tc.err)
}
})
}
}
// Runs through all defined properties and checks for error cases
func TestShow(t *testing.T) {
if testing.Short() {