mirror of
https://github.com/fleetdm/fleet
synced 2026-05-21 16:08:47 +00:00
122 lines
3.4 KiB
Go
122 lines
3.4 KiB
Go
package update
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/fleetdm/fleet/v4/server/fleet"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestRenewEnrollmentProfile(t *testing.T) {
|
|
var logBuf bytes.Buffer
|
|
|
|
oldLog := log.Logger
|
|
log.Logger = log.Output(&logBuf)
|
|
t.Cleanup(func() { log.Logger = oldLog })
|
|
|
|
cases := []struct {
|
|
desc string
|
|
renewFlag bool
|
|
cmdErr error
|
|
wantCmdCalled bool
|
|
wantLog string
|
|
}{
|
|
{"renew=false", false, nil, false, ""},
|
|
{"renew=true; success", true, nil, true, "successfully called /usr/bin/profiles to renew enrollment profile"},
|
|
{"renew=true; fail", true, io.ErrUnexpectedEOF, true, "calling /usr/bin/profiles to renew enrollment profile failed"},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.desc, func(t *testing.T) {
|
|
logBuf.Reset()
|
|
|
|
fetcher := &dummyConfigFetcher{
|
|
cfg: &fleet.OrbitConfig{Notifications: fleet.OrbitConfigNotifications{RenewEnrollmentProfile: c.renewFlag}},
|
|
}
|
|
|
|
var cmdGotCalled bool
|
|
renewFetcher := &RenewEnrollmentProfileConfigFetcher{
|
|
Fetcher: fetcher,
|
|
Frequency: time.Hour, // doesn't matter for this test
|
|
runCmdFn: func() error {
|
|
cmdGotCalled = true
|
|
return c.cmdErr
|
|
},
|
|
}
|
|
|
|
cfg, err := renewFetcher.GetConfig()
|
|
require.NoError(t, err) // the dummy fetcher never returns an error
|
|
require.Equal(t, fetcher.cfg, cfg) // the renew enrollment wrapper properly returns the expected config
|
|
|
|
require.Equal(t, c.wantCmdCalled, cmdGotCalled)
|
|
require.Contains(t, logBuf.String(), c.wantLog)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRenewEnrollmentProfilePrevented(t *testing.T) {
|
|
var logBuf bytes.Buffer
|
|
|
|
oldLog := log.Logger
|
|
log.Logger = log.Output(&logBuf)
|
|
t.Cleanup(func() { log.Logger = oldLog })
|
|
|
|
fetcher := &dummyConfigFetcher{
|
|
cfg: &fleet.OrbitConfig{Notifications: fleet.OrbitConfigNotifications{RenewEnrollmentProfile: true}},
|
|
}
|
|
|
|
var cmdCallCount int
|
|
chProceed := make(chan struct{})
|
|
renewFetcher := &RenewEnrollmentProfileConfigFetcher{
|
|
Fetcher: fetcher,
|
|
Frequency: 2 * time.Second, // just to be safe with slow environments (CI)
|
|
runCmdFn: func() error {
|
|
<-chProceed // will be unblocked only when allowed
|
|
cmdCallCount++ // no need for sync, single-threaded call of this func is guaranteed by the fetcher's mutex
|
|
return nil
|
|
},
|
|
}
|
|
|
|
assertResult := func(cfg *fleet.OrbitConfig, err error) {
|
|
require.NoError(t, err)
|
|
require.Equal(t, fetcher.cfg, cfg)
|
|
}
|
|
|
|
started := make(chan struct{})
|
|
go func() {
|
|
close(started)
|
|
|
|
// the first call will block in runCmdFn
|
|
cfg, err := renewFetcher.GetConfig()
|
|
assertResult(cfg, err)
|
|
}()
|
|
|
|
<-started
|
|
// this call will happen while the first call is blocked in runCmdFn, so it
|
|
// won't call the command (won't be able to lock the mutex). However it will
|
|
// still complete successfully without being blocked by the other call in
|
|
// progress.
|
|
cfg, err := renewFetcher.GetConfig()
|
|
assertResult(cfg, err)
|
|
|
|
// unblock the first call
|
|
close(chProceed)
|
|
|
|
// this next call won't execute the command because of the frequency
|
|
// restriction (it got called less than N seconds ago)
|
|
cfg, err = renewFetcher.GetConfig()
|
|
assertResult(cfg, err)
|
|
|
|
// wait for the fetcher's frequency to pass
|
|
time.Sleep(renewFetcher.Frequency)
|
|
|
|
// this call executes the command
|
|
cfg, err = renewFetcher.GetConfig()
|
|
assertResult(cfg, err)
|
|
|
|
require.Equal(t, 2, cmdCallCount) // the initial call and the one after sleep
|
|
}
|