vrr/Pulse
2
0
Fork 0
mirror of https://github.com/rcourtman/Pulse.git synced 2026-04-28 03:20:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Fix grant refresh callback race test

Browse source
This commit is contained in:
rcourtman 2026-04-11 19:15:27 +01:00
parent 47b6d0fb1c
commit 166b4768db
1 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

13
pkg/licensing/grant_refresh_test.go
View file

@ -141,9 +141,11 @@ func TestGrantRefreshLoop_401ClearsActivation(t *testing.T) {
svc := NewService()
svc.SetLicenseServerClient(NewLicenseServerClient(server.URL))
svc.SetPersistence(p)
var callbackState *ActivationState
var callbackState atomic.Pointer[ActivationState]
var callbackCalled atomic.Bool
svc.SetActivationStateChangeCallback(func(state *ActivationState) {
callbackState = state
callbackCalled.Store(true)
callbackState.Store(state)
})
state := &ActivationState{
@ -202,8 +204,11 @@ func TestGrantRefreshLoop_401ClearsActivation(t *testing.T) {
if svc.IsActivated() {
t.Error("expected activation cleared after 401 revocation")
}
if callbackState != nil {
t.Fatalf("expected nil activation-state callback after 401 revocation, got %+v", callbackState)
if !callbackCalled.Load() {
t.Fatal("expected activation-state callback after 401 revocation")
}
if state := callbackState.Load(); state != nil {
t.Fatalf("expected nil activation-state callback after 401 revocation, got %+v", state)
}
// Verify the persisted state was also cleared.