mirror of
https://github.com/fleetdm/fleet
synced 2026-05-23 08:58:41 +00:00
666482245a
> Related issue: #20311 # Checklist for submitter If some of the following don't apply, delete the relevant line. <!-- Note that API documentation changes are now addressed by the product design team. --> - [x] Manual QA for all new/changed functionality - For Orbit and Fleet Desktop changes: - [x] Orbit runs on macOS, Linux and Windows. Check if the orbit feature/bugfix should only apply to one platform (`runtime.GOOS`). - [ ] Manual QA must be performed in the three main OSs, macOS, Windows and Linux. - [x] Auto-update manual QA, from released version of component to new version (see [tools/tuf/test](../tools/tuf/test/README.md)).
97 lines
2.2 KiB
Go
97 lines
2.2 KiB
Go
package useraction
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type dummyHandler struct{}
|
|
|
|
func (d dummyHandler) NotifyRemote() error {
|
|
return nil
|
|
}
|
|
|
|
func (d dummyHandler) ShowInstructions() error { return nil }
|
|
|
|
func TestWaitForUnenrollment(t *testing.T) {
|
|
t.Parallel()
|
|
m := &swiftDialogMDMMigrator{
|
|
handler: dummyHandler{},
|
|
baseDialog: newBaseDialog("foo/bar"),
|
|
frequency: 15 * time.Minute,
|
|
unenrollmentRetryInterval: 300 * time.Millisecond,
|
|
}
|
|
|
|
cases := []struct {
|
|
name string
|
|
enrollErr error
|
|
unenrollAfterNTries int
|
|
wantErr bool
|
|
}{
|
|
{"unenroll after 3 tries", nil, 3, false},
|
|
{"unenroll after one try", nil, 1, false},
|
|
{"error after max number of tries is exceeded", nil, 1000, true},
|
|
{"always error calling profiles func", errors.New("test"), 1, true},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
tries := 0
|
|
m.testEnrollmentCheckFileFn = func() (bool, error) {
|
|
if tries >= c.unenrollAfterNTries {
|
|
return false, c.enrollErr
|
|
}
|
|
tries++
|
|
return true, c.enrollErr
|
|
}
|
|
|
|
m.testEnrollmentCheckStatusFn = func() (bool, string, error) {
|
|
return true, "example.com", nil
|
|
}
|
|
|
|
outErr := m.waitForUnenrollment(true)
|
|
if c.wantErr {
|
|
require.Error(t, outErr)
|
|
} else {
|
|
require.NoError(t, outErr)
|
|
require.Equal(t, c.unenrollAfterNTries, tries)
|
|
}
|
|
})
|
|
}
|
|
|
|
t.Run("fallback to enrollment check file", func(t *testing.T) {
|
|
m.testEnrollmentCheckFileFn = func() (bool, error) {
|
|
return true, nil
|
|
}
|
|
|
|
m.testEnrollmentCheckStatusFn = func() (bool, string, error) {
|
|
return false, "", nil
|
|
}
|
|
|
|
outErr := m.waitForUnenrollment(true)
|
|
require.NoError(t, outErr)
|
|
})
|
|
|
|
t.Run("only check file during ADE enrollment", func(t *testing.T) {
|
|
var fileWasChecked bool
|
|
m.testEnrollmentCheckFileFn = func() (bool, error) {
|
|
fileWasChecked = true
|
|
return true, nil
|
|
}
|
|
|
|
m.testEnrollmentCheckStatusFn = func() (bool, string, error) {
|
|
return false, "", nil
|
|
}
|
|
|
|
err := m.waitForUnenrollment(false)
|
|
require.NoError(t, err)
|
|
require.False(t, fileWasChecked)
|
|
|
|
err = m.waitForUnenrollment(true)
|
|
require.NoError(t, err)
|
|
require.True(t, fileWasChecked)
|
|
})
|
|
}
|